export backends (#145)

Author: alex-oleshkevich

broadcaster/__init__.py

@@ -1,5 +1,5 @@
-from ._backends.base import BroadcastBackend
 from ._base import Broadcast, Event
+from .backends.base import BroadcastBackend
 
 __version__ = "0.3.1"
 __all__ = ["Broadcast", "Event", "BroadcastBackend"]

broadcaster/_base.py

@@ -6,7 +6,7 @@
 from urllib.parse import urlparse
 
 if TYPE_CHECKING:  # pragma: no cover
-    from broadcaster._backends.base import BroadcastBackend
+    from broadcaster.backends.base import BroadcastBackend
 
 
 class Event:
@@ -34,27 +34,27 @@ def __init__(self, url: str | None = None, *, backend: BroadcastBackend | None =
     def _create_backend(self, url: str) -> BroadcastBackend:
         parsed_url = urlparse(url)
         if parsed_url.scheme in ("redis", "rediss"):
-            from broadcaster._backends.redis import RedisBackend
+            from broadcaster.backends.redis import RedisBackend
 
             return RedisBackend(url)
 
         elif parsed_url.scheme == "redis-stream":
-            from broadcaster._backends.redis import RedisStreamBackend
+            from broadcaster.backends.redis import RedisStreamBackend
 
             return RedisStreamBackend(url)
 
         elif parsed_url.scheme in ("postgres", "postgresql"):
-            from broadcaster._backends.postgres import PostgresBackend
+            from broadcaster.backends.postgres import PostgresBackend
 
             return PostgresBackend(url)
 
         if parsed_url.scheme == "kafka":
-            from broadcaster._backends.kafka import KafkaBackend
+            from broadcaster.backends.kafka import KafkaBackend
 
             return KafkaBackend(url)
 
         elif parsed_url.scheme == "memory":
-            from broadcaster._backends.memory import MemoryBackend
+            from broadcaster.backends.memory import MemoryBackend
 
             return MemoryBackend(url)
         raise ValueError(f"Unsupported backend: {parsed_url.scheme}")

broadcaster/_backends/__init__.py broadcaster/backends/__init__.pybroadcaster/_backends/__init__.py renamed to broadcaster/backends/__init__.py

@@ -6,7 +6,7 @@
 import pytest
 
 from broadcaster import Broadcast, BroadcastBackend, Event
-from broadcaster._backends.kafka import KafkaBackend
+from broadcaster.backends.kafka import KafkaBackend
 
 
 class CustomBackend(BroadcastBackend):