planetstack/model_policies/model_policy_User.py

   1 from core.models import *
   2
   3 def handle(user):
   4         deployments = Deployment.objects.all()
   5         site_deployments = SiteDeployments.objects.all()
   6         site_deploy_lookup = defaultdict(list)
   7         for site_deployment in site_deployments:
   8                 site_deploy_lookup[site_deployment.site].append(site_deployment.deployment)
   9
  10         user_deploy_lookup = defaultdict(list)
  11         for user_deployment in UserDeployments.objects.all():
  12                 user_deploy_lookup[user_deployment.user].append(user_deployment.deployment)
  13
  14         all_deployments = Deployment.objects.filter()
  15         if user.is_admin:
  16                 # admins should have an account at all deployments
  17                 expected_deployments = deployments
  18         else:
  19                 # normal users should have an account at their site's deployments
  20                 #expected_deployments = site_deploy_lookup[user.site]
  21                 # users are added to all deployments for now
  22                 expected_deployments = deployments
  23
  24         for expected_deployment in expected_deployments:
  25                 if not user in user_deploy_lookup or \
  26                   expected_deployment not in user_deploy_lookup[user]:
  27                         # add new record
  28                         ud = UserDeployments(user=user, deployment=expected_deployment)
  29                         ud.save()
  30