plstackapi/planetstack/api/users.py

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