Renamed SliceDeployment back to SliceDeployments

[plstackapi.git] / planetstack / model_policies / model_policy_User.py

diff --git a/planetstack/model_policies/model_policy_User.py b/planetstack/model_policies/model_policy_User.py
index 1b1895e..7af6ff3 100644 (file)
--- a/planetstack/model_policies/model_policy_User.py
+++ b/planetstack/model_policies/model_policy_User.py
@@ -1,6 +1,8 @@
 from core.models import *
 
 def handle(user):
+       from core.models import Deployment,SiteDeployments,UserDeployment
+       from collections import defaultdict
        deployments = Deployment.objects.all()
        site_deployments = SiteDeployments.objects.all()
        site_deploy_lookup = defaultdict(list)
@@ -8,7 +10,7 @@ def handle(user):
                site_deploy_lookup[site_deployment.site].append(site_deployment.deployment)
 
        user_deploy_lookup = defaultdict(list)
-       for user_deployment in UserDeployments.objects.all():
+       for user_deployment in UserDeployment.objects.all():
                user_deploy_lookup[user_deployment.user].append(user_deployment.deployment)
    
        all_deployments = Deployment.objects.filter() 
@@ -25,6 +27,6 @@ def handle(user):
                if not user in user_deploy_lookup or \
                  expected_deployment not in user_deploy_lookup[user]: 
                        # add new record
-                       ud = UserDeployments(user=user, deployment=expected_deployment)
+                       ud = UserDeployment(user=user, deployment=expected_deployment)
                        ud.save()