typed-ffmpeg-compatible 2.6.4__py3-none-any.whl → 2.7.2__py3-none-any.whl

typed_ffmpeg/common/schema.py

@@ -12,7 +12,10 @@ from dataclasses import dataclass
 from enum import Enum
 from typing import Literal
 
+from .serialize import Serializable, serializable
 
+
+@serializable
 class StreamType(str, Enum):
     """
     Enumeration of possible stream types in FFmpeg.
@@ -198,7 +201,7 @@ class FFMpegFilterDef:
 
 
 @dataclass(frozen=True, kw_only=True)
-class FFMpegFilter:
+class FFMpegFilter(Serializable):
     """
     Comprehensive representation of an FFmpeg filter with all its metadata.
 

typed_ffmpeg/common/serialize.py

@@ -9,7 +9,6 @@ saved to disk and loaded back.
 
 from __future__ import annotations
 
-import importlib
 import json
 from dataclasses import fields, is_dataclass
 from enum import Enum
@@ -19,43 +18,62 @@ from typing import Any
 
 from ..utils.forzendict import FrozenDict
 
+CLASS_REGISTRY: dict[str, type[Serializable | Enum]] = {}
+"""
+A registry of classes that have been loaded, and can be deserialized.
+"""
+
+
+def serializable(
+    cls: type[Serializable] | type[Enum],
+) -> type[Serializable] | type[Enum]:
+    """
+    Register a class with the serialization system.
+    """
+    assert cls.__name__ not in CLASS_REGISTRY, (
+        f"Class {cls.__name__} already registered"
+    )
+    CLASS_REGISTRY[cls.__name__] = cls
+
+    return cls
+
 
-def load_class(path: str, strict: bool = True) -> Any:
+class Serializable:
     """
-    Load a class from a string path.
+    A base class for all serializable classes.
+    """
+
+    def __init_subclass__(cls, **kwargs: Any) -> None:
+        super().__init_subclass__(**kwargs)
+        serializable(cls)
+
+
+def load_class(name: str) -> type[Serializable] | type[Enum]:
+    """
+    Load a class from its name.
 
-    This function dynamically imports a class based on its fully qualified
-    path (e.g., 'ffmpeg.dag.nodes.FilterNode'). It's used during deserialization
-    to reconstruct objects from their class names.
+    This function looks up a class by its name in the CLASS_REGISTRY. It's used during
+    deserialization to reconstruct objects from their class names.
 
     Args:
-        path: The fully qualified path to the class (module.submodule.ClassName)
-        strict: If True, only allow loading classes from the ffmpeg package
-               as a security measure
+        name: The simple class name (e.g., 'FilterNode')
 
     Returns:
         The class object that can be instantiated
 
     Raises:
-        AssertionError: If strict is True and the path doesn't start with 'ffmpeg.'
-        ImportError: If the module or class cannot be found
+        AssertionError: If the class name is not found in the registry
 
     Example:
         ```python
         # Load the FilterNode class
-        FilterNode = load_class('ffmpeg.dag.nodes.FilterNode')
+        FilterNode = load_class('FilterNode')
         # Create an instance
         node = FilterNode(name='scale', ...)
         ```
     """
-    if strict:
-        assert path.startswith("ffmpeg."), (
-            f"Only support loading class from ffmpeg package: {path}"
-        )
-
-    module_path, class_name = path.rsplit(".", 1)
-    module = importlib.import_module(module_path)
-    return getattr(module, class_name)
+    assert name in CLASS_REGISTRY, f"Class {name} not registered"
+    return CLASS_REGISTRY[name]
 
 
 def frozen(v: Any) -> Any:
@@ -110,7 +128,7 @@ def object_hook(obj: Any, strict: bool = True) -> Any:
         ```python
         # A JSON object with class information
         json_obj = {
-            "__class__": "ffmpeg.dag.nodes.FilterNode",
+            "__class__": "FilterNode",
             "name": "scale",
             "kwargs": {"width": 1280, "height": 720},
         }
@@ -119,7 +137,7 @@ def object_hook(obj: Any, strict: bool = True) -> Any:
     """
     if isinstance(obj, dict):
         if obj.get("__class__"):
-            cls = load_class(obj.pop("__class__"), strict=strict)
+            cls = load_class(obj.pop("__class__"))
 
             if is_dataclass(cls):
                 # NOTE: in our application, the dataclass is always frozen
@@ -148,7 +166,7 @@ def loads(raw: str, strict: bool = True) -> Any:
     Example:
         ```python
         # Deserialize a filter graph from JSON
