plstackapi/core/api/users.py

   1 from plstackapi.openstack.client import OpenStackClient
   2 from plstackapi.openstack.driver import OpenStackDriver
   3 from plstackapi.core.api.auth import auth_check
   4 from plstackapi.core.models import User, Site
   5
   6 def lookup_site(fields):
   7     site = None
   8     if 'site' in fields:
   9         if isinstance(fields['site'], int):
  10             sites = Site.objects.filter(id=fields['site'])
  11         else:
  12             sites = Site.objects.filter(login_base=fields['site'])
  13         if sites:
  14             site = sites[0]
  15     if not site:
  16         raise Exception, "No such site: %s" % fields['site']
  17     return site
  18
  19 def add_user(auth, fields):
  20     driver = OpenStackDriver(client = auth_check(auth))
  21     site = lookup_site(fields)
  22     if site: fields['site'] = site
  23     user = User(**fields)
  24     nova_fields = {'name': user.email[:user.email.find('@')],
  25                    'email': user.email,
  26                    'password': fields.get('password'),
  27                    'enabled': user.enabled}
  28     nova_user = driver.create_user(**nova_fields)
  29     #driver.add_user_role(user.id, user.site.tenant_id, 'user')
  30     user.user_id=nova_user.id
  31     user.save()
  32     return user
  33
  34 def update_user(auth, id, **fields):
  35     driver = OpenStackDriver(client = auth_check(auth))
  36     users = User.objects.filter(id=id)
  37     if not users:
  38         return
  39
  40     user = users[0]
  41     nova_fields = {}
  42     if 'email' in fields:
  43         nova_fields['name'] = fields['email'][:self.email.find('@')]
  44         nova_fields['email'] = fields['email']
  45     if 'password' in fields:
  46         nova_fields['password'] = fields['password']
  47     if 'enabled' in fields:
  48         nova_fields['enabled'] = fields['enabled']
  49     driver.update_user(user.user_id, **nova_fields)
  50     site = lookup_site(fields)
  51     if site: fields['site'] = site
  52     user.update(**fields)
  53     return user
  54
  55 def delete_user(auth, filter={}):
  56     driver = OpenStackDriver(client = auth_check(auth))
  57     users = User.objects.filter(**filter)
  58     for user in users:
  59         driver.delete_user(id=user.user_id)
  60         user.delete()
  61     return 1
  62
  63 def get_users(auth, filter={}):
  64     client = auth_check(auth)
  65     users = User.objects.filter(**filter)
  66     return users
  67
  68
  69