dataclass-extensions 0.1.0__py3-none-any.whl → 0.2.1__py3-none-any.whl

dataclass_extensions/decode.py

@@ -8,6 +8,7 @@ from datetime import datetime
 from enum import Enum
 from typing import Any, Callable, ClassVar, Type, TypeVar
 
+from .registrable import Registrable
 from .types import *
 
 C = TypeVar("C", bound=Dataclass)
@@ -163,6 +164,10 @@ def _coerce(
             # e.g. TypedDict
             or _safe_issubclass(allowed_type, dict)
         ) and _safe_isinstance(value, dict):
+            if _safe_issubclass(allowed_type, Registrable):
+                type_name = value.get("type", allowed_type._default_type)
+                if type_name is not None:
+                    allowed_type = allowed_type.get_registered_class(type_name)
             type_hints = typing.get_type_hints(allowed_type)
             kwargs = {}
             for k, v in value.items():

dataclass_extensions/registrable.py

@@ -1,16 +1,22 @@
 from __future__ import annotations
 
 import dataclasses
+import sys
 import typing
 from dataclasses import dataclass
 from typing import Callable, ClassVar, Type, TypeVar
 
 R = TypeVar("R", bound="Registrable")
 
+if sys.version_info < (3, 11):
+    # HACK: work-around for https://stackoverflow.com/q/70400639/4151392
+    dataclasses.InitVar.__call__ = lambda *_: None  # type: ignore
+
 
 @dataclass
 class Registrable:
     _registry: ClassVar[dict[str, Type[Registrable]]]
+    _default_type: ClassVar[str | None]
 
     type: dataclasses.InitVar[str | None] = dataclasses.field(
         default=None, kw_only=True, repr=False
@@ -27,16 +33,20 @@ class Registrable:
                     f"Available choices are: {list(cls._registry.keys())}"
                 )
             return super().__new__(cls._registry[type])
+        elif cls._default_type is not None and not hasattr(cls, "registered_name"):
+            return super().__new__(cls._registry[cls._default_type])
         else:
             return super().__new__(cls)
 
     def __init_subclass__(cls, **kwargs):
         super().__init_subclass__(**kwargs)
-        if not hasattr(cls, "_choice_registry"):
+        if not hasattr(cls, "_registry"):
             cls._registry = {}
+        if not hasattr(cls, "_default_type"):
+            cls._default_type = None
 
     @classmethod
