sfa/rspecs/version_manager.py

   1 import os
   2 from sfa.util.faults import InvalidRSpec
   3 from sfa.rspecs.rspec_version import BaseVersion
   4 from sfa.util.sfalogging import logger
   5
   6 class VersionManager:
   7     default_type = 'SFA'
   8     default_version_num = '1'
   9
  10     def __init__(self):
  11         self.versions = []
  12         self.load_versions()
  13
  14     def load_versions(self):
  15         path = os.path.dirname(os.path.abspath( __file__ ))
  16         versions_path = path + os.sep + 'versions'
  17         versions_module_path = 'sfa.rspecs.versions'
  18         valid_module = lambda x: os.path.isfile(os.sep.join([versions_path, x])) \
  19                         and not x.endswith('.pyc') and x not in ['__init__.py']
  20         files = [f for f in os.listdir(versions_path) if valid_module(f)]
  21         for filename in files:
  22             basename = filename.split('.')[0]
  23             module_path = versions_module_path +'.'+basename
  24             module = __import__(module_path, fromlist=module_path)
  25             for attr_name in dir(module):
  26                 attr = getattr(module, attr_name)
  27                 if hasattr(attr, 'version') and hasattr(attr, 'enabled') and attr.enabled == True:
  28                     self.versions.append(attr())
  29
  30     def _get_version(self, type, version_num=None, content_type=None):
  31         retval = None
  32         for version in self.versions:
  33             if type is None or type.lower() == version.type.lower():
  34                 if version_num is None or version_num == version.version:
  35                     if content_type is None or content_type.lower() == version.content_type.lower():
  36                         retval = version
  37         if not retval:
  38             raise InvalidRSpec("No such version format: %s version: %s type:%s "% (type, version_num, content_type))
  39         return retval
  40
  41     def get_version(self, version=None):
  42         retval = None
  43         if isinstance(version, dict):
  44             retval =  self._get_version(version.get('type'), version.get('version'), version.get('content_type'))
  45         elif isinstance(version, basestring):
  46             version_parts = version.split(' ')
  47             num_parts = len(version_parts)
  48             type = version_parts[0]
  49             version_num = None
  50             content_type = None
  51             if num_parts > 1:
  52                 version_num = version_parts[1]
  53             if num_parts > 2:
  54                 content_type = version_parts[2]
  55             retval = self._get_version(type, version_num, content_type)
  56         elif isinstance(version, BaseVersion):
  57             retval = version
  58         else:
  59             retval = self._get_version(self.default_type, self.default_version_num)
  60
  61         return retval
  62
  63     def get_version_by_schema(self, schema):
  64         retval = None
  65         for version in self.versions:
  66             if schema == version.schema:
  67                 retval = version
  68         if not retval:
  69             raise InvalidRSpec("Unkwnown RSpec schema: %s" % schema)
  70         return retval
  71
  72 if __name__ == '__main__':
  73     v = VersionManager()
  74     print v.versions
  75     print v.get_version('sfa 1')
  76     print v.get_version('protogeni 2')
  77     print v.get_version('protogeni 2 advertisement')
  78     print v.get_version_by_schema('http://www.protogeni.net/resources/rspec/2/ad.xsd')
  79