chore: enable new ruff rules

Author: danfimov

docs/examples/extending/broker.py

@@ -1,4 +1,4 @@
-from typing import AsyncGenerator, Union
+from collections.abc import AsyncGenerator
 
 from taskiq import AckableMessage, AsyncBroker, BrokerMessage
 
@@ -23,7 +23,7 @@ async def kick(self, message: BrokerMessage) -> None:
         # Send a message.message.
         pass
 
-    async def listen(self) -> AsyncGenerator[Union[bytes, AckableMessage], None]:
+    async def listen(self) -> AsyncGenerator[bytes | AckableMessage, None]:
         while True:
             # Get new message.
             new_message: bytes = ...  # type: ignore

docs/examples/extending/cli.py

@@ -1,5 +1,5 @@
 from argparse import ArgumentParser
-from typing import Sequence
+from collections.abc import Sequence
 
 from taskiq.abc.cmd import TaskiqCMD
 

docs/examples/extending/schedule_source.py

@@ -1,5 +1,3 @@
-from typing import List
-
 from taskiq import ScheduledTask, ScheduleSource
 
 
@@ -10,7 +8,7 @@ async def startup(self) -> None:
     async def shutdown(self) -> None:
         """Do something on shutdown."""
 
-    async def get_schedules(self) -> List["ScheduledTask"]:
+    async def get_schedules(self) -> list["ScheduledTask"]:
         # Here you must return list of scheduled tasks from your source.
         return [
             ScheduledTask(

docs/examples/schedule/intro.py

@@ -1,7 +1,7 @@
 from taskiq_aio_pika import AioPikaBroker
 
-from taskiq.schedule_sources import LabelScheduleSource
 from taskiq import TaskiqScheduler
+from taskiq.schedule_sources import LabelScheduleSource
 
 broker = AioPikaBroker("amqp://guest:guest@localhost:5672/")
 

docs/examples/schedule/redis_schedule.py

@@ -15,4 +15,4 @@
 @broker.task
 async def my_task(arg1: int, arg2: str) -> None:
     """Example task."""
-    print("Hello from my_task!", arg1, arg2)  # noqa: T201
+    print("Hello from my_task!", arg1, arg2)

docs/examples/state/async_generator_deps.py

@@ -1,5 +1,5 @@
 import asyncio
-from typing import AsyncGenerator
+from collections.abc import AsyncGenerator
 
 from taskiq import TaskiqDepends
 

docs/examples/state/async_generator_deps_annot.py

@@ -1,5 +1,6 @@
 import asyncio
-from typing import Annotated, AsyncGenerator
+from collections.abc import AsyncGenerator
+from typing import Annotated
 
 from taskiq import TaskiqDepends
 

docs/examples/state/events_example.py

@@ -1,5 +1,4 @@
 import asyncio
-from typing import Optional
 
 from redis.asyncio import ConnectionPool, Redis  # type: ignore
 from taskiq_aio_pika import AioPikaBroker
@@ -30,7 +29,7 @@ async def shutdown(state: TaskiqState) -> None:
 
 
 @broker.task
-async def get_val(key: str, context: Context = TaskiqDepends()) -> Optional[str]:
+async def get_val(key: str, context: Context = TaskiqDepends()) -> str | None:
     # Now we can use our pool.
     redis = Redis(connection_pool=context.state.redis, decode_responses=True)
     return await redis.get(key)

docs/examples/state/events_example_annot.py

@@ -1,5 +1,5 @@
 import asyncio
-from typing import Annotated, Optional
+from typing import Annotated
 
 from redis.asyncio import ConnectionPool, Redis  # type: ignore
 from taskiq_aio_pika import AioPikaBroker
@@ -33,7 +33,7 @@ async def shutdown(state: TaskiqState) -> None:
 async def get_val(
     key: str,
     context: Annotated[Context, TaskiqDepends()],
-) -> Optional[str]:
+) -> str | None:
     # Now we can use our pool.
     redis = Redis(connection_pool=context.state.redis, decode_responses=True)
     return await redis.get(key)

docs/examples/state/generator_deps.py

@@ -1,4 +1,4 @@
-from typing import Generator
+from collections.abc import Generator
 
 from taskiq import TaskiqDepends