update linting tools, sync lint configs with starlette, improve typings

.github/workflows/test-suite.yml

@@ -14,7 +14,7 @@ jobs:
 
     strategy:
       matrix:
-        python-version: ["3.8", "3.9", "3.10", "3.11-dev"]
+        python-version: ["3.8", "3.9", "3.10", "3.11", "3.12", "3.13-dev"]
 
     services:
       zookeeper:

broadcaster/_backends/kafka.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import typing
 from urllib.parse import urlparse
 
@@ -10,7 +12,7 @@
 class KafkaBackend(BroadcastBackend):
     def __init__(self, url: str):
         self._servers = [urlparse(url).netloc]
-        self._consumer_channels: typing.Set = set()
+        self._consumer_channels: set[str] = set()
 
     async def connect(self) -> None:
         self._producer = AIOKafkaProducer(bootstrap_servers=self._servers)

broadcaster/_backends/memory.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import asyncio
 import typing
 
@@ -7,10 +9,10 @@
 
 class MemoryBackend(BroadcastBackend):
     def __init__(self, url: str):
-        self._subscribed: typing.Set = set()
+        self._subscribed: set[str] = set()
 
     async def connect(self) -> None:
-        self._published: asyncio.Queue = asyncio.Queue()
+        self._published: asyncio.Queue[Event] = asyncio.Queue()
 
     async def disconnect(self) -> None:
         pass

broadcaster/_backends/postgres.py

@@ -13,7 +13,7 @@ def __init__(self, url: str):
 
     async def connect(self) -> None:
         self._conn = await asyncpg.connect(self._url)
-        self._listen_queue: asyncio.Queue = asyncio.Queue()
+        self._listen_queue: asyncio.Queue[Event] = asyncio.Queue()
 
     async def disconnect(self) -> None:
         await self._conn.close()

broadcaster/_base.py

@@ -1,6 +1,8 @@
+from __future__ import annotations
+
 import asyncio
 from contextlib import asynccontextmanager
-from typing import Any, AsyncGenerator, AsyncIterator, Dict, Optional
+from typing import Any, AsyncGenerator, AsyncIterator
 from urllib.parse import urlparse
 
 
@@ -10,11 +12,7 @@ def __init__(self, channel: str, message: str) -> None:
         self.message = message
 
     def __eq__(self, other: object) -> bool:
-        return (
-            isinstance(other, Event)
-            and self.channel == other.channel
-            and self.message == other.message
-        )
+        return isinstance(other, Event) and self.channel == other.channel and self.message == other.message
 
     def __repr__(self) -> str:
         return f"Event(channel={self.channel!r}, message={self.message!r})"
@@ -30,7 +28,7 @@ def __init__(self, url: str):
 
         parsed_url = urlparse(url)
         self._backend: BroadcastBackend
-        self._subscribers: Dict[str, Any] = {}
+        self._subscribers: dict[str, set[asyncio.Queue[Event | None]]] = {}
         if parsed_url.scheme in ("redis", "rediss"):
             from broadcaster._backends.redis import RedisBackend
 
@@ -51,7 +49,7 @@ def __init__(self, url: str):
 
             self._backend = MemoryBackend(url)
 
-    async def __aenter__(self) -> "Broadcast":
+    async def __aenter__(self) -> Broadcast:
         await self.connect()
         return self
 
@@ -79,8 +77,8 @@ async def publish(self, channel: str, message: Any) -> None:
         await self._backend.publish(channel, message)
 
     @asynccontextmanager
-    async def subscribe(self, channel: str) -> AsyncIterator["Subscriber"]:
-        queue: asyncio.Queue = asyncio.Queue()
+    async def subscribe(self, channel: str) -> AsyncIterator[Subscriber]:
+        queue: asyncio.Queue[Event | None] = asyncio.Queue()
 
         try:
             if not self._subscribers.get(channel):
@@ -99,10 +97,10 @@ async def subscribe(self, channel: str) -> AsyncIterator["Subscriber"]:
 
 
 class Subscriber:
