planetstack/observer/event_manager.py

   1 import threading
   2 import requests, json
   3
   4 from core.models import *
   5 #from openstack.manager import OpenStackManager
   6 from planetstack.config import Config
   7 from observer.deleters import deleters
   8
   9 import os
  10 import base64
  11 from fofum import Fofum
  12 import json
  13
  14 # decorator that marks dispatachable event methods
  15 def event(func):
  16         setattr(func, 'event', func.__name__)
  17         return func
  18
  19 class EventHandler:
  20         # This code is currently not in use.
  21         def __init__(self):
  22                 pass #self.manager = OpenStackManager()
  23
  24         @staticmethod
  25         def get_events():
  26                 events = []
  27                 for name in dir(EventHandler):
  28                         attribute = getattr(EventHandler, name)
  29                         if hasattr(attribute, 'event'):
  30                                 events.append(getattr(attribute, 'event'))
  31                 return events
  32
  33         def dispatch(self, event, *args, **kwds):
  34                 if hasattr(self, event):
  35                         return getattr(self, event)(*args, **kwds)
  36
  37
  38 class EventSender:
  39         def __init__(self,user=None,clientid=None):
  40                 try:
  41                         clid = Config().feefie_client_id
  42                         user = Config().feefie_client_user
  43                 except:
  44                         clid = 'planetstack_core_team'
  45                         user = 'pl'
  46
  47                 self.fofum = Fofum(user=user)
  48                 self.fofum.make(clid)
  49
  50         def fire(self,**args):
  51                 self.fofum.fire(json.dumps(args))
  52
  53 class EventListener:
  54         def __init__(self,wake_up=None):
  55                 self.handler = EventHandler()
  56                 self.wake_up = wake_up
  57
  58         def handle_event(self, payload):
  59                 payload_dict = json.loads(payload)
  60
  61                 try:
  62                         deletion = payload_dict['deletion_flag']
  63                         if (deletion):
  64                                 model = payload_dict['model']
  65                                 pk = payload_dict['pk']
  66
  67                                 for deleter in deleters[model]:
  68                                         deleter(pk)
  69                 except:
  70                         deletion = False
  71
  72                 if (not deletion and self.wake_up):
  73                         self.wake_up()
  74
  75
  76         def run(self):
  77                 # This is our unique client id, to be used when firing and receiving events
  78                 # It needs to be generated once and placed in the config file
  79
  80                 try:
  81                         clid = Config().feefie_client_id
  82                         user = Config().feefie_client_user
  83                 except:
  84                         clid = 'planetstack_core_team'
  85                         user = 'pl'
  86
  87                 f = Fofum(user=user)
  88
  89                 listener_thread = threading.Thread(target=f.listen_for_event,args=(clid,self.handle_event))
  90                 listener_thread.start()