plstackapi/core/api/users.py

   1 from types import StringTypes
   2 from django.contrib.auth import authenticate
   3 from plstackapi.openstack.manager import OpenStackManager
   4 from plstackapi.core.models import PLUser, Site
   5 from plstackapi.core.api.sites import _get_sites
   6
   7 def _get_users(filter):
   8     if isinstance(filter, StringTypes) and filter.isdigit():
   9         filter = int(filter)
  10     if isinstance(filter, int):
  11         users = PLUser.objects.filter(id=filter)
  12     elif isinstance(filter, StringTypes):
  13         users = PLUser.objects.filter(email=filter)
  14     elif isinstance(filter, dict):
  15         users = PLUser.objects.filter(**filter)
  16     else:
  17         users = []
  18     return users
  19
  20 def add_user(auth, fields):
  21     user = authenticate(username=auth.get('username'),
  22                         password=auth.get('password'))
  23     auth['tenant'] = user.site.login_base
  24
  25     sites = _get_sites(fields.get('site'))
  26     if sites: fields['site'] = sites[0]
  27     user = PLUser(**fields)
  28     user.os_manager = OpenStackManager(auth=auth, caller = user)
  29     user.save()
  30     return user
  31
  32 def update_user(auth, id, **fields):
  33     user = authenticate(username=auth.get('username'),
  34                         password=auth.get('password'))
  35     auth['tenant'] = user.site.login_base
  36
  37     users = PLUser.objects.filter(id=id)
  38     if not users:
  39         return
  40
  41     user = users[0]
  42     nova_fields = {}
  43     if 'email' in fields:
  44         nova_fields['name'] = fields['email'][:self.email.find('@')]
  45         nova_fields['email'] = fields['email']
  46     if 'password' in fields:
  47         nova_fields['password'] = fields['password']
  48     if 'enabled' in fields:
  49         nova_fields['enabled'] = fields['enabled']
  50
  51
  52     sites = _get_sites(fields.get('site'))
  53     if sites: fields['site'] = sites[0]
  54     user.os_manager = OpenStackManager(auth=auth, caller = user)
  55     for (k,v) in fields.items():
  56         setattr(user, k, v)
  57     user.save()
  58     return user
  59
  60 def delete_user(auth, filter={}):
  61     user = authenticate(username=auth.get('username'),
  62                         password=auth.get('password'))
  63     auth['tenant'] = user.site.login_base
  64     users = _get_users(filter)
  65     for user in users:
  66         user.os_manager = OpenStackManager(auth=auth, caller = user)
  67         user.delete()
  68     return 1
  69
  70 def get_users(auth, filter={}):
  71     user = authenticate(username=auth.get('username'),
  72                         password=auth.get('password'))
  73     users = _get_users(filter)
  74     return users
  75
  76
  77