mrtoth/baseimage-tutorial-framework
1
0
Fork 0
mirror of https://github.com/avatao-content/baseimage-tutorial-framework synced 2024-11-22 12:51:32 +00:00
Code Issues Releases Wiki Activity

Refactor reset in StatefulEventHandler

Browse Source
This commit is contained in:
Bálint Bokros 2017-12-05 18:35:22 +01:00
parent 6f16a335db
commit ad89c8fb57
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/event_handlers/stateful_event_handler.py
View File

@ -16,10 +16,7 @@ class StatefulEventHandler(EventHandlerBase):
def event_handler_callback(msg_parts): def event_handler_callback(msg_parts):
anchor, message = msg_parts anchor, message = msg_parts
if anchor == b'reset': if anchor == b'reset':
self.generator = None self.reset()
self.unsubscribe_all()
self.subscribe(self.anchor)
self.subscribe('reset')
return return
data_json = json.loads(message) data_json = json.loads(message)
if self.generator is None: if self.generator is None:
@ -36,3 +33,12 @@ class StatefulEventHandler(EventHandlerBase):
self.zmq_push_socket.send_multipart(response) self.zmq_push_socket.send_multipart(response)
self.zmq_sub_stream.on_recv(event_handler_callback) self.zmq_sub_stream.on_recv(event_handler_callback)
def unsubscribe_all(self):
super().unsubscribe_all()
self.subscribe('reset')
def reset(self):
self.generator = None
self.unsubscribe_all()
self.subscribe(self.anchor)