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

Allow passing an iterable with keys to subscribe to in EventHandlerBase.__init__()

Browse Source
This commit is contained in:
Kristóf Tóth 2019-05-06 17:11:56 +02:00
parent 90b780a5c0
commit 9ad77eaed8
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lib/tfw/event_handler_base.py
View File

@ -3,6 +3,7 @@
from abc import ABC, abstractmethod from abc import ABC, abstractmethod
from inspect import currentframe from inspect import currentframe
from typing import Iterable
from tfw.networking.event_handlers import ServerConnector from tfw.networking.event_handlers import ServerConnector
from tfw.crypto import message_checksum, KeyManager, verify_message from tfw.crypto import message_checksum, KeyManager, verify_message
@ -20,7 +21,12 @@ class EventHandlerBase(ABC):
""" """
def __init__(self, key): def __init__(self, key):
self.server_connector = ServerConnector() self.server_connector = ServerConnector()
self.keys = [key] self.keys = []
if isinstance(key, str):
self.keys.append(key)
elif isinstance(key, Iterable):
self.keys = list(key)
self.subscribe(*self.keys) self.subscribe(*self.keys)
self.server_connector.register_callback(self.event_handler_callback) self.server_connector.register_callback(self.event_handler_callback)