planetstack/observer/steps/sync_users.py

   1 import os
   2 import base64
   3 from django.db.models import F, Q
   4 from planetstack.config import Config
   5 from observer.openstacksyncstep import OpenStackSyncStep
   6 from core.models.user import User
   7
   8 class SyncUsers(OpenStackSyncStep):
   9     provides=[User]
  10     requested_interval=0
  11
  12     def fetch_pending(self):
  13         return User.objects.filter(Q(enacted__lt=F('updated')) | Q(enacted=None))
  14
  15     def sync_record(self, user):
  16         name = user.email[:user.email.find('@')]
  17         user_fields = {'name': name,
  18                        'email': user.email,
  19                        'password': hashlib.md5(user.password).hexdigest()[:6],
  20                        'enabled': True}
  21         if not user.kuser_id:
  22             keystone_user = self.driver.create_user(**user_fields)
  23             user.kuser_id = keystone_user.id
  24         else:
  25             self.driver.update_user(user.kuser_id, user_fields)
  26
  27         if user.site:
  28             self.driver.add_user_role(user.kuser_id, user.site.tenant_id, 'user')
  29             if user.is_admin:
  30                 self.driver.add_user_role(user.kuser_id, user.site.tenant_id, 'admin')
  31             else:
  32                 # may have admin role so attempt to remove it
  33                 self.driver.delete_user_role(user.kuser_id, user.site.tenant_id, 'admin')
  34
  35         if user.public_key:
  36             driver = self.driver.client_driver(caller=user, tenant=user.site.login_base)
  37             key_fields =  {'name': user.keyname,
  38                            'public_key': user.public_key}
  39             driver.create_keypair(**key_fields)
  40
  41         user.save()