X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=sfa%2Futil%2Fstorage.py;h=9b682064eed12f71068bf94e4738ce2ab6cb3581;hb=052e281f8c79237b6759cbb259407f071adda1cc;hp=5d915398efc6df679733afd8181f811853b63f35;hpb=5f4f788db1dd59e8a5968e9118ff1feda89389dc;p=sfa.git

diff --git a/sfa/util/storage.py b/sfa/util/storage.py
index 5d915398..9b682064 100644
--- a/sfa/util/storage.py
+++ b/sfa/util/storage.py
@@ -1,6 +1,6 @@
 import os
+from sfa.util.xml import XML
 
-from sfa.util.rspec import RecordSpec
 
 class SimpleStorage(dict):
     """
@@ -9,62 +9,61 @@ class SimpleStorage(dict):
     """
     db_filename = None
     type = 'dict'
-    
-    def __init__(self, db_filename, db = {}):
 
+    def __init__(self, db_filename, db=None):
+        if db is None:
+            db = {}
         dict.__init__(self, db)
         self.db_filename = db_filename
-    
+
     def load(self):
         if os.path.exists(self.db_filename) and os.path.isfile(self.db_filename):
             db_file = open(self.db_filename, 'r')
             dict.__init__(self, eval(db_file.read()))
         elif os.path.exists(self.db_filename) and not os.path.isfile(self.db_filename):
-            raise IOError, '%s exists but is not a file. please remove it and try again' \
-                           % self.db_filename
+            raise IOError('%s exists but is not a file. please remove it and try again'
+                          % self.db_filename)
         else:
             self.write()
             self.load()
- 
+
     def write(self):
-        db_file = open(self.db_filename, 'w')  
+        db_file = open(self.db_filename, 'w')
         db_file.write(str(self))
         db_file.close()
-    
+
     def sync(self):
         self.write()
 
+
 class XmlStorage(SimpleStorage):
     """
     Handles storing and loading python dictionaries. The storage file created
     is a xml representation of the python dictionary.
-    """ 
+    """
     db_filename = None
     type = 'xml'
 
     def load(self):
         """
         Parse an xml file and store it as a dict
-        """ 
-        data = RecordSpec()
+        """
         if os.path.exists(self.db_filename) and os.path.isfile(self.db_filename):
-            data.parseFile(self.db_filename)
-            dict.__init__(self, data.toDict())
+            xml = XML(self.db_filename)
+            dict.__init__(self, xml.todict())
         elif os.path.exists(self.db_filename) and not os.path.isfile(self.db_filename):
-            raise IOError, '%s exists but is not a file. please remove it and try again' \
-                           % self.db_filename
+            raise IOError('%s exists but is not a file. please remove it and try again'
+                          % self.db_filename)
         else:
             self.write()
             self.load()
 
     def write(self):
-        data = RecordSpec()
-        data.parseDict(self)
+        xml = XML()
+        xml.parseDict(self)
         db_file = open(self.db_filename, 'w')
         db_file.write(data.toprettyxml())
         db_file.close()
 
     def sync(self):
         self.write()
-
-