-        json_str = '{"__class__": "ffmpeg.dag.nodes.FilterNode", "name": "scale", ...}'
+        json_str = '{"__class__": "FilterNode", "name": "scale", ...}'
         filter_node = loads(json_str)
         # filter_node is now a FilterNode instance
         ```
@@ -192,7 +210,7 @@ def to_dict_with_class_info(instance: Any) -> Any:
         return str(instance)
     elif is_dataclass(instance):
         return {
-            "__class__": f"{instance.__class__.__module__}.{instance.__class__.__name__}",
+            "__class__": instance.__class__.__name__,
             **{
                 k.name: to_dict_with_class_info(getattr(instance, k.name))
                 for k in fields(instance)
@@ -200,7 +218,7 @@ def to_dict_with_class_info(instance: Any) -> Any:
         }
     elif isinstance(instance, Enum):
         return {
-            "__class__": f"{instance.__class__.__module__}.{instance.__class__.__name__}",
+            "__class__": instance.__class__.__name__,
             "value": instance.value,
         }
     return instance

typed_ffmpeg/dag/nodes.py

@@ -665,10 +665,7 @@ class OutputNode(Node):
         # !handle mapping
         commands = []
 
-        if context and (
-            any(isinstance(k.node, FilterNode) for k in self.inputs)
-            or len([k for k in context.all_nodes if isinstance(k, OutputNode)]) > 1
-        ):
+        if context:
             for input in self.inputs:
                 if isinstance(input.node, InputNode):
                     commands += ["-map", input.label(context)]

typed_ffmpeg/dag/schema.py

@@ -5,6 +5,7 @@ from dataclasses import dataclass, replace
 from functools import cached_property
 from typing import TYPE_CHECKING, Literal
 
+from ..common.serialize import Serializable
 from ..utils.forzendict import FrozenDict
 from ..utils.lazy_eval.schema import LazyValue
 from .utils import is_dag
@@ -14,7 +15,7 @@ if TYPE_CHECKING:
 
 
 @dataclass(frozen=True, kw_only=True)
-class HashableBaseModel:
+class HashableBaseModel(Serializable):
     """
     A base class for hashable dataclasses.
     """

{typed_ffmpeg_compatible-2.6.4.dist-info → typed_ffmpeg_compatible-2.7.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: typed-ffmpeg-compatible
-Version: 2.6.4
+Version: 2.7.2
 Summary: Modern Python FFmpeg wrappers offer comprehensive support for complex filters, complete with detailed typing and documentation.
 Home-page: https://livingbio.github.io/typed-ffmpeg/
 License: MIT

{typed_ffmpeg_compatible-2.6.4.dist-info → typed_ffmpeg_compatible-2.7.2.dist-info}/RECORD

@@ -1,8 +1,8 @@
 typed_ffmpeg/__init__.py,sha256=NL_jH6nUb5pF8pMt3S51243qBTUvZdN-9sJpeuiznRE,1532
 typed_ffmpeg/base.py,sha256=C5Tqbx2I0c-09D7aXKZoGkspu-lAAeAhuOns5zr3PXQ,6304
 typed_ffmpeg/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-typed_ffmpeg/common/schema.py,sha256=NSr1GnREvAhdshlBigRSUu12NTleOwhspJrbVCpGgNo,19468
-typed_ffmpeg/common/serialize.py,sha256=Be0SgrsfD76kWZYT1IKFKQItuOgGLYWJITjDXGvbzcs,7687
+typed_ffmpeg/common/schema.py,sha256=Gfq_C8T4F0Ie6E9vzmWMn-mzjN0tOfTK0mKHugW5vEs,19547
+typed_ffmpeg/common/serialize.py,sha256=QH7UVPCXAHrqbAMK11DzFfy03XgFEHXWOrCVUJxBxR0,7764
 typed_ffmpeg/dag/__init__.py,sha256=qAApSNqjbZ1DtUaV5bSku9RwG7MpMPa1HJO764cSBt4,849
 typed_ffmpeg/dag/compile.py,sha256=GXMTFE1vqRDI-nuIZPtJwXQhCc61uCIGh5u4OBRp8gg,3129
 typed_ffmpeg/dag/context.py,sha256=pv77lltW6yk-PRYlZ0MC6FICVxYLuyxQAS9Vx5vko0s,11691
@@ -14,8 +14,8 @@ typed_ffmpeg/dag/io/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSu
 typed_ffmpeg/dag/io/_input.py,sha256=KRLTSQPEfmgPcPEAJdeWRHZhNsClaJCB9Ac6czMOrmE,7214
 typed_ffmpeg/dag/io/_output.py,sha256=_no6ffAOABznbLNTki8CYr7pvr4Sa0LweRfn38-cszs,12470
 typed_ffmpeg/dag/io/output_args.py,sha256=SThIhZh9PXs2m6Fz5JsSy8oS-Te7GM_oz7HRuZo0-eI,13901
-typed_ffmpeg/dag/nodes.py,sha256=yJcFjN85aFbk_4SKI7pbe-JkTSUJbfrpSXpqYiGbWGs,29133
-typed_ffmpeg/dag/schema.py,sha256=gKW1eMMiVHvsbFnw4jlh22QhnSDgk_AJCBWvlF73ZZs,6040
+typed_ffmpeg/dag/nodes.py,sha256=rV6pGPOo6hCuI_4Iipq8pwEz32U2iz7vITaG4C6dmCs,28964
+typed_ffmpeg/dag/schema.py,sha256=51t38NHgztXp5RAPKgSKqFoSQaU5IsnhIxuxnx2Udl4,6098
 typed_ffmpeg/dag/utils.py,sha256=hydh7_kjpOCw8WEGhXMxIXR4Ek-3DeoOt6esInuK2Xw,1941
 typed_ffmpeg/dag/validate.py,sha256=tWswVibP_BFNOiWX0LgnfMrg3TBm7QIt060Ofd9ychE,9569
 typed_ffmpeg/exceptions.py,sha256=D4SID6WOwkjVV8O8mAjrEDHWn-8BRDnK_jteaDof1SY,2474
@@ -41,8 +41,8 @@ typed_ffmpeg/utils/run.py,sha256=mSoAdcvD-InldqkRgWNc8iXKgJJoEMAOE4PL2gVmtqw,217
 typed_ffmpeg/utils/snapshot.py,sha256=mKILRm6qiQV2egaD-70MSUEl-DFoLD5w_v9GZIequI4,2181
 typed_ffmpeg/utils/typing.py,sha256=DBQn_gCF8C_DTwsfMHeCgfnNUROwAjlIcHrQ7lNDOoE,1187
 typed_ffmpeg/utils/view.py,sha256=4DJmyGIkUeE6-V0azudDCWm608mRdeRhTZHMgnew9Kw,3409
-typed_ffmpeg_compatible-2.6.4.dist-info/LICENSE,sha256=8Aaya5i_09Cou2i3QMxTwz6uHGzi_fGA4uhkco07-A4,1066
-typed_ffmpeg_compatible-2.6.4.dist-info/METADATA,sha256=AB_nBB0d5PwrLzSZ29cy_Xc01YKTAM8tWj9X8BEnIJs,7249
-typed_ffmpeg_compatible-2.6.4.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-typed_ffmpeg_compatible-2.6.4.dist-info/entry_points.txt,sha256=KfZmNsM16GT_lF1otASIN6E3i6xXHXoB1gMeEdlptjA,44
-typed_ffmpeg_compatible-2.6.4.dist-info/RECORD,,
+typed_ffmpeg_compatible-2.7.2.dist-info/LICENSE,sha256=8Aaya5i_09Cou2i3QMxTwz6uHGzi_fGA4uhkco07-A4,1066
+typed_ffmpeg_compatible-2.7.2.dist-info/METADATA,sha256=djaV8dOPAbYJax4MQV1G0kDMuZXwlOxNbCpxzbgD3pw,7249
+typed_ffmpeg_compatible-2.7.2.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+typed_ffmpeg_compatible-2.7.2.dist-info/entry_points.txt,sha256=KfZmNsM16GT_lF1otASIN6E3i6xXHXoB1gMeEdlptjA,44
+typed_ffmpeg_compatible-2.7.2.dist-info/RECORD,,