fix: ValueCodec as generic class

i2mint · Nov 17, 2023 · 182ef2c · 182ef2c
1 parent 7773e98
commit 182ef2c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/dol/kv_codecs.py b/dol/kv_codecs.py
@@ -265,7 +265,7 @@ def is_value_codec(attr_val):
 
     # Note: Note clear if escaping or unescaping is the encoder or decoder here
     # I have never had the need for stores using it, so will omit for now
-    # html: Codec[str, str] = value_wrap(html.unescape, html.escape)
+    # html: ValueCodec[str, str] = value_wrap(html.unescape, html.escape)
 
     # Compression
     zipfile: ValueCodec[bytes, bytes] = value_wrap(zip_compress, zip_decompress)
@@ -284,7 +284,7 @@ def is_value_codec(attr_val):
     )
 
     # Any is really xml.etree.ElementTree.Element, but didn't want to import
-    xml_etree: ValueCodec[Any, bytes] = value_wrap(_xml_tree_encode, _xml_tree_decode)
+    xml_etree: Codec[Any, bytes] = value_wrap(_xml_tree_encode, _xml_tree_decode)
 
 
 from dol.paths import KeyTemplate

diff --git a/dol/trans.py b/dol/trans.py
@@ -3028,11 +3028,11 @@ def __add__(self, other):
         )
 
 
-class ValueCodec(Codec):
+class ValueCodec(Generic[DecodedType, EncodedType], Codec[DecodedType, EncodedType]):
     def __call__(self, obj):
         return wrap_kvs(obj, data_of_obj=self.encoder, obj_of_data=self.decoder)
 
 
-class KeyCodec(Codec):
+class KeyCodec(Generic[DecodedType, EncodedType], Codec[DecodedType, EncodedType]):
     def __call__(self, obj):
-        return wrap_kvs(obj, id_of_key=self.encoder, key_of_id=self.decoder)
+        return wrap_kvs(obj, id_of_key=self.encoder, key_of_id=self.decoder)