Common: Fix type hints in IEventQueue and PyPubSubEventQueue

This commit is contained in:
Shreya Malviya 2022-08-09 11:46:19 +05:30
parent 9210023876
commit e92d67bfe8
2 changed files with 10 additions and 6 deletions

View File

@ -10,7 +10,7 @@ class IEventQueue(ABC):
"""
@abstractstaticmethod
def subscribe_all(subscriber: Callable[..., Any]):
def subscribe_all(subscriber: Callable[[AbstractEvent], None]):
"""
Subscribes a subscriber to all events
@ -20,7 +20,9 @@ class IEventQueue(ABC):
pass
@abstractstaticmethod
def subscribe_types(types: Sequence[AbstractEvent], subscriber: Callable[..., Any]):
def subscribe_types(
types: Sequence[AbstractEvent], subscriber: Callable[[AbstractEvent], None]
):
"""
Subscribes a subscriber to all specifed event types
@ -31,7 +33,7 @@ class IEventQueue(ABC):
pass
@abstractstaticmethod
def subscribe_tags(tags: Sequence[str], subscriber: Callable[..., Any]):
def subscribe_tags(tags: Sequence[str], subscriber: Callable[[AbstractEvent], None]):
"""
Subscribes a subscriber to all specified event tags

View File

@ -9,18 +9,20 @@ from .i_event_queue import IEventQueue
class PyPubSubEventQueue(IEventQueue):
@staticmethod
def subscribe_all(subscriber: Callable[..., Any]):
def subscribe_all(subscriber: Callable[[AbstractEvent], None]):
pub.subscribe(listener=subscriber, topicName=pub.ALL_TOPICS)
@staticmethod
def subscribe_types(types: Sequence[AbstractEvent], subscriber: Callable[..., Any]):
def subscribe_types(
types: Sequence[AbstractEvent], subscriber: Callable[[AbstractEvent], None]
):
for event_type in types:
# pypubsub.pub.subscribe needs a string as the topic/event name
event_type_name = event_type.__name__
pub.subscribe(listener=subscriber, topicName=event_type_name)
@staticmethod
def subscribe_tags(tags: Sequence[str], subscriber: Callable[..., Any]):
def subscribe_tags(tags: Sequence[str], subscriber: Callable[[AbstractEvent], None]):
for tag in tags:
pub.subscribe(listener=subscriber, topicName=tag)