typed-ffmpeg-compatible 2.5.0__py3-none-any.whl → 2.6.0__py3-none-any.whl

typed_ffmpeg/__init__.py

@@ -2,6 +2,7 @@ from . import dag, filters
 from .base import afilter, filter_multi_output, input, merge_outputs, output, vfilter
 from .dag import Stream
 from .exceptions import FFMpegExecuteError, FFMpegTypeError, FFMpegValueError
+from .info import get_codecs, get_decoders, get_encoders
 from .probe import probe
 from .streams import AudioStream, AVStream, VideoStream
 
@@ -22,4 +23,7 @@ __all__ = [
     "afilter",
     "filter_multi_output",
     "dag",
+    "get_codecs",
+    "get_decoders",
+    "get_encoders",
 ]

typed_ffmpeg/dag/nodes.py

@@ -351,7 +351,8 @@ class InputNode(Node):
             if isinstance(value, bool):
                 if value is True:
                     commands += [f"-{key}"]
-                # NOTE: the -nooption is not supported
+                elif value is False:
+                    commands += [f"-no{key}"]
             else:
                 commands += [f"-{key}", str(value)]
         commands += ["-i", self.filename]
@@ -398,7 +399,8 @@ class OutputNode(Node):
             if isinstance(value, bool):
                 if value is True:
                     commands += [f"-{key}"]
-                # NOTE: the -nooption is not supported
+                elif value is False:
+                    commands += [f"-no{key}"]
             else:
                 commands += [f"-{key}", str(value)]
         commands += [self.filename]
@@ -453,7 +455,6 @@ class GlobalNode(Node):
                     commands += [f"-{key}"]
                 elif value is False:
                     commands += [f"-no{key}"]
-                # NOTE: the -no{key} format since not really for global options
             else:
                 commands += [f"-{key}", str(value)]
         return commands

typed_ffmpeg/info.py

