Fix bugs. Reefactor.
[plstackapi.git] / planetstack / openstack_observer / steps / sync_site_privileges.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 import User, ControllerUsers, SitePrivilege, SiteDeployments   
7
8 class SyncSitePrivileges(OpenStackSyncStep):
9     requested_interval=0
10     provides=[SitePrivilege]
11
12     def fetch_pending(self, deleted):
13         # Deleting site privileges is not supported yet
14         if (deleted):
15             return []
16
17         return SitePrivilege.objects.filter(Q(enacted__lt=F('updated')) | Q(enacted=None))
18
19     def sync_record(self, site_priv):
20         # sync site privileges at all site controllers
21         ctrl_site_deployments = SiteDeployments.objects.filter(site_deployment__site=site_priv.site)
22         for ctrl_site_deployment in ctrl_site_deployments:
23             controller_users = ControllerUsers.objects.filter(controller=ctrl_site_deployment.controller,
24                                                               user=site_priv.user)
25             if controller_users:
26                 kuser_id  = controller_users[0].kuser_id
27                 driver = self.driver.admin_driver(controller=ctrl_site_deployment.controller)
28                 driver.add_user_role(kuser_id,
29                                      ctrl_site_deployment.controller.tenant_id,
30                                      site_priv.role.role)