-    def register(cls, name: str) -> Callable[[Type[R]], Type[R]]:
+    def register(cls, name: str, default: bool = False) -> Callable[[Type[R]], Type[R]]:
         def register_subclass(subclass: Type[R]) -> Type[R]:
             if not issubclass(subclass, cls):
                 raise TypeError(
@@ -46,6 +56,13 @@ class Registrable:
                 raise TypeError(
                     f"class {subclass.__name__} must be a dataclass in order to register it"
                 )
+            if default:
+                if cls._default_type is not None:
+                    raise ValueError(
+                        f"A default implementation for {cls.__name__} has already been registered"
+                    )
+                else:
+                    cls._default_type = name
 
             fields = [
                 (f.name, f.type, f) for f in dataclasses.fields(subclass) if f.name != "type"  # type: ignore
@@ -94,3 +111,9 @@ class Registrable:
     @classmethod
     def get_registered_names(cls) -> list[str]:
         return list(cls._registry.keys())
+
+    @classmethod
+    def get_default(cls: Type[R]) -> Type[R]:
+        if cls._default_type is None:
+            raise ValueError(f"A default implementation of {cls.__name__} has not been registered")
+        return cls.get_registered_class(cls._default_type)

dataclass_extensions/version.py

@@ -1 +1 @@
-VERSION = "0.1.0"
+VERSION = "0.2.1"

{dataclass_extensions-0.1.0.dist-info → dataclass_extensions-0.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dataclass-extensions
-Version: 0.1.0
+Version: 0.2.1
 Summary: Additional functionality for Python dataclasses
 Author-email: Pete Walsh <epwalsh10@gmail.com>
 License:                                  Apache License

dataclass_extensions-0.2.1.dist-info/RECORD

@@ -0,0 +1,13 @@
+dataclass_extensions/__init__.py,sha256=jpgRh5jjj9jM1tkhTVvB-aDThI2v4tAlq_WpR0-NOSI,255
+dataclass_extensions/decode.py,sha256=baPXLrWznJRA-JxPpSDFLB0y4OCvpnu1qNY13gIjnSo,6622
+dataclass_extensions/encode.py,sha256=rctJHAwlSnjOw0rTGbHRWigb2BT4DI3FYGiDzAjDmEQ,3233
+dataclass_extensions/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+dataclass_extensions/registrable.py,sha256=TWdJ0vLkOTsw5Zm8uOk7Ueh6ZODn-q_2jht1uWRkpzE,4608
+dataclass_extensions/types.py,sha256=UwSV0MKjFDaK4dlgm72vJHHr6-8QNe8jMhbTZGOMIJM,247
+dataclass_extensions/utils.py,sha256=8cigzY9JDEWexWXlcMxXyf27S50-1ZPMNmn4zUSrc6k,935
+dataclass_extensions/version.py,sha256=Vdi6OffDRorPQeWjvXo2MPbidl7CNworxvziT78bjl0,18
+dataclass_extensions-0.2.1.dist-info/licenses/LICENSE,sha256=YvuKOpYh3COIF0yqq-nCMXtpS7mh1GyYvPVlW2j1G-M,11359
+dataclass_extensions-0.2.1.dist-info/METADATA,sha256=HfEEj5f6qnWVrx7LBzosDJ2dz7fkiQBQ_ktO_UeY81w,15215
+dataclass_extensions-0.2.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dataclass_extensions-0.2.1.dist-info/top_level.txt,sha256=njfdeqLtTThNGwqnoKu4JV8J1287xGAeEwyESpoU-k8,21
+dataclass_extensions-0.2.1.dist-info/RECORD,,

dataclass_extensions-0.1.0.dist-info/RECORD

@@ -1,13 +0,0 @@
-dataclass_extensions/__init__.py,sha256=jpgRh5jjj9jM1tkhTVvB-aDThI2v4tAlq_WpR0-NOSI,255
-dataclass_extensions/decode.py,sha256=dstNb-RU8i3jGoTlo165UlSbyG3MhAgbc-O54mdlxqo,6329
-dataclass_extensions/encode.py,sha256=rctJHAwlSnjOw0rTGbHRWigb2BT4DI3FYGiDzAjDmEQ,3233
-dataclass_extensions/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-dataclass_extensions/registrable.py,sha256=QJN-QSzxC-vxm1w_wqzRkhRr-GcPFAlIGfbbw5m1o68,3572
-dataclass_extensions/types.py,sha256=UwSV0MKjFDaK4dlgm72vJHHr6-8QNe8jMhbTZGOMIJM,247
-dataclass_extensions/utils.py,sha256=8cigzY9JDEWexWXlcMxXyf27S50-1ZPMNmn4zUSrc6k,935
-dataclass_extensions/version.py,sha256=CpXi3jGlx23RvRyU7iytOMZrnspdWw4yofS8lpP1AJU,18
-dataclass_extensions-0.1.0.dist-info/licenses/LICENSE,sha256=YvuKOpYh3COIF0yqq-nCMXtpS7mh1GyYvPVlW2j1G-M,11359
-dataclass_extensions-0.1.0.dist-info/METADATA,sha256=UwZk8Z5EKDiN_rdxjsDqjYcNwQHiT7qpIVVPJxUsfqY,15215
-dataclass_extensions-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dataclass_extensions-0.1.0.dist-info/top_level.txt,sha256=njfdeqLtTThNGwqnoKu4JV8J1287xGAeEwyESpoU-k8,21
-dataclass_extensions-0.1.0.dist-info/RECORD,,