Lint

ypy_websocket/awareness.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import json
 import time
 from typing import Any, Dict
@@ -11,7 +13,7 @@ def __init__(self, ydoc):
         self.meta = {}
         self.states = {}
 
-    def get_changes(self, message: bytes) -> Dict[str, Any]:
+    def get_changes(self, message: bytes) -> dict[str, Any]:
         message = read_message(message)
         decoder = Decoder(message)
         timestamp = int(time.time() * 1000)

ypy_websocket/websocket_server.py

@@ -54,7 +54,7 @@ def delete_room(self, *, name: str | None = None, room: YRoom | None = None):
         if name is None:
             name = self.get_room_name(room)
         room = self.rooms[name]
-        room.exit()
+        room.stop()
         del self.rooms[name]
 
     async def serve(self, websocket):

ypy_websocket/yroom.py

@@ -3,7 +3,7 @@
 import logging
 from contextlib import AsyncExitStack
 from functools import partial
-from typing import Callable, List, Optional
+from typing import Callable
 
 import y_py as Y
 from anyio import Event, create_memory_object_stream, create_task_group

ypy_websocket/ystore.py

@@ -7,7 +7,7 @@
 from abc import ABC, abstractmethod
 from contextlib import AsyncExitStack
 from pathlib import Path
-from typing import AsyncIterator, Callable, Optional, Tuple
+from typing import AsyncIterator, Callable
 
 import aiosqlite
 import anyio

ypy_websocket/yutils.py

@@ -1,6 +1,7 @@
+from __future__ import annotations
+
 from enum import IntEnum
 from pathlib import Path
-from typing import Optional
 
 import anyio
 import y_py as Y
@@ -70,7 +71,7 @@ def read_var_uint(self) -> int:
                 break
         return uint
 
-    def read_message(self) -> Optional[bytes]:
+    def read_message(self) -> bytes | None:
         if self.length == 0:
             return None
         length = self.read_var_uint()