sfa/openstack/euca_shell.py

   1 try:
   2     import boto
   3     from boto.ec2.regioninfo import RegionInfo
   4     from boto.exception import EC2ResponseError
   5     has_boto=True
   6 except:
   7     has_boto=False
   8
   9 from sfa.util.sfalogging import logger
  10 from sfa.openstack.nova_shell import NovaShell
  11 from sfa.util.config import Config
  12
  13 class EucaShell:
  14     """
  15     A xmlrpc connection to the euca api.
  16     """
  17
  18     def __init__(self, config):
  19         self.config = config
  20         self.nova_shell = NovaShell(config)
  21         self.access_key = None
  22         self.secret_key = None
  23
  24     def init_context(self, project_name=None):
  25
  26         # use the context of the specified  project's project
  27         # manager.
  28         if project_name:
  29             project = self.nova_shell.auth_manager.get_project(project_name)
  30             self.access_key = "%s:%s" % (project.project_manager.name, project_name)
  31             self.secret_key = project.project_manager.secret
  32         else:
  33             # use admin user's context
  34             admin_user = self.nova_shell.auth_manager.get_user(self.config.SFA_NOVA_USER)
  35             #access_key = admin_user.access
  36             self.access_key = '%s' % admin_user.name
  37             self.secret_key = admin_user.secret
  38
  39     def get_euca_connection(self, project_name=None):
  40         if not has_boto:
  41             logger.info('Unable to access EC2 API - boto library not found.')
  42             return None
  43
  44         if not self.access_key or not self.secret_key:
  45             self.init_context(project_name)
  46
  47         url = self.config.SFA_NOVA_API_URL
  48         host = None
  49         port = None
  50         path = "/"
  51         use_ssl = False
  52         # Split the url into parts
  53         if url.find('https://') >= 0:
  54             use_ssl  = True
  55             url = url.replace('https://', '')
  56         elif url.find('http://') >= 0:
  57             use_ssl  = False
  58             url = url.replace('http://', '')
  59         parts = url.split(':')
  60         host = parts[0]
  61         if len(parts) > 1:
  62             parts = parts[1].split('/')
  63             port = int(parts[0])
  64             parts = parts[1:]
  65             path = '/'+'/'.join(parts)
  66         return boto.connect_ec2(aws_access_key_id=self.access_key,
  67                                 aws_secret_access_key=self.secret_key,
  68                                 is_secure=use_ssl,
  69                                 region=RegionInfo(None, 'eucalyptus', host),
  70                                 host=host,
  71                                 port=port,
  72                                 path=path)
  73
  74     def __getattr__(self, name):
  75         def func(*args, **kwds):
  76             conn = self.get_euca_connection()
  77