X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=planetstack%2Fmodel_policy.py;h=8ddb82ecbb408f65943e541717e82c1b29aadc96;hb=e42cedcb2caf20b1118db570a9650348958d7590;hp=e12b15cb8bc61a1c876bce2c19ee409763c347da;hpb=f6288f8c4299a9bad92c012c11e5d401b74ad20c;p=plstackapi.git

diff --git a/planetstack/model_policy.py b/planetstack/model_policy.py
index e12b15c..8ddb82e 100644
--- a/planetstack/model_policy.py
+++ b/planetstack/model_policy.py
@@ -1,10 +1,14 @@
 from django.db.models.signals import post_save
 from django.dispatch import receiver
 import pdb
-from core.models import *
 from dependency_walker import *
 import model_policies
 from util.logger import logger
+from datetime import datetime
+import time
+from core.models import *
+from django.db.transaction import atomic
+from django.db.models import F, Q
 
 modelPolicyEnabled = True
 
@@ -17,32 +21,52 @@ def update_dep(d, o):
 		if (d.updated < o.updated):
 			d.save(update_fields=['updated'])
 	except AttributeError,e:
-		pdb.set_trace()
 		raise e
 	
 def delete_if_inactive(d, o):
 	#print "Deleting %s (%s)"%(d,d.__class__.__name__)
-	d.delete()	
+	# d.delete()	
 	return
 
-@receiver(post_save)
-def post_save_handler(sender, instance, **kwargs):
-        if not modelPolicyEnabled:
-            return
+@atomic
+def execute_model_policy(instance, deleted):
+	# Automatic dirtying
+	walk_inv_deps(update_dep, instance)
 
-	sender_name = sender.__name__
+	sender_name = instance.__class__.__name__
 	policy_name = 'model_policy_%s'%sender_name
-	
-	if (not kwargs['update_fields']):
-		# Automatic dirtying
-		walk_inv_deps(update_dep, instance)
+	noargs = False
 
+	if deleted:
+		walk_inv_deps(delete_if_inactive, instance)
+	else:
 		try:
 			policy_handler = getattr(model_policies, policy_name, None)
+			logger.error("POLICY HANDLER: %s %s" % (policy_name, policy_handler))                       
 			if policy_handler is not None:
 				policy_handler.handle(instance)
 		except:
 			logger.log_exc("Model Policy Error:") 
-	elif 'deleted' in kwargs['update_fields']:
-		walk_inv_deps(delete_if_inactive, instance)
-	
+			print "Policy Exceution Error"
+
+	instance.policed=datetime.now()
+        instance.save(update_fields=['policed'])
+
+def run_policy():
+        from core.models import Slice,Controller,Network,User,SlicePrivilege,Site,SitePrivilege,Image,ControllerSlice
+	while (True):
+		start = time.time()
+		models = [Slice, Controller, Network, User, SlicePrivilege, Site, SitePrivilege, Image, ControllerSlice]
+		objects = []
+		
+		for m in models:
+        		res = m.objects.filter(Q(policed__lt=F('updated')) | Q(policed=None))
+			objects.extend(res)	
+
+		for o in objects:
+			print "Working on %r"%o
+			execute_model_policy(o, False)
+		
+		
+		if (time.time()-start<1):
+			time.sleep(1)