Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

Support one-shot event hooks (src/EventManager.py)

Browse source
This commit is contained in:
jesopo 2019-02-14 10:36:11 +00:00
parent 33cacbb904
commit 708ba2ddc7
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
src/EventManager.py
View file

@ -28,11 +28,13 @@ class Event(object):
self.eaten = True self.eaten = True
class EventCallback(object): class EventCallback(object):
def __init__(self, function: CALLBACK_TYPE, priority: int, kwargs: dict): def __init__(self, function: CALLBACK_TYPE, priority: int, kwargs: dict,
one_shot: bool=False):
self.function = function self.function = function
self.priority = priority self.priority = priority
self.kwargs = kwargs self.kwargs = kwargs
self.docstring = utils.parse.docstring(function.__doc__ or "") self.docstring = utils.parse.docstring(function.__doc__ or "")
self._one_shot = one_shot
def call(self, event: Event) -> typing.Any: def call(self, event: Event) -> typing.Any:
return self.function(event) return self.function(event)
@ -206,6 +208,9 @@ class EventHook(object):
else: else:
raise raise
if hook._one_shot:
self.unhook(hook)
total_milliseconds = (time.monotonic() - start) * 1000 total_milliseconds = (time.monotonic() - start) * 1000
self.log.trace("event \"%s\" called in %fms", self.log.trace("event \"%s\" called in %fms",
[event_path, total_milliseconds]) [event_path, total_milliseconds])