From: Sapan Bhatia <gwsapan@gmail.com>
Date: Fri, 19 Sep 2014 20:47:07 +0000 (-0400)
Subject: Deployment manager for filtering deployments based on backend
X-Git-Url: http://git.onelab.eu/?a=commitdiff_plain;h=ac34c81fd2aa38e0c53e8dcd9db62c97f4ebc1bc;p=plstackapi.git

Deployment manager for filtering deployments based on backend
---

diff --git a/planetstack/core/models/site.py b/planetstack/core/models/site.py
index bc4c5ca..ccd5d39 100644
--- a/planetstack/core/models/site.py
+++ b/planetstack/core/models/site.py
@@ -1,11 +1,23 @@
 import os
 from django.db import models
-from core.models import PlCoreBase
+from core.models import PlCoreBase,PlCoreBaseManager,PlCoreBaseDeletionManager
 from core.models import Tag
 from django.contrib.contenttypes import generic
 from geoposition.fields import GeopositionField
 from core.acl import AccessControlList
 
+class DeploymentManager(PlCoreBaseManager):
+    def get_queryset(self):
+        parent=super(DeploymentManager, self)
+        if hasattr(parent, "get_queryset"):
+            return parent.get_queryset().filter(Q(backend_type=config.observer_backend_type)|Q(backend_type=None))
+        else:
+            return parent.get_queryset().filter(Q(backend_type=config.observer_backend_type)|Q(backend_type=None))
+
+    # deprecated in django 1.7 in favor of get_queryset().
+    def get_query_set(self):
+        return self.get_queryset()
+
 class Site(PlCoreBase):
     """
         A logical grouping of Nodes that are co-located at the same geographic location, which also typically corresponds to the Nodes' location in the physical network.