4 from planetstack.config import Config
5 from observer.deleter import Deleter
12 from fofum import Fofum
17 def get_random_client_id():
18 global random_client_id
20 if (random_client_id is None) and os.path.exists("/opt/planetstack/random_client_id"):
21 # try to use the last one we used, if we saved it
23 random_client_id = open("/opt/planetstack/random_client_id","r").readline().strip()
24 print "get_random_client_id: loaded %s" % random_client_id
26 print "get_random_client_id: failed to read /opt/planetstack/random_client_id"
28 if random_client_id is None:
29 random_client_id = base64.urlsafe_b64encode(os.urandom(12))
30 print "get_random_client_id: generated new id %s" % random_client_id
32 # try to save it for later (XXX: could race with another client here)
34 open("/opt/planetstack/random_client_id","w").write("%s\n" % random_client_id)
36 print "get_random_client_id: failed to write /opt/planetstack/random_client_id"
38 return random_client_id
40 # decorator that marks dispatachable event methods
42 setattr(func, 'event', func.__name__)
46 # This code is currently not in use.
53 for name in dir(EventHandler):
54 attribute = getattr(EventHandler, name)
55 if hasattr(attribute, 'event'):
56 events.append(getattr(attribute, 'event'))
59 def dispatch(self, event, *args, **kwds):
60 if hasattr(self, event):
61 return getattr(self, event)(*args, **kwds)
65 def __init__(self,user=None,clientid=None):
67 user = Config().feefie_client_user
72 clid = Config().feefie_client_id
74 clid = get_random_client_id()
75 print "EventSender: no feefie_client_id configured. Using random id %s" % clid
77 self.fofum = Fofum(user=user)
80 def fire(self,**kwargs):
81 kwargs["uuid"] = str(uuid.uuid1())
82 self.fofum.fire(json.dumps(kwargs))
85 def __init__(self,wake_up=None):
86 self.handler = EventHandler()
87 self.wake_up = wake_up
89 self.load_deleter_modules()
91 def load_deleter_modules(self, deleter_dir=None):
92 if deleter_dir is None:
93 if hasattr(Config(), "observer_deleters_dir"):
94 deleter_dir = Config().observer_deleters_dir
96 deleter_dir = "/opt/planetstack/observer/deleters"
98 for fn in os.listdir(deleter_dir):
99 pathname = os.path.join(deleter_dir,fn)
100 if os.path.isfile(pathname) and fn.endswith(".py") and (fn!="__init__.py"):
101 module = imp.load_source(fn[:-3],pathname)
102 for classname in dir(module):
103 c = getattr(module, classname, None)
105 # make sure 'c' is a descendent of Deleter and has a
106 # provides field (this eliminates the abstract base classes
107 # since they don't have a provides)
109 if inspect.isclass(c) and issubclass(c, Deleter) and hasattr(c,"model") and c.model!=None:
111 if not modelName in self.deleters:
112 self.deleters[modelName] = []
113 if not (c in self.deleters[modelName]):
114 self.deleters[modelName].append(c)
115 print 'loaded deleters: %s' % ",".join(self.deleters.keys())
118 def handle_event(self, payload):
119 payload_dict = json.loads(payload)
122 deletion = payload_dict.get('delete_flag', False)
124 model = payload_dict['model']
125 pk = payload_dict['pk']
126 model_dict = payload_dict['model_dict']
128 for deleter in self.deleters[model]:
130 deleter()(pk, model_dict)
132 # something is silently eating these
134 traceback.print_exc()
140 if (not deletion and self.wake_up):
144 # This is our unique client id, to be used when firing and receiving events
145 # It needs to be generated once and placed in the config file
148 user = Config().feefie_client_user
153 clid = Config().feefie_client_id
155 clid = get_random_client_id()
156 print "EventListener: no feefie_client_id configured. Using random id %s" % clid
160 listener_thread = threading.Thread(target=f.listen_for_event,args=(clid,self.handle_event))
161 listener_thread.start()