|
7
|
1 """
|
|
|
2 Project that can be stored to and loaded from XML.
|
|
|
3 """
|
|
|
4
|
|
|
5 from xml.sax import ContentHandler, make_parser
|
|
|
6 import xml.dom.minidom as md
|
|
|
7 import os.path
|
|
|
8
|
|
|
9 class ProjectContentHandler(ContentHandler):
|
|
|
10 def __init__(self, project):
|
|
|
11 self.project = project
|
|
|
12 self.inFiles = False
|
|
|
13 def startElement(self, name, attrs):
|
|
|
14 if name == 'Project':
|
|
|
15 self.project.name = attrs['name']
|
|
|
16 if name == 'Files':
|
|
|
17 self.inFiles = True
|
|
|
18 if name == 'File' and self.inFiles:
|
|
|
19 self.project.files.append(attrs['Filename'])
|
|
|
20 def endElement(self, name):
|
|
|
21 if name == 'Files':
|
|
|
22 self.inFiles = False
|
|
|
23
|
|
|
24 class Project:
|
|
|
25 def __init__(self):
|
|
|
26 self.name = ""
|
|
|
27 self.files = []
|
|
|
28 self.settings = {}
|
|
8
|
29 self.filename = ""
|
|
7
|
30
|
|
|
31 def save(self, filename):
|
|
|
32 """ Save the project in XML format """
|
|
|
33 # Create document:
|
|
|
34 doc = md.Document()
|
|
|
35 # Add project:
|
|
|
36 project = doc.createElement("Project")
|
|
|
37 project.setAttribute("name", self.name)
|
|
|
38 doc.appendChild(project)
|
|
|
39
|
|
|
40 # Add project files:
|
|
|
41 filesNode = doc.createElement("Files")
|
|
|
42 project.appendChild(filesNode)
|
|
|
43 for f in self.files:
|
|
|
44 fe = doc.createElement("File")
|
|
|
45 fe.setAttribute("Filename", f)
|
|
|
46 filesNode.appendChild(fe)
|
|
|
47
|
|
|
48 # Write the XML file:
|
|
|
49 xml = doc.toprettyxml()
|
|
|
50 with open(filename, 'w') as f:
|
|
|
51 f.write(xml)
|
|
|
52
|
|
|
53 def load(self, filename):
|
|
|
54 """ Load the project from the XML file """
|
|
|
55 if not os.path.exists(filename):
|
|
|
56 return
|
|
8
|
57 self.filename = filename
|
|
|
58 print(self.filename)
|
|
7
|
59 parser = make_parser()
|
|
|
60 handler = ProjectContentHandler(self)
|
|
|
61 parser.setContentHandler(handler)
|
|
|
62 parser.parse(filename)
|
|
|
63
|
