refactor(playout): add typings and fix linting errors
move EVENT_KEY_FORMAT to events module properly type fetch queue event start/end can be str or datetime
This commit is contained in:
parent
3fba7c73d3
commit
e88e843b65
11 changed files with 286 additions and 213 deletions
|
@ -3,10 +3,11 @@ import logging
|
|||
from queue import Queue as ThreadQueue
|
||||
from signal import SIGTERM, signal
|
||||
from time import sleep
|
||||
from typing import Any, Dict
|
||||
from typing import Any, Dict, Union
|
||||
|
||||
# For RabbitMQ
|
||||
from kombu.connection import Connection
|
||||
from kombu.message import Message
|
||||
from kombu.messaging import Exchange, Queue
|
||||
from kombu.mixins import ConsumerMixin
|
||||
|
||||
|
@ -35,7 +36,7 @@ class MessageHandler(ConsumerMixin):
|
|||
Consumer(queues, callbacks=[self.on_message], accept=["text/plain"]),
|
||||
]
|
||||
|
||||
def on_message(self, body, message):
|
||||
def on_message(self, body, message: Message):
|
||||
logger.debug("received message: %s", body)
|
||||
try:
|
||||
try:
|
||||
|
@ -78,7 +79,7 @@ class MessageListener:
|
|||
def __init__(
|
||||
self,
|
||||
config: Config,
|
||||
fetch_queue: ThreadQueue[Dict[str, Any]],
|
||||
fetch_queue: ThreadQueue[Union[str, bytes]],
|
||||
recorder_queue: ThreadQueue[Dict[str, Any]],
|
||||
) -> None:
|
||||
self.config = config
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue