-from plstackapi.openstack.client import OpenStackClient
-from plstackapi.openstack.driver import OpenStackDriver
-from plstackapi.core.api.auth import auth_check
-from plstackapi.core.models import User, Site
-
-def lookup_site(fields):
- site = None
- if 'site' in fields:
- if isinstance(fields['site'], int):
- sites = Site.objects.filter(id=fields['site'])
- else:
- sites = Site.objects.filter(login_base=fields['site'])
- if sites:
- site = sites[0]
- if not site:
- raise Exception, "No such site", fields['site']
- return site
+from types import StringTypes
+from django.contrib.auth import authenticate
+from plstackapi.openstack.manager import OpenStackManager
+from plstackapi.core.models import PLUser, Site
+from plstackapi.core.api.sites import _get_sites
+
+def _get_users(filter):
+ if isinstance(filter, StringTypes) and filter.isdigit():
+ filter = int(filter)
+ if isinstance(filter, int):
+ users = PLUser.objects.filter(id=filter)
+ elif isinstance(filter, StringTypes):
+ users = PLUser.objects.filter(email=filter)
+ elif isinstance(filter, dict):
+ users = PLUser.objects.filter(**filter)
+ else:
+ users = []
+ return users
def add_user(auth, fields):
- driver = OpenStackDriver(client = auth_check(auth))
- site = lookup_site(fields)
- if site: fields['site'] = site
- user = User(**fields)
- nova_fields = {'name': user.email[:user.email.find('@')],
- 'email': user.email,
- 'password': fields.get('password'),
- 'enabled': user.enabled}
- nova_user = driver.create_user(**nova_fields)
- #driver.add_user_role(user.id, user.site.tenant_id, 'user')
- user.user_id=nova_user.id
+ user = authenticate(username=auth.get('username'),
+ password=auth.get('password'))
+ auth['tenant'] = user.site.login_base
+
+ sites = _get_sites(fields.get('site'))
+ if sites: fields['site'] = sites[0]
+ user = PLUser(**fields)
+ user.os_manager = OpenStackManager(auth=auth, caller = user)
user.save()
return user
def update_user(auth, id, **fields):
- driver = OpenStackDriver(client = auth_check(auth))
- users = User.objects.filter(id=id)
+ user = authenticate(username=auth.get('username'),
+ password=auth.get('password'))
+ auth['tenant'] = user.site.login_base
+
+ users = PLUser.objects.filter(id=id)
if not users:
return
nova_fields['password'] = fields['password']
if 'enabled' in fields:
nova_fields['enabled'] = fields['enabled']
- driver.update_user(user.user_id, **nova_fields)
- site = lookup_site(fields)
- if site: fields['site'] = site
- user.update(**fields)
+
+
+ sites = _get_sites(fields.get('site'))
+ if sites: fields['site'] = sites[0]
+ user.os_manager = OpenStackManager(auth=auth, caller = user)
+ for (k,v) in fields.items():
+ setattr(user, k, v)
+ user.save()
return user
def delete_user(auth, filter={}):
- driver = OpenStackDriver(client = auth_check(auth))
- users = User.objects.filter(**filter)
+ user = authenticate(username=auth.get('username'),
+ password=auth.get('password'))
+ auth['tenant'] = user.site.login_base
+ users = _get_users(filter)
for user in users:
- driver.delete_user(id=user.user_id)
+ user.os_manager = OpenStackManager(auth=auth, caller = user)
user.delete()
return 1
def get_users(auth, filter={}):
- client = auth_check(auth)
- users = User.objects.filter(**filter)
+ user = authenticate(username=auth.get('username'),
+ password=auth.get('password'))
+ users = _get_users(filter)
return users