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
7 def _get_users(filter):
8 if isinstance(filter, StringTypes) and filter.isdigit():
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)
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
25 sites = _get_sites(fields.get('site'))
26 if sites: fields['site'] = sites[0]
28 user.os_manager = OpenStackManager(auth=auth, caller = user)
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
37 users = User.objects.filter(id=id)
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():
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)
64 user.os_manager = OpenStackManager(auth=auth, caller = user)
68 def get_users(auth, filter={}):
69 user = authenticate(username=auth.get('username'),
70 password=auth.get('password'))
71 users = _get_users(filter)