rustico/core
2
0
Fork 0
Code Issues 9 Pull Requests Projects Releases Wiki Activity

added any_state/event in handler

Browse Source
This commit is contained in:
Giulio De Pasquale 2020-12-04 12:13:13 +00:00
parent 5d5fbc91e1
commit 8391cec49e
1 changed files with 34 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
bfxbot/models.py
View File

@ -164,23 +164,37 @@ class EventHandler:
def __init__(self):
self.event_handlers = {}
self.state_handlers = {}
self.any_events = []
self.any_state = []
async def call_event(self, status: SymbolStatus, event: Event):
async def call_event(self, event: Event, status: SymbolStatus):
value = event.kind.value
if value in self.event_handlers:
for h in self.event_handlers[value]:
if inspect.iscoroutinefunction(h):
await h(status, event)
await h(event, status)
else:
h(status, event)
h(event, status)
async def call_state(self, status: SymbolStatus, pw: PositionWrapper):
if pw.state in self.state_handlers:
for h in self.state_handlers[pw.state]:
for h in self.any_events:
if inspect.iscoroutinefunction(h):
await h(event, status)
else:
h(event, status)
async def call_state(self, state: PositionState, status: SymbolStatus):
if state in self.state_handlers:
for h in self.state_handlers[state]:
if inspect.iscoroutinefunction(h):
await h(status, pw)
await h(status)
else:
h(status, pw)
h(status)
for h in self.any_state:
if inspect.iscoroutinefunction(h):
await h(status)
else:
h(status)
def on_event(self, kind: EventKind):
value = kind.value
@ -203,3 +217,15 @@ class EventHandler:
return handler
return registerhandler
def on_any_event(self):
def registerhandle(handler):
self.any_events.append(handler)
return handler
return registerhandle
def on_any_state(self):
def registerhandle(handler):
self.any_state.append(handler)
return handler
return registerhandle