planetstack/observer/event_manager.py

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