Add PubSubBroker in __init__.py (#31)

Author: booqoffsky

README.md

@@ -20,8 +20,7 @@ Let's see the example with the redis broker and redis async result:
 ```python
 import asyncio
 
-from taskiq_redis.redis_broker import ListQueueBroker
-from taskiq_redis.redis_backend import RedisAsyncResultBackend
+from taskiq_redis import ListQueueBroker, RedisAsyncResultBackend
 
 redis_async_result = RedisAsyncResultBackend(
     redis_url="redis://localhost:6379",

taskiq_redis/__init__.py

@@ -1,5 +1,9 @@
 """Package for redis integration."""
 from taskiq_redis.redis_backend import RedisAsyncResultBackend
-from taskiq_redis.redis_broker import ListQueueBroker
+from taskiq_redis.redis_broker import ListQueueBroker, PubSubBroker
 
-__all__ = ["RedisAsyncResultBackend", "ListQueueBroker"]
+__all__ = [
+    "RedisAsyncResultBackend",
+    "ListQueueBroker",
+    "PubSubBroker",
+]

tests/test_backend.py

@@ -5,11 +5,11 @@
 import pytest
 from taskiq import TaskiqResult
 
+from taskiq_redis import RedisAsyncResultBackend
 from taskiq_redis.exceptions import (
     DuplicateExpireTimeSelectedError,
     ExpireTimeMustBeMoreThanZeroError,
 )
-from taskiq_redis.redis_backend import RedisAsyncResultBackend
 
 _ReturnType = TypeVar("_ReturnType")
 

tests/test_broker.py

@@ -4,7 +4,7 @@
 import pytest
 from taskiq import AsyncBroker, BrokerMessage
 
-from taskiq_redis.redis_broker import ListQueueBroker, PubSubBroker
+from taskiq_redis import ListQueueBroker, PubSubBroker
 
 
 async def get_message(broker: AsyncBroker) -> BrokerMessage:  # type: ignore

tests/test_result_backend.py

@@ -3,7 +3,7 @@
 import pytest
 from taskiq import TaskiqResult
 
-from taskiq_redis.redis_backend import RedisAsyncResultBackend
+from taskiq_redis import RedisAsyncResultBackend
 
 
 @pytest.mark.anyio