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