planetstack/openstack_observer/steps/sync_controller_users.py

   1 import os
   2 import base64
   3 from collections import defaultdict
   4 from django.db.models import F, Q
   5 from planetstack.config import Config
   6 from observer.openstacksyncstep import OpenStackSyncStep
   7 from core.models.site import Controller, SiteDeployments, SiteDeployments
   8 from core.models.user import User
   9 from core.models.controllerusers import ControllerUsers
  10 from util.logger import Logger, logging
  11
  12 from observer.ansible import *
  13
  14 logger = Logger(level=logging.INFO)
  15
  16 class SyncControllerUsers(OpenStackSyncStep):
  17     provides=[ControllerUsers, User]
  18     requested_interval=0
  19
  20     def fetch_pending(self, deleted):
  21
  22         if (deleted):
  23             return ControllerUsers.deleted_objects.all()
  24         else:
  25             return ControllerUsers.objects.filter(Q(enacted__lt=F('updated')) | Q(enacted=None))
  26
  27     def sync_record(self, controller_user):
  28         logger.info("sync'ing user %s at controller %s" % (controller_user.user, controller_user.controller))
  29
  30         if not controller_user.controller.admin_user:
  31             logger.info("controller %r has no admin_user, skipping" % controller_user.controller)
  32             return
  33
  34         template = os_template_env.get_template('sync_controller_users.yaml')
  35
  36         # All users will have at least the 'user' role at their home site/tenant.
  37         # We must also check if the user should have the admin role
  38         roles = ['user']
  39         if controller_user.user.is_admin:
  40             roles.append('admin')
  41
  42         # setup user home site roles at controller
  43         if not controller_user.user.site:
  44             raise Exception('Siteless user %s'%controller_user.user.email)
  45         else:
  46             # look up tenant id for the user's site at the controller
  47             #ctrl_site_deployments = SiteDeployments.objects.filter(
  48             #  site_deployment__site=controller_user.user.site,
  49             #  controller=controller_user.controller)
  50
  51             #if ctrl_site_deployments:
  52             #    # need the correct tenant id for site at the controller
  53             #    tenant_id = ctrl_site_deployments[0].tenant_id
  54             #    tenant_name = ctrl_site_deployments[0].site_deployment.site.login_base
  55             user_fields = {
  56                        'endpoint':controller_user.controller.auth_url,
  57                        'name': controller_user.user.email,
  58                        'email': controller_user.user.email,
  59                        'password': controller_user.user.remote_password,
  60                        'admin_user': controller_user.controller.admin_user,
  61                        'admin_password': controller_user.controller.admin_password,
  62                        'ansible_tag':'%s@%s'%(controller_user.user.email.replace('@','-at-'),controller_user.controller.name),
  63                        'admin_tenant': controller_user.controller.admin_tenant,
  64                        'roles':roles,
  65                        'tenant':controller_user.user.site.login_base}
  66
  67             rendered = template.render(user_fields)
  68             res = run_template('sync_controller_users.yaml', user_fields,path='controller_users')
  69
  70             # results is an array in which each element corresponds to an
  71             # "ok" string received per operation. If we get as many oks as
  72             # the number of operations we issued, that means a grand success.
  73             # Otherwise, the number of oks tell us which operation failed.
  74             expected_length = len(roles) + 1
  75             if (len(res)==expected_length):
  76                 controller_user.kuser_id = res[0]['id']
  77                 controller_user.save()
  78             elif (len(res)):
  79                 raise Exception('Could not assign roles for user %s'%user_fields['name'])
  80             else:
  81                 raise Exception('Could not create or update user %s'%user_fields['name'])
  82
  83     def delete_record(self, controller_user):
  84         if controller_user.kuser_id:
  85             driver = self.driver.admin_driver(controller=controller_user.controller)
  86             driver.delete_user(controller_user.kuser_id)