pass pk to deleter, support model as a class or instance, give each event a uuid...

[plstackapi.git] / planetstack / observer / event_manager.py

diff --git a/planetstack/observer/event_manager.py b/planetstack/observer/event_manager.py
index f24026a..8ec3fc2 100644 (file)
--- a/planetstack/observer/event_manager.py
+++ b/planetstack/observer/event_manager.py
@@ -1,23 +1,24 @@
 import threading
 import requests, json
 
-from core.models import *
-#from openstack.manager import OpenStackManager
 from planetstack.config import Config
+from observer.deleters import deleters
 
+import uuid
 import os
 import base64
 from fofum import Fofum
+import json
 
-# decorator that marks dispatachable event methods     
+# decorator that marks dispatachable event methods       
 def event(func):
        setattr(func, 'event', func.__name__)
-       return func              
+       return func                     
 
 class EventHandler:
        # This code is currently not in use.
        def __init__(self):
-               pass #self.manager = OpenStackManager()
+               pass 
 
        @staticmethod
        def get_events():
@@ -33,71 +34,26 @@ class EventHandler:
                        return getattr(self, event)(*args, **kwds)
                        
                
-       @event
-       def save_site(self, id):
-               sites = Site.objects.filter(id=id)
-               if sites:
-                       self.manager.save_site(sites[0])
-       
-       @event
-       def delete_site(self, tenant_id):
-               self.manager.driver.delete_tenant(tenant_id)
-
-       @event
-       def save_site_privilege(self, id):
-               site_privileges = SitePrivilege.objects.filter(id=id)
-               if site_privileges:
-                       site_priv = self.manager.save_site_privilege(site_privileges[0])
-
-       @event
-       def delete_site_privilege(self, kuser_id, tenant_id, role_type):
-               self.manager.driver.delete_user_role(kuser_id, tenant_id, role_type)
-
-       @event
-       def save_slice(self, id):
-               slices = Slice.objects.filter(id=id)
-               if slices:
-                       self.manager.save_slice(slices[0])
-       
-       @event
-       def delete_slice(self, tenant_id, network_id, router_id, subnet_id):
-               self.manager._delete_slice(tenant_id, network_id, router_id, subnet_id)
-
-       @event
-       def save_user(self, id):
-               users = User.objects.filter(id=id)
-               if users:
-                       self.manager.save_user(users[0])
-               
-       @event
-       def delete_user(self, kuser_id):
-               self.manager.driver.delete_user(kuser_id)
-       
-       @event
-       def save_sliver(self, id):
-               slivers = Sliver.objects.filter(id=id)
-               if slivers:
-                       self.manager.save_sliver(slivers[0])
-
-       @event
-       def delete_sliver(self, instance_id):
-               self.manager.destroy_instance(instance_id)                                                        
-
-       
 class EventSender:
        def __init__(self,user=None,clientid=None):
                try:
-                       clid = Config().feefie_client_id
                        user = Config().feefie_client_user
                except:
-                       clid = 'planetstack_core_team'
                        user = 'pl'
 
+               try:
+                       clid = Config().feefie_client_id
+               except:
+                       clid = self.random_client_id()
+                       
+
                self.fofum = Fofum(user=user)
                self.fofum.make(clid)
 
-       def fire(self):
-               self.fofum.fire()
+       def fire(self,**kwargs):
+                kwargs["uuid"] = str(uuid.uuid1())
+                print "YYY fire", kwargs
+               self.fofum.fire(json.dumps(kwargs))
 
 class EventListener:
        def __init__(self,wake_up=None):
@@ -107,37 +63,45 @@ class EventListener:
        def handle_event(self, payload):
                payload_dict = json.loads(payload)
 
-       # The code below will come back when we optimize the observer syncs
-       # into 'small' and 'big' syncs.
-
-               #event = payload_dict['event']
-               #ctx = payload_dict['ctx']
-               #self.handler.dispatch(event,**ctx)   
-
                try:
-                       deletion = payload_dict['deletion_flag']
+                       deletion = payload_dict.get('delete_flag', False)
+                        print "XXX", payload_dict, deletion
                        if (deletion):
-                               cmd = payload_dict['command']
-                               ctx = payload_dict['ctx']
-                               self.handler.dispatch(cmd,**ctx)
+                               model = payload_dict['model']
+                               pk = payload_dict['pk']
+
+                                print "XXX", model, pk, deleters
+
+                               for deleter in deleters[model]:
+                                        print "ZZZ executing deleter"
+                                       deleter(pk)
                except:
                        deletion = False
 
                if (not deletion and self.wake_up):
                        self.wake_up()
-               
+
+       def random_client_id(self):
+               try:
+                       return self.client_id
+               except AttributeError:
+                       self.client_id = base64.urlsafe_b64encode(os.urandom(12))
+                       return self.client_id
 
        def run(self):
                # This is our unique client id, to be used when firing and receiving events
                # It needs to be generated once and placed in the config file
 
                try:
-                       clid = Config().feefie_client_id
                        user = Config().feefie_client_user
                except:
-                       clid = 'planetstack_core_team'
                        user = 'pl'
 
+               try:
+                       clid = Config().feefie_client_id
+               except:
+                       clid = self.random_client_id()
+
                f = Fofum(user=user)
                
                listener_thread = threading.Thread(target=f.listen_for_event,args=(clid,self.handle_event))