-    def __init__(self, queue: asyncio.Queue) -> None:
+    def __init__(self, queue: asyncio.Queue[Event | None]) -> None:
         self._queue = queue
 
-    async def __aiter__(self) -> Optional[AsyncGenerator]:
+    async def __aiter__(self) -> AsyncGenerator[Event | None, None] | None:
         try:
             while True:
                 yield await self.get()

pyproject.toml

@@ -24,6 +24,8 @@ classifiers = [
     "Programming Language :: Python :: 3.9",
     "Programming Language :: Python :: 3.10",
     "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
 ]
 dependencies = [
     "anyio>=3.4.0,<5",
@@ -48,14 +50,17 @@ include = [
 ]
 
 [tool.ruff]
-ignore = []
 line-length = 120
-select = ["E","F","W"]
+
+[tool.ruff.lint]
+select = ["E", "F", "I", "FA", "UP"]
 
 [tool.ruff.isort]
 combine-as-imports = true
 
 [tool.mypy]
+strict = true
+python_version = "3.8"
 disallow_untyped_defs = true
 ignore_missing_imports = true
 

requirements.txt

@@ -1,18 +1,17 @@
 -e .[redis,postgres,kafka]
 
 # Documentation
-mkdocs==1.3.1
+mkdocs==1.5.3
+mkdocs-material==9.5.12
 mkautodoc==0.2.0
-mkdocs-material==8.4.0
 
 # Packaging
-build==0.10.0
-twine==4.0.1
+build==1.1.1
+twine==5.0.0
 
 # Tests & Linting
-ruff==0.0.277
-black==22.6.0
-coverage==6.4.4
-mypy==0.971
-pytest==7.1.2
-pytest-asyncio==0.19.0
+ruff==0.1.15
+coverage==7.4.3
+mypy==1.8.0
+pytest==8.0.2
+pytest-asyncio==0.23.6

scripts/check

@@ -8,6 +8,6 @@ export SOURCE_FILES="broadcaster tests"
 
 set -x
 
-${PREFIX}black --check --diff --target-version=py37 $SOURCE_FILES
-${PREFIX}ruff check $SOURCE_FILES
+${PREFIX}ruff format --check --diff $SOURCE_FILES
 ${PREFIX}mypy $SOURCE_FILES
+${PREFIX}ruff check $SOURCE_FILES

scripts/lint

@@ -1,12 +1,12 @@
 #!/bin/sh -e
 
 export PREFIX=""
-if [ -d 'venv' ]; then
+if [ -d 'venv' ] ; then
     export PREFIX="venv/bin/"
 fi
 export SOURCE_FILES="broadcaster tests"
 
 set -x
 
-${PREFIX}ruff --fix $SOURCE_FILES
-${PREFIX}black --target-version=py37 $SOURCE_FILES
+${PREFIX}ruff format $SOURCE_FILES
+${PREFIX}ruff --fix $SOURCE_FILES

tests/test_broadcast.py

@@ -9,6 +9,7 @@ async def test_memory():
         async with broadcast.subscribe("chatroom") as subscriber:
             await broadcast.publish("chatroom", "hello")
             event = await subscriber.get()
+            assert event
             assert event.channel == "chatroom"
             assert event.message == "hello"
 
@@ -25,9 +26,7 @@ async def test_redis():
 
 @pytest.mark.asyncio
 async def test_postgres():
-    async with Broadcast(
-        "postgres://postgres:postgres@localhost:5432/broadcaster"
-    ) as broadcast:
+    async with Broadcast("postgres://postgres:postgres@localhost:5432/broadcaster") as broadcast:
         async with broadcast.subscribe("chatroom") as subscriber:
             await broadcast.publish("chatroom", "hello")
             event = await subscriber.get()

tests/test_unsubscribe.py

@@ -1,4 +1,5 @@
 import pytest
+
 from broadcaster import Broadcast