improved typing

bs_rpc/__init__.py

@@ -46,7 +46,7 @@ def __new_call_id(self):
             call_id += 1
         return call_id
 
-    def call_for_response(self, body: Body) -> Body:
+    def call_for_response(self, body: Body) -> Body | None:
         ch = Channel[Body]()
         call_id = self.__new_call_id()
         self.calls[call_id] = ch
@@ -99,7 +99,7 @@ def listen(self, context):
                 msg_parts = msg_buf.split(END_SECTION)
 
                 msg: Msg = json.loads(msg_parts[0])
-                #logging.info("[BsRpc] Received data: %s", msg)
+                # logging.info("[BsRpc] Received data: %s", msg)
 
                 body: Body = self.reg.get(msg["type"])(
                     data=msg["body"]

bs_rpc/channel.py

@@ -3,6 +3,7 @@
 
 T = TypeVar("T")
 
+
 class Channel(Generic[T]):
     def __init__(self):
         self.__state: list[T] = []

bs_rpc/channel_generator.py

@@ -5,6 +5,7 @@
 CloseFn = Callable[[], None]
 T = TypeVar("T")
 
+
 class ChannelGenerator(Generic[T]):
     def __init__(self, ch: Channel[T], close_fn: CloseFn | None = None):
         self.__ch: Channel[T] = ch

bs_rpc/msg.py

@@ -4,6 +4,7 @@
 
 ResponseWriter = Callable[["Body"], None]
 
+
 class Method(Enum):
     CALL_FOR_RESPONSE = "call_for_response"
     CALL_FOR_RESPONSES = "call_for_responses"
@@ -20,7 +21,9 @@ def call_for_response(self, context) -> "Body":
         """returns Body"""
         raise Exception(f"call_for_response not implemented for {self.type}")
 
-    def call_for_responses(self, context, response_writer:ResponseWriter) -> None:
+    def call_for_responses(
+        self, context, response_writer: ResponseWriter
+    ) -> None:
         raise Exception(f"call_for_responses not implemented for {self.type}")
 
     def get_type(self):

pokete_classes/multiplayer/remote_fight/figure_wrapper.py

@@ -1,5 +1,5 @@
 import random
-from typing_extensions import Generator
+from typing import Generator
 
 import bs_rpc
 from pokete_classes.context import Context