This change renamed a number of misnamed related fields (e.g. site_privileges->sitepr...

[plstackapi.git] / planetstack / core / models / userdeployments.py

diff --git a/planetstack/core/models/userdeployments.py b/planetstack/core/models/userdeployments.py
index 3422156..d0337ba 100644 (file)
--- a/planetstack/core/models/userdeployments.py
+++ b/planetstack/core/models/userdeployments.py
@@ -4,10 +4,14 @@ from collections import defaultdict
 from django.db import models
 from django.db.models import F, Q
 from core.models import PlCoreBase,Site,User,Deployment
+from core.models import Deployment,DeploymentLinkManager,DeploymentLinkDeletionManager
 
-class UserDeployments(PlCoreBase):
-    user = models.ForeignKey(User)
-    deployment = models.ForeignKey(Deployment)
+class UserDeployment(PlCoreBase):
+    objects = DeploymentLinkManager()
+    deleted_objects = DeploymentLinkDeletionManager()
+
+    user = models.ForeignKey(User,related_name='userdeployments')
+    deployment = models.ForeignKey(Deployment,related_name='userdeployments')
     kuser_id = models.CharField(null=True, blank=True, max_length=200, help_text="Keystone user id")
 
     def __unicode__(self):  return u'%s %s' % (self.user, self.deployment.name)
@@ -15,7 +19,7 @@ class UserDeployments(PlCoreBase):
     @staticmethod
     def select_by_user(user):
         if user.is_admin:
-            qs = UserDeployments.objects.all()
+            qs = UserDeployment.objects.all()
         else:
             users = Users.select_by_user(user)
             qs = Usereployments.objects.filter(user__in=slices)