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

dataclass_extensions/registrable.py

@@ -11,6 +11,7 @@ R = TypeVar("R", bound="Registrable")
 @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 +28,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 +51,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 +106,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.0"

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

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

{dataclass_extensions-0.1.0.dist-info → dataclass_extensions-0.2.0.dist-info}/RECORD

@@ -2,12 +2,12 @@ dataclass_extensions/__init__.py,sha256=jpgRh5jjj9jM1tkhTVvB-aDThI2v4tAlq_WpR0-N
 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/registrable.py,sha256=6lpMHnOypZ_CPgTwiYglt0egpmhsFaOUmKC-LOBoR_Y,4425
 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,,
+dataclass_extensions/version.py,sha256=w9Iw7QVvd8lme2wKwEbCo5IgetVjSfFBOOYAcA_Q0Ns,18
+dataclass_extensions-0.2.0.dist-info/licenses/LICENSE,sha256=YvuKOpYh3COIF0yqq-nCMXtpS7mh1GyYvPVlW2j1G-M,11359
+dataclass_extensions-0.2.0.dist-info/METADATA,sha256=QXmnDLRIZD1WwOtOGOVFBv0uj1AXxDcHu6McG_-pvz4,15215
+dataclass_extensions-0.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dataclass_extensions-0.2.0.dist-info/top_level.txt,sha256=njfdeqLtTThNGwqnoKu4JV8J1287xGAeEwyESpoU-k8,21
+dataclass_extensions-0.2.0.dist-info/RECORD,,