diff --git a/taskiq_faststream/broker.py b/taskiq_faststream/broker.py index 34b6b2a..36b55cc 100644 --- a/taskiq_faststream/broker.py +++ b/taskiq_faststream/broker.py @@ -10,7 +10,7 @@ from taskiq.decor import AsyncTaskiqDecoratedTask from typing_extensions import TypeAlias -from taskiq_faststream.formatter import PatchedFormatter, PathcedMessage +from taskiq_faststream.formatter import PatchedFormatter, PatchedMessage from taskiq_faststream.types import ScheduledTask from taskiq_faststream.utils import resolve_msg @@ -46,7 +46,7 @@ async def shutdown(self) -> None: await self.broker.close() await super().shutdown() - async def kick(self, message: PathcedMessage) -> None: # type: ignore[override] + async def kick(self, message: PatchedMessage) -> None: # type: ignore[override] """Call wrapped FastStream broker `publish` method.""" await _broker_publish(self.broker, message) @@ -123,7 +123,7 @@ async def shutdown(self) -> None: await self.app._shutdown() # noqa: SLF001 await super(BrokerWrapper, self).shutdown() - async def kick(self, message: PathcedMessage) -> None: # type: ignore[override] + async def kick(self, message: PatchedMessage) -> None: # type: ignore[override] """Call wrapped FastStream broker `publish` method.""" assert ( # noqa: S101 self.app.broker @@ -133,7 +133,7 @@ async def kick(self, message: PathcedMessage) -> None: # type: ignore[override] async def _broker_publish( broker: Any, - message: PathcedMessage, + message: PatchedMessage, ) -> None: async for msg in resolve_msg(message.body): await broker.publish(msg, **message.labels) diff --git a/taskiq_faststream/formatter.py b/taskiq_faststream/formatter.py index 5ed1cc2..146f04a 100644 --- a/taskiq_faststream/formatter.py +++ b/taskiq_faststream/formatter.py @@ -6,7 +6,7 @@ @dataclass -class PathcedMessage: +class PatchedMessage: """DTO to transfer data to `broker.kick`.""" body: Any @@ -19,7 +19,7 @@ class PatchedFormatter(TaskiqFormatter): def dumps( # type: ignore[override] self, message: TaskiqMessage, - ) -> PathcedMessage: + ) -> PatchedMessage: """ Dumps taskiq message to some broker message format. @@ -30,7 +30,7 @@ def dumps( # type: ignore[override] labels.pop("schedule", None) labels.pop("schedule_id", None) - return PathcedMessage( + return PatchedMessage( body=labels.pop("message", None), labels=labels, )