sfa/rspecs/version_manager.py

   1
   2
   3 import os
   4 from sfa.util.faults import InvalidRSpec, UnsupportedRSpecVersion
   5 from sfa.rspecs.version import RSpecVersion
   6 from sfa.util.sfalogging import logger
   7
   8
   9 class VersionManager:
  10
  11     def __init__(self):
  12         self.versions = []
  13         self.load_versions()
  14
  15     def __repr__(self):
  16         return "<VersionManager with {} flavours: [{}]>"\
  17             .format(len(self.versions),
  18                     ", ".join([str(x) for x in self.versions]))
  19
  20     def load_versions(self):
  21         path = os.path.dirname(os.path.abspath(__file__))
  22         versions_path = path + os.sep + 'versions'
  23         versions_module_path = 'sfa.rspecs.versions'
  24         valid_module = lambda x: os.path.isfile(os.sep.join([versions_path, x])) \
  25             and x.endswith('.py') and x != '__init__.py'
  26         files = [f for f in os.listdir(versions_path) if valid_module(f)]
  27         for filename in files:
  28             basename = filename.split('.')[0]
  29             module_path = versions_module_path + '.' + basename
  30             module = __import__(module_path, fromlist=module_path)
  31             for attr_name in dir(module):
  32                 attr = getattr(module, attr_name)
  33                 if hasattr(attr, 'version') and hasattr(attr, 'enabled') and attr.enabled == True:
  34                     self.versions.append(attr())
  35
  36     def _get_version(self, type, version_num=None, content_type=None):
  37         retval = None
  38         for version in self.versions:
  39             if type is None or type.lower() == version.type.lower():
  40                 if version_num is None or str(float(version_num)) == str(float(version.version)):
  41                     if content_type is None or content_type.lower() == version.content_type.lower() \
  42                             or version.content_type == '*':
  43                         retval = version
  44                         # sounds like we should be glad with the first match,
  45                         # not the last one
  46                         break
  47         if not retval:
  48             raise UnsupportedRSpecVersion(
  49                 "[%s %s %s] is not suported here" % (type, version_num, content_type))
  50         return retval
  51
  52     def get_version(self, version=None):
  53         retval = None
  54         if isinstance(version, dict):
  55             retval = self._get_version(version.get('type'), version.get(
  56                 'version'), version.get('content_type'))
  57         elif isinstance(version, str):
  58             version_parts = version.split(' ')
  59             num_parts = len(version_parts)
  60             type = version_parts[0]
  61             version_num = None
  62             content_type = None
  63             if num_parts > 1:
  64                 version_num = version_parts[1]
  65             if num_parts > 2:
  66                 content_type = version_parts[2]
  67             retval = self._get_version(type, version_num, content_type)
  68         elif isinstance(version, RSpecVersion):
  69             retval = version
  70         elif not version:
  71             retval = self.versions[0]
  72         else:
  73             raise UnsupportedRSpecVersion(
  74                 "No such version: %s " % str(version))
  75
  76         return retval
  77
  78     def get_version_by_schema(self, schema):
  79         retval = None
  80         for version in self.versions:
  81             if schema == version.schema:
  82                 retval = version
  83         if not retval:
  84             raise InvalidRSpec("Unkwnown RSpec schema: %s" % schema)
  85         return retval
  86
  87     def show_by_string(self, string):
  88         try:
  89             print(self.get_version(string))
  90         except Exception as e:
  91             print(e)
  92
  93     def show_by_schema(self, string):
  94         try:
  95             print(self.get_version_by_schema(string))
  96         except Exception as e:
  97             print(e)
  98
  99 if __name__ == '__main__':
 100     manager = VersionManager()
 101     print(manager)
 102     manager.show_by_string('sfa 1')
 103     manager.show_by_string('protogeni 2')
 104     manager.show_by_string('protogeni 2 advertisement')
 105     manager.show_by_schema('http://www.protogeni.net/resources/rspec/2/ad.xsd')
 106     manager.show_by_schema('http://sorch.netmode.ntua.gr/ws/RSpec/ad.xsd')