lint: apply ruff changes

Author: Lancetnik

taskiq_faststream/broker.py

@@ -1,14 +1,13 @@
 import typing
 import warnings
-from typing import Any
+from typing import Any, TypeAlias
 
 import anyio
 from faststream._internal.application import Application
 from faststream.types import SendableMessage
 from taskiq import AsyncBroker
 from taskiq.acks import AckableMessage
 from taskiq.decor import AsyncTaskiqDecoratedTask
-from typing_extensions import TypeAlias
 
 from taskiq_faststream.formatter import PatchedFormatter, PatchedMessage
 from taskiq_faststream.types import ScheduledTask
@@ -52,7 +51,7 @@ async def kick(self, message: PatchedMessage) -> None:  # type: ignore[override]
 
     async def listen(
         self,
-    ) -> typing.AsyncGenerator[typing.Union[bytes, AckableMessage], None]:
+    ) -> typing.AsyncGenerator[bytes | AckableMessage, None]:
         """Not supported method."""
         while True:
             warnings.warn(
@@ -68,14 +67,12 @@ async def listen(
 
     def task(  # type: ignore[override]
         self,
-        message: typing.Union[
-            None,
-            SendableMessage,
-            typing.Callable[[], SendableMessage],
-            typing.Callable[[], typing.Awaitable[SendableMessage]],
-            typing.Callable[[], typing.Generator[SendableMessage, None, None]],
-            typing.Callable[[], typing.AsyncGenerator[SendableMessage, None]],
-        ] = None,
+        message: None
+        | SendableMessage
+        | typing.Callable[[], SendableMessage]
+        | typing.Callable[[], typing.Awaitable[SendableMessage]]
+        | typing.Callable[[], typing.Generator[SendableMessage, None, None]]
+        | typing.Callable[[], typing.AsyncGenerator[SendableMessage, None]] = None,
         *,
         schedule: list[ScheduledTask],
         **kwargs: PublishParameters,

taskiq_faststream/types.py

@@ -1,5 +1,4 @@
 from datetime import datetime, timedelta
-from typing import Optional, Union
 
 from typing_extensions import TypedDict
 
@@ -11,5 +10,5 @@ class ScheduledTask(TypedDict, total=False):
     """
 
     cron: str
-    cron_offset: Union[str, timedelta, None]
-    time: Optional[datetime]
+    cron_offset: str | timedelta | None
+    time: datetime | None

taskiq_faststream/utils.py

@@ -10,14 +10,12 @@
 
 
 async def resolve_msg(
-    msg: typing.Union[
-        None,
-        SendableMessage,
-        typing.Callable[[], SendableMessage],
-        typing.Callable[[], typing.Awaitable[SendableMessage]],
-        typing.Callable[[], typing.Iterator[SendableMessage]],
-        typing.Callable[[], typing.AsyncIterator[SendableMessage]],
-    ],
+    msg: None
+    | SendableMessage
+    | typing.Callable[[], SendableMessage]
+    | typing.Callable[[], typing.Awaitable[SendableMessage]]
+    | typing.Callable[[], typing.Iterator[SendableMessage]]
+    | typing.Callable[[], typing.AsyncIterator[SendableMessage]],
 ) -> typing.AsyncIterator[SendableMessage]:
     """Resolve message generation callback.
 

tests/test_resolve_message.py

@@ -21,14 +21,12 @@
 )
 @pytest.mark.anyio
 async def test_resolve_msg(
-    msg: typing.Union[
-        None,
-        SendableMessage,
-        typing.Callable[[], SendableMessage],
-        typing.Callable[[], typing.Awaitable[SendableMessage]],
-        typing.Callable[[], typing.Generator[SendableMessage, None, None]],
-        typing.Callable[[], typing.AsyncGenerator[SendableMessage, None]],
-    ],
+    msg: None
+    | SendableMessage
+    | typing.Callable[[], SendableMessage]
+    | typing.Callable[[], typing.Awaitable[SendableMessage]]
+    | typing.Callable[[], typing.Generator[SendableMessage, None, None]]
+    | typing.Callable[[], typing.AsyncGenerator[SendableMessage, None]],
 ) -> None:
     async for m in resolve_msg(msg):
         assert m == messages.message

tests/testcase.py

@@ -89,14 +89,12 @@ async def test_task_multiple_schedules_by_cron(
         subject: str,
         broker: Any,
         event: asyncio.Event,
-        msg: typing.Union[
-            None,
-            SendableMessage,
-            typing.Callable[[], SendableMessage],
-            typing.Callable[[], typing.Awaitable[SendableMessage]],
-            typing.Callable[[], typing.Generator[SendableMessage, None, None]],
-            typing.Callable[[], typing.AsyncGenerator[SendableMessage, None]],
-        ],
+        msg: None
+        | SendableMessage
+        | typing.Callable[[], SendableMessage]
+        | typing.Callable[[], typing.Awaitable[SendableMessage]]
+        | typing.Callable[[], typing.Generator[SendableMessage, None, None]]
+        | typing.Callable[[], typing.AsyncGenerator[SendableMessage, None]],
     ) -> None:
         """Test cron runs twice via StreamScheduler."""
         received_message = []