Mirrored_Repos/houdini
1
0
Fork 0
mirror of https://github.com/solero/houdini.git synced 2024-11-14 14:48:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add support for ordering handlers by referencing before and after

Browse Source
This commit is contained in:
Ben 2019-06-21 01:41:14 +01:00
parent 9bb32d94d2
commit 1a5cf4868c
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
houdini/handlers/__init__.py
View File

@ -54,13 +54,16 @@ class Priority(enum.Enum):
class _Listener(_ArgumentDeserializer): class _Listener(_ArgumentDeserializer):
__slots__ = ['priority', 'packet', 'overrides'] __slots__ = ['priority', 'packet', 'overrides', 'before', 'after']
def __init__(self, packet, callback, **kwargs): def __init__(self, packet, callback, **kwargs):
super().__init__(packet.id, callback, **kwargs) super().__init__(packet.id, callback, **kwargs)
self.packet = packet self.packet = packet
self.priority = kwargs.get('priority', Priority.Low) self.priority = kwargs.get('priority', Priority.Low)
self.before = kwargs.get('before')
self.after = kwargs.get('after')
self.overrides = kwargs.get('overrides', []) self.overrides = kwargs.get('overrides', [])
if type(self.overrides) is not list: if type(self.overrides) is not list:
@ -146,6 +149,14 @@ class _ListenerManager(_AbstractManager):
self[listener_object.packet].append(listener_object) self[listener_object.packet].append(listener_object)
for listener_name, listener_object in listener_objects: for listener_name, listener_object in listener_objects:
if listener_object.before is not None:
index_of_before = self[listener_object.packet].index(listener_object.before)
old_index = self[listener_object.packet].index(listener_object)
self[listener_object.packet].insert(index_of_before, self[listener_object.packet].pop(old_index))
if listener_object.after is not None:
index_of_after = self[listener_object.packet].index(listener_object.after)
old_index = self[listener_object.packet].index(listener_object)
self[listener_object.packet].insert(index_of_after + 1, self[listener_object.packet].pop(old_index))
for override in listener_object.overrides: for override in listener_object.overrides:
self[override.packet].remove(override) self[override.packet].remove(override)