Merge pull request #130 from imnotjames/refactor/xadd-xread-xreadgroup-dict-typing

refactor: updating typing for xreadgroup / xread / xadd

Author: Mikhail Koviazin

valkey/commands/core.py

@@ -24,7 +24,10 @@
 from valkey.exceptions import ConnectionError, DataError, NoScriptError, ValkeyError
 from valkey.typing import (
     AbsExpiryT,
+    AnyEncodableT,
+    AnyFieldT,
     AnyKeyT,
+    AnyStreamIdT,
     BitfieldOffsetT,
     ChannelT,
     CommandsProtocol,
@@ -3508,7 +3511,7 @@ def xack(self, name: KeyT, groupname: GroupT, *ids: StreamIdT) -> ResponseT:
     def xadd(
         self,
         name: KeyT,
-        fields: Dict[FieldT, EncodableT],
+        fields: Mapping[AnyFieldT, AnyEncodableT],
         id: StreamIdT = "*",
         maxlen: Union[int, None] = None,
         approximate: bool = True,
@@ -3933,7 +3936,7 @@ def xrange(
 
     def xread(
         self,
-        streams: Dict[KeyT, StreamIdT],
+        streams: Mapping[AnyKeyT, AnyStreamIdT],
         count: Union[int, None] = None,
         block: Union[int, None] = None,
     ) -> ResponseT:
@@ -3973,7 +3976,7 @@ def xreadgroup(
         self,
         groupname: str,
         consumername: str,
-        streams: Dict[KeyT, StreamIdT],
+        streams: Mapping[AnyKeyT, AnyStreamIdT],
         count: Union[int, None] = None,
         block: Union[int, None] = None,
         noack: bool = False,

valkey/typing.py

@@ -47,6 +47,8 @@
 AnyKeyT = TypeVar("AnyKeyT", bytes, str, memoryview)
 AnyFieldT = TypeVar("AnyFieldT", bytes, str, memoryview)
 AnyChannelT = TypeVar("AnyChannelT", bytes, str, memoryview)
+AnyStreamIdT = TypeVar("AnyStreamIdT", int, bytes, str, memoryview)
+AnyEncodableT = TypeVar("AnyEncodableT", int, float, bytes, str, memoryview)
 
 ExceptionMappingT = Mapping[str, Union[Type[Exception], Mapping[str, Type[Exception]]]]