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):
  19         self.nova_shell = NovaShell(config)
  20         self.access_key = None
  21         self.secret_key = None
  22
  23     def _init_ctx(self, username=None, project=None):
  24         if username and project:
  25             self.access_key = "%s:%s" % (username, project)
  26             self.secret_key = self.nova_shell.auth_manager.get_user(username).secret
  27         else:
  28             admin_user = self.nova_shell.auth_manager.get_user(self.config.SFA_NOVA_USER)
  29             #access_key = admin_user.access
  30             self.access_key = '%s' % admin_user.name
  31             self.secret_key = admin_user.secret
  32
  33
  34
  35     def get_euca_connection(self, username=None, project=None):
  36         if not has_boto:
  37             logger.info('Unable to access EC2 API - boto library not found.')
  38             return None
  39
  40         if username and project:
  41             self._init_ctx(username, project)
  42         elif not self.access_key or not self.secret_key
  43             self._init_ctx()
  44
  45         url = self.config.SFA_NOVA_API_URL
  46         host = None
  47         port = None
  48         path = "/"
  49         use_ssl = False
  50         # Split the url into parts
  51         if url.find('https://') >= 0:
  52             use_ssl  = True
  53             url = url.replace('https://', '')
  54         elif url.find('http://') >= 0:
  55             use_ssl  = False
  56             url = url.replace('http://', '')
  57         parts = url.split(':')
  58         host = parts[0]
  59         if len(parts) > 1:
  60             parts = parts[1].split('/')
  61             port = int(parts[0])
  62             parts = parts[1:]
  63             path = '/'+'/'.join(parts)
  64         return boto.connect_ec2(aws_access_key_id=self.access_key,
  65                                 aws_secret_access_key=self.secret_key,
  66                                 is_secure=use_ssl,
  67                                 region=RegionInfo(None, 'eucalyptus', host),
  68                                 host=host,
  69                                 port=port,
  70                                 path=path)
  71
  72     def __getattr__(self, name):
  73         def func(*args, **kwds):
  74             conn = self.get_euca_connection()
  75