import os class Scene: def __init__(self, name, size): self.name = name self.items = [] self.size = size def add(self, item): self.items.append(item) def svgstr(self): svgstr = "\n" svgstr += "\n"%self.size svgstr += " \n" for item in self.items: svgstr += item.svgstr() svgstr += " \n\n" return svgstr def write(self, filename=None): if not filename: filename = self.name + ".svg" with open(filename, "w") as f: f.write(self.svgstr()) class Rectangle: def __init__(self, pos, size, color): self.pos = pos self.size = size self.color = color def svgstr(self): svgstr = " \n"%self.color return svgstr if __name__ == "__main__": scene = Scene('test', (400, 400)) scene.add(Rectangle((0,0),(100,100),(255,0,0))) scene.write()