planetstack/core/api/users.py

   1 from types import StringTypes
   2 from django.contrib.auth import authenticate
   3 from openstack.manager import OpenStackManager
   4 from core.models import User, Site
   5 from 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 = User.objects.filter(id=filter)
  12     elif isinstance(filter, StringTypes):
  13         users = User.objects.filter(email=filter)
  14     elif isinstance(filter, dict):
  15         users = User.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 = User(**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 = User.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     sites = _get_sites(fields.get('site'))
  51     if sites: fields['site'] = sites[0]
  52     user.os_manager = OpenStackManager(auth=auth, caller = user)
  53     for (k,v) in fields.items():
  54         setattr(user, k, v)
  55     user.save()
  56     return user
  57
  58 def delete_user(auth, filter={}):
  59     user = authenticate(username=auth.get('username'),
  60                         password=auth.get('password'))
  61     auth['tenant'] = user.site.login_base
  62     users = _get_users(filter)
  63     for user in users:
  64         user.os_manager = OpenStackManager(auth=auth, caller = user)
  65         user.delete()
  66     return 1
  67
  68 def get_users(auth, filter={}):
  69     user = authenticate(username=auth.get('username'),
  70                         password=auth.get('password'))
  71     users = _get_users(filter)
  72     return users
  73
  74
  75