pass pk to deleter, support model as a class or instance, give each event a uuid...
[plstackapi.git] / planetstack / observer / event_manager.py
index ab50f66..8ec3fc2 100644 (file)
 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      
+       setattr(func, 'event', func.__name__)
+       return func                     
 
 class EventHandler:
-    # This code is currently not in use.
-    def __init__(self):
-        pass #self.manager = OpenStackManager()
-
-    @staticmethod
-    def get_events():
-        events = []
-        for name in dir(EventHandler):
-            attribute = getattr(EventHandler, name)
-            if hasattr(attribute, 'event'):
-                events.append(getattr(attribute, 'event'))
-        return events
-
-    def dispatch(self, event, *args, **kwds):
-        if hasattr(self, event):
-            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)                            
-
-    
+       # This code is currently not in use.
+       def __init__(self):
+               pass 
+
+       @staticmethod
+       def get_events():
+               events = []
+               for name in dir(EventHandler):
+                       attribute = getattr(EventHandler, name)
+                       if hasattr(attribute, 'event'):
+                               events.append(getattr(attribute, 'event'))
+               return events
+
+       def dispatch(self, event, *args, **kwds):
+               if hasattr(self, event):
+                       return getattr(self, event)(*args, **kwds)
+                       
+               
 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'
-
-        self.fofum = Fofum(user=user)
-        self.fofum.make(clid)
-
-    def fire(self):
-        self.fofum.fire()
+       def __init__(self,user=None,clientid=None):
+               try:
+                       user = Config().feefie_client_user
+               except:
+                       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,**kwargs):
+                kwargs["uuid"] = str(uuid.uuid1())
+                print "YYY fire", kwargs
+               self.fofum.fire(json.dumps(kwargs))
 
 class EventListener:
-    def __init__(self,wake_up=None):
-        self.handler = EventHandler()
-        self.wake_up = wake_up
-
-    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)   
-
-        if (self.wake_up):
-            self.wake_up()
-        
-
-    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'
-
-        f = Fofum(user=user)
-        
-        listener_thread = threading.Thread(target=f.listen_for_event,args=(clid,self.handle_event))
-        listener_thread.start()
+       def __init__(self,wake_up=None):
+               self.handler = EventHandler()
+               self.wake_up = wake_up
+
+       def handle_event(self, payload):
+               payload_dict = json.loads(payload)
+
+               try:
+                       deletion = payload_dict.get('delete_flag', False)
+                        print "XXX", payload_dict, deletion
+                       if (deletion):
+                               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:
+                       user = Config().feefie_client_user
+               except:
+                       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))
+               listener_thread.start()