sfa/rspecs/version_manager.py

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