add networkDeployments and userDeployments

[plstackapi.git] / planetstack / model_policy.py

diff --git a/planetstack/model_policy.py b/planetstack/model_policy.py
index 32d8f8e..9224c50 100644 (file)
--- a/planetstack/model_policy.py
+++ b/planetstack/model_policy.py
@@ -4,14 +4,24 @@ import pdb
 from core.models import *
 import model_policies
 
+modelPolicyEnabled = True
+
+def EnableModelPolicy(x):
+    global modelPolicyEnabled
+    modelPolicyEnabled = x
+
 @receiver(post_save)
 def post_save_handler(sender, instance, **kwargs):
        sender_name = sender.__name__
        policy_name = 'model_policy_%s'%sender_name
+
+        if not modelPolicyEnabled:
+            return
        
        if (not kwargs['update_fields']):
                try:
-                       policy_handler = getattr(model_policies, policy_name)
-                       policy_handler.handle(instance)
+                       policy_handler = getattr(model_policies, policy_name, None)
+                       if policy_handler is not None:
+                               policy_handler.handle(instance)
                except:
                        pass