initial checkin of event listener. refactor/cleanup

[plstackapi.git] / planetstack / openstack / event_listener.py

diff --git a/planetstack/openstack/event_listener.py b/planetstack/openstack/event_listener.py
new file mode 100644 (file)
index 0000000..d3f0abf
--- /dev/null
+++ b/planetstack/openstack/event_listener.py
@@ -0,0 +1,114 @@
+import threading
+import requests, json
+from core.models import *
+from openstack.manager import OpenStackManager
+
+# decorator that marks dispatachable event methods  
+def event(func):
+    setattr(func, 'event', func.__name__)
+    return func      
+
+class EventHandler:
+
+    def __init__(self):
+        self.manager = OpenStackManager()
+
+    def get_events(self):
+        events = []
+        for attrib in dir(self):
+            if hasattr(attrib, 'event'):
+                events.append(getattr(attrib, '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)                            
+
+    
+
+class EventListener:
+
+    def __init__(self):
+        self.handler = EventHandler()
+
+    def listen_for_event(self, event, hash):
+        url = 'http://www.feefie.com/command'
+        params = {'action': 'subscribe',
+                  'hash': hash,
+                  'htm': 1}
+        while True:
+            r = requests.get(url, params=params)
+            r_data = json.loads(r)
+            payload = r_data.get('payload')
+            self.handler.dispatch(event, **payload)
+
+
+    def run(self):
+        # register events
+        event_names = [{'title': name} for name in self.handler.get_events()]
+        url = 'http://www.feefie.com/command'
+        params = {'action': 'add',
+                  'u': 'pl',
+                  'events': event_names}
+        r = requests.get(url, params=params)
+        print dir(r)
+        print r
+        r_data = json.loads(r)
+        events = r_data.get('events', [])
+        # spanw a  thread for each event
+        for event in events:
+            args = (event['title'], event['hash'])
+            listener_thread = threading.Thread(target=self.listen_for_event, args=args)
+            listener_tread.start()
+