planetstack/openstack_observer/steps/sync_slice_memberships.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.slice import *
   7 from core.models.controllerusers import ControllerUser
   8 from util.logger import Logger, logging
   9
  10 logger = Logger(level=logging.INFO)
  11
  12 class SyncSliceMemberships(OpenStackSyncStep):
  13     requested_interval=0
  14     provides=[SlicePrivilege]
  15
  16     def fetch_pending(self, deleted):
  17         # Deleting site memberships is not supported yet
  18         if (deleted):
  19             return []
  20         return SlicePrivilege.objects.filter(Q(enacted__lt=F('updated')) | Q(enacted=None))
  21
  22     def sync_record(self, slice_memb):
  23         # sync slice memberships at all slice controllers
  24         logger.info("syncing slice privilege: %s %s" % (slice_memb.slice.name, slice_memb.user.email))
  25         slice_controllers = ControllerSlice.objects.filter(slice=slice_memb.slice)
  26         for slice_controller in slice_controllers:
  27             if not slice_controller.tenant_id:
  28                 continue
  29             controller_users = ControllerUser.objects.filter(controller=slice_controller.controller,
  30                                                               user=slice_memb.user)
  31             if controller_users:
  32                 kuser_id  = controller_users[0].kuser_id
  33                 driver = self.driver.admin_driver(controller=slice_controller.controller.name)
  34                 driver.add_user_role(kuser_id,
  35                                      slice_controller.tenant_id,
  36                                      slice_memb.role.role)