@@ -0,0 +1,161 @@
+import logging
+import subprocess
+from dataclasses import dataclass
+from enum import Flag, auto
+
+from .exceptions import FFMpegExecuteError
+from .utils.run import command_line
+
+logger = logging.getLogger(__name__)
+
+
+class CodecFlags(Flag):
+    video = auto()
+    audio = auto()
+    subtitle = auto()
+    frame_level_multithreading = auto()
+    slice_level_multithreading = auto()
+    experimental = auto()
+    draw_horiz_band = auto()
+    direct_rendering_method_1 = auto()
+
+
+@dataclass(frozen=True)
+class Codec:
+    name: str
+    flags: CodecFlags
+    description: str
+
+
+def parse_codec_flags(flags: str) -> CodecFlags:
+    flags_enum = CodecFlags(0)
+    if flags[0] == "V":
+        flags_enum |= CodecFlags.video
+    if flags[0] == "A":
+        flags_enum |= CodecFlags.audio
+    if flags[0] == "S":
+        flags_enum |= CodecFlags.subtitle
+    if flags[1] == "F":
+        flags_enum |= CodecFlags.frame_level_multithreading
+    if flags[2] == "S":
+        flags_enum |= CodecFlags.slice_level_multithreading
+    if flags[3] == "X":
+        flags_enum |= CodecFlags.experimental
+    if flags[4] == "B":
+        flags_enum |= CodecFlags.draw_horiz_band
+    if flags[5] == "D":
+        flags_enum |= CodecFlags.direct_rendering_method_1
+    return flags_enum
+
+
+def get_codecs() -> tuple[Codec, ...]:
+    args = ["ffmpeg", "-hide_banner", "-codecs"]
+    logger.info("Running ffmpeg command: %s", command_line(args))
+    p = subprocess.Popen(args, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    out, err = p.communicate()
+
+    retcode = p.poll()
+    if p.returncode != 0:
+        raise FFMpegExecuteError(retcode=retcode, cmd=command_line(args), stdout=out, stderr=err)
+
+    codecs = out.decode("utf-8")
+    codecs_lines = codecs.strip().split("\n")
+    # Skip header lines until we find the separator
+    for i, line in enumerate(codecs_lines):
+        if line.startswith(" ------"):
+            codecs_lines = codecs_lines[i + 1 :]
+            break
+    return tuple(
+        Codec(
+            name=parts[1],
+            flags=parse_codec_flags(parts[0]),
+            description=parts[2],
+        )
+        for line in codecs_lines
+        for parts in [line.split(None, 3)]
+    )
+
+
+class CoderFlags(Flag):
+    video = auto()
+    audio = auto()
+    subtitle = auto()
+    frame_level_multithreading = auto()
+    slice_level_multithreading = auto()
+    experimental = auto()
+    draw_horiz_band = auto()
+    direct_rendering_method_1 = auto()
+
+
+def parse_coder_flags(flags: str) -> CoderFlags:
+    flags_enum = CoderFlags(0)
+    if flags[0] == "V":
+        flags_enum |= CoderFlags.video
+    if flags[0] == "A":
+        flags_enum |= CoderFlags.audio
+    if flags[0] == "S":
+        flags_enum |= CoderFlags.subtitle
+    if flags[1] == "F":
+        flags_enum |= CoderFlags.frame_level_multithreading
+    if flags[2] == "S":
+        flags_enum |= CoderFlags.slice_level_multithreading
+    if flags[3] == "X":
+        flags_enum |= CoderFlags.experimental
+    if flags[4] == "B":
+        flags_enum |= CoderFlags.draw_horiz_band
+    if flags[5] == "D":
+        flags_enum |= CoderFlags.direct_rendering_method_1
+    return flags_enum
+
+
+@dataclass
+class Coder:
+    name: str
+    flags: CoderFlags
+    description: str
+
+
+def get_coders(codes: str) -> tuple[Coder, ...]:
+    codecs_lines = codes.strip().split("\n")
+    # Skip header lines until we find the separator
+    for i, line in enumerate(codecs_lines):
+        if line.startswith(" ------"):
+            codecs_lines = codecs_lines[i + 1 :]
+            break
+    return tuple(
+        Coder(
+            name=parts[1],
+            flags=parse_coder_flags(parts[0]),
+            description=parts[2],
+        )
+        for line in codecs_lines
+        for parts in [line.split(None, 3)]
+    )
+
+
+def get_decoders() -> tuple[Coder, ...]:
+    args = ["ffmpeg", "-hide_banner", "-decoders"]
+    logger.info("Running ffmpeg command: %s", command_line(args))
+    p = subprocess.Popen(args, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    out, err = p.communicate()
+
+    retcode = p.poll()
+    if p.returncode != 0:
+        raise FFMpegExecuteError(retcode=retcode, cmd=command_line(args), stdout=out, stderr=err)
+
+    decoders = out.decode("utf-8")
+    return get_coders(decoders)
+
+
+def get_encoders() -> tuple[Coder, ...]:
+    args = ["ffmpeg", "-hide_banner", "-encoders"]
+    logger.info("Running ffmpeg command: %s", command_line(args))
+    p = subprocess.Popen(args, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    out, err = p.communicate()
+
+    retcode = p.poll()
+    if p.returncode != 0:
+        raise FFMpegExecuteError(retcode=retcode, cmd=command_line(args), stdout=out, stderr=err)
+
+    encoders = out.decode("utf-8")
+    return get_coders(encoders)

typed_ffmpeg/utils/snapshot.py

@@ -1,12 +1,13 @@
+from dataclasses import asdict
 from typing import Optional
 
-from syrupy.extensions.image import PNGImageSnapshotExtension
+from syrupy.extensions.json import JSONSnapshotExtension
 from syrupy.types import PropertyFilter, PropertyMatcher, SerializableData, SerializedData
 
 from ..dag.schema import Stream
 
 
-class DAGSnapshotExtenstion(PNGImageSnapshotExtension):
+class DAGSnapshotExtenstion(JSONSnapshotExtension):
     """
     A snapshot extension for the DAG. This extension is used to serialize and match the DAG.
     """
@@ -20,7 +21,5 @@ class DAGSnapshotExtenstion(PNGImageSnapshotExtension):
         matcher: Optional["PropertyMatcher"] = None,
     ) -> "SerializedData":
         stream = Stream(node=data)
-        graph_path = stream.view()
 
-        with open(graph_path, "rb") as ifile:
-            return super().serialize(ifile.read())
+        return super().serialize(asdict(stream))

{typed_ffmpeg_compatible-2.5.0.dist-info → typed_ffmpeg_compatible-2.6.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: typed-ffmpeg-compatible
-Version: 2.5.0
+Version: 2.6.0
 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.5.0.dist-info → typed_ffmpeg_compatible-2.6.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-typed_ffmpeg/__init__.py,sha256=WnVr15iPyHWL5ZXtAKah50Jf-rR4QEA8qisBWhMY3KI,596
+typed_ffmpeg/__init__.py,sha256=122K8EWO5F-1re6YjI_VrTo8tlGGkNbT9Dqlppt5bUI,711
 typed_ffmpeg/base.py,sha256=aa0-lPC_P_P61nnQRK0VR9a-b-XhNCPJUdtYWkQA6yE,3169
 typed_ffmpeg/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 typed_ffmpeg/common/schema.py,sha256=9oiURyju0YDYPdhg0afrSQ1nOyAK-0v1xSkA6S7Psok,9460
@@ -14,12 +14,13 @@ typed_ffmpeg/dag/io/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSu
 typed_ffmpeg/dag/io/_input.py,sha256=xz5MWVmQJNd2ryFoDNy094hIfj2RCngIxu12AwPwnUU,7162
 typed_ffmpeg/dag/io/_output.py,sha256=OCY7XxEdEOfvVqjEsXq5SWPt02Kkx8GkYphIWiAkLts,12415
 typed_ffmpeg/dag/io/output_args.py,sha256=tilJd5xssVahZP88J8A2urZ3PHJG80Lj6kv1QS8WZzE,13875
-typed_ffmpeg/dag/nodes.py,sha256=hdgwepV3U-GU2rHCgWor2pYuNyCJcCL2mJtuOWgSGuw,14449
+typed_ffmpeg/dag/nodes.py,sha256=BJ-fQfiwJrz7i0F5CQ-ZFVyb014_h7XRTOUtcxRSAuU,14425
 typed_ffmpeg/dag/schema.py,sha256=WfY6gJtjnOxJV-Z5hdBcrKUYzpbHv4dT7mYs0Zv9lS0,5926
 typed_ffmpeg/dag/utils.py,sha256=ooGrCOnsvRjjUMhC4JSCc3T2FRL1JLz0CeVSiE9f1hI,1141
 typed_ffmpeg/dag/validate.py,sha256=smkTzh4ex6Omqcm5iUWeKX5-B8c4e0oTTVOGWSl9i5I,5786
 typed_ffmpeg/exceptions.py,sha256=Zz-lKGildItY9a3dD1NHqx5b1U9hO6splbtA8JGce0U,968
 typed_ffmpeg/filters.py,sha256=OEIG8nWGMHBenm5EJmng6nHYCL-_t5KVun5quF4GhLE,107702
+typed_ffmpeg/info.py,sha256=vgJo0QVxxZeXIe3GUjo060MQ7ligzlNmxbRXgogHsVw,4724
 typed_ffmpeg/probe.py,sha256=srLG1HIK4V4bX3GdgETiLrkc7LJuTAoG7B36L6htteQ,1408
 typed_ffmpeg/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 typed_ffmpeg/schema.py,sha256=5ITK1t7g_OLZRyuN69DI920wERsxZZxrV3mV57tql7A,534
@@ -35,11 +36,11 @@ typed_ffmpeg/utils/lazy_eval/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NM
 typed_ffmpeg/utils/lazy_eval/operator.py,sha256=AFUUdse1M5QvyO6i78xO3e3hxG2lwITSmgywsQbDWNs,2787
 typed_ffmpeg/utils/lazy_eval/schema.py,sha256=JCiwxNElHzeSm5ehfgpwgV1fNbZ7cuQzs2u605zEw5Q,4840
 typed_ffmpeg/utils/run.py,sha256=OotkfIFCU9KD_SzhigCtUeqqvcEWlFiMUsBcxms1k3M,709
-typed_ffmpeg/utils/snapshot.py,sha256=OTEH0Eyzam_r1i5oIC0nKD3hGpVmq4aV5TB2qd5hKuM,813
+typed_ffmpeg/utils/snapshot.py,sha256=0kWMjuwxhwaMtiXwHb4qH_ejsuAYpJyj9gHcLE2Rxps,752
 typed_ffmpeg/utils/typing.py,sha256=sx4IN38NGQxgMs5GOtPouM02TcHINxvTreXPUbAbGRc,316
 typed_ffmpeg/utils/view.py,sha256=rS8gC_uwlUs9Y5UQFj1ur2NjVbZXkZGdGHND0btJpuM,1705
-typed_ffmpeg_compatible-2.5.0.dist-info/LICENSE,sha256=8Aaya5i_09Cou2i3QMxTwz6uHGzi_fGA4uhkco07-A4,1066
-typed_ffmpeg_compatible-2.5.0.dist-info/METADATA,sha256=bhXG9rmI7iIzYn0yRU6JVeP1k3FTxHsa_AznmKgSe9E,7249
-typed_ffmpeg_compatible-2.5.0.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-typed_ffmpeg_compatible-2.5.0.dist-info/entry_points.txt,sha256=KfZmNsM16GT_lF1otASIN6E3i6xXHXoB1gMeEdlptjA,44
-typed_ffmpeg_compatible-2.5.0.dist-info/RECORD,,
+typed_ffmpeg_compatible-2.6.0.dist-info/LICENSE,sha256=8Aaya5i_09Cou2i3QMxTwz6uHGzi_fGA4uhkco07-A4,1066
+typed_ffmpeg_compatible-2.6.0.dist-info/METADATA,sha256=08agGVp8FMLf-GrIHbmlRI_HyBx_EbEnXMMM2gN8dN0,7249
+typed_ffmpeg_compatible-2.6.0.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+typed_ffmpeg_compatible-2.6.0.dist-info/entry_points.txt,sha256=KfZmNsM16GT_lF1otASIN6E3i6xXHXoB1gMeEdlptjA,44
+typed_ffmpeg_compatible-2.6.0.dist-info/RECORD,,