Commit before the split.

1 files changed, 89 insertions, 0 deletions

diff --git a/model.py b/model.py
new file mode 100644
index 0000000..723e27c
--- /dev/null
+++ b/model.py
@@ -0,0 +1,89 @@
+import libxml2
+import os
+import logging
+
+class Game(object):
+    def __init__(self, gamepath, dtdpath, name='noname'):
+        self.name = name
+        self.pairs = {}
+        self.gamepath = gamepath
+        self.dtdpath = dtdpath
+
+        try:
+            self.dtd = libxml2.parseDTD(None, os.path.join(self.dtdpath, 'memosono.dtd'))
+        except libxml2.parserError, e:
+            logging.error('No memosono.dtd found ' +str(e))
+            self.dtd = None
+        self.ctxt = libxml2.newValidCtxt()               
+        
+    def read(self, filename):
+        try:
+            doc = libxml2.parseFile(os.path.join(self.gamepath, filename))            
+            if doc.validateDtd(self.ctxt, self.dtd):
+        
+                # get the requested nodes
+                xpa = doc.xpathNewContext()
+                res = xpa.xpathEval("//*")
+
+                # write their content to the data structure
+                for elem in res:
+                    attributes = elem.get_properties()
+                    pair = []
+                    idpair = ''
+                    for attribute in attributes:
+                        if(attribute.name == 'text'):
+                            pass
+                        if(attribute.name == 'id'):
+                            idpair = attribute.content
+                        if(attribute.name == 'mother'):
+                            pair.append(attribute.content)
+                        if(attribute.name == 'child'):
+                            pair.append(attribute.content)
+                        if(attribute.name == 'color'):
+                            pair.append(int(attribute.content))
+                        if( elem.name == 'memosono' ):    
+                            self.name = attribute.content            
+                        if( elem.name != 'memosono' ):
+                            self.pairs[idpair] = pair
+                xpa.xpathFreeContext()
+            else:
+                logging.error('Error in validation of the file')
+            doc.freeDoc()
+        except libxml2.parserError, e:
+            logging.error('Error parsing file ' +str(e))
+            
+    def save(self, filename):
+        doc = libxml2.newDoc("1.0")
+        root = doc.newChild(None, "memosono", None)
+        root.setProp("name", self.name)
+        for key in self.pairs:
+            
+            elem = root.newChild(None, "pair", None)
+            elem.setProp("id", key)
+            elem.setProp("mother", self.pairs[key][0])
+            elem.setProp("child", self.pairs[key][1])
+            elem.setProp("color", self.pairs[key][2])
+        
+        if doc.validateDtd(self.ctxt, self.dtd):
+            doc.saveFormatFile(filename, 1)
+        doc.freeDoc()    
+            
+if __name__ == '__main__':
+    
+    print "[Read game from file] "        
+    game = Game()
+    game.read('memosono.xml')
+    print "   name=" + game.name
+    print "   pairs=%s" %game.pairs
+
+    elemkey = '0'
+    if game.pairs.has_key(elemkey):
+        del game.pairs[elemkey]
+        
+    game.pairs['2'] = ['frettchen.jpg', 'frettchen.wav']
+
+    print "[Write game to file] "        
+    game.save('memosono.xml')
+        
+
+