update how RSpecVersion is used

[sfa.git] / sfa / rspecs / rspec_version.py

diff --git a/sfa/rspecs/rspec_version.py b/sfa/rspecs/rspec_version.py
old mode 100644 (file)
new mode 100755 (executable)
index 4deb924..b96a765
--- a/sfa/rspecs/rspec_version.py
+++ b/sfa/rspecs/rspec_version.py
@@ -1,32 +1,42 @@
 #!/usr/bin/python
-from sfa.util.sfalogging import sfa_logger
+from sfa.util.sfalogging import _SfaLogger
 
-class RSpecVersion:
+class RSpecVersion(dict):
 
-    format = None
-    version = None
-    schema = None
-    namespace = None
-    extensions = []
+    fields = {'type': None,
+              'version': None,
+              'schema': None,
+              'namespace': None,
+              'extensions': []
+        }
+    def __init__(self, version={}):
+        
+        self.logger = _SfaLogger('/var/log/sfa.log')
+        dict.__init__(self, self.fields)
 
-    def __init__(self, version_raw):
-        self.logger = sfa_logger() 
-        self.parse_version(version_raw)
+        if not version:
+            from sfa.rspecs.sfa_rspec import sfa_rspec_version
+            self.update(sfa_rspec_version)          
+        elif isinstance(version, dict):
+            self.update(version)
+        elif isinstance(version, basestring):
+            version_parts = version.split(' ')
+            num_parts = len(version_parts)
+            self['type'] = version_parts[0]
+            if num_parts > 1:
+                self['version'] = version_parts[1]
+        else:
+            logger.info("Unable to parse rspec version, using default")
 
-    def parse_version(self, version_raw):
-        # version_raw is currently a string but will 
-        # eventually be a struct.
-        try:
-            format_split = version_raw.split(' ')
-            format, version = format_split[0].lower(), format_split[1]
-        except:
-            self.logger.info("RSpecVersion: invalid rspec version: %s , using default" \
-                        % version_raw)
-            # invalid format. Just continue
-            format, version = 'sfa', '1'
+    def get_version_name(self):
+        return "%s %s" % (str(self['type']), str(self['version']))
 
-        self.format = format
-        self.version = version 
-    
+if __name__ == '__main__':
+
+    from sfa.rspecs.pl_rspec_version import ad_rspec_versions
+    for version in [RSpecVersion(), 
+                    RSpecVersion("SFA"), 
+                    RSpecVersion("SFA 1"),
+                    RSpecVersion(ad_rspec_versions[0])]: 
+        print version.get_version_name() + ": " + str(version)
 
-