Make pix_fmt optional

WyattBlue · Nov 28, 2023 · a357fdf · a357fdf
1 parent 2ca0e98
commit a357fdf
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/auto_editor/ffwrapper.py b/auto_editor/ffwrapper.py
@@ -126,7 +126,7 @@ def pipe(self, cmd: list[str]) -> str:
  return output
 
 
-@dataclass(slots=True)
+@dataclass(slots=True, frozen=True)
 class VideoStream:
  width: int
  height: int
@@ -135,7 +135,7 @@ class VideoStream:
  duration: float
  sar: Fraction
  time_base: Fraction
- pix_fmt: str
+ pix_fmt: str | None
  color_range: str | None
  color_space: str | None
  color_primaries: str | None
@@ -144,7 +144,7 @@ class VideoStream:
  lang: str | None
 
 
-@dataclass(slots=True)
+@dataclass(slots=True, frozen=True)
 class AudioStream:
  codec: str
  samplerate: int
@@ -154,14 +154,14 @@ class AudioStream:
  lang: str | None
 
 
-@dataclass(slots=True)
+@dataclass(slots=True, frozen=True)
 class SubtitleStream:
  codec: str
  ext: str
  lang: str | None
 
 
-@dataclass(slots=True)
+@dataclass(slots=True, frozen=True)
 class FileInfo:
  path: Path
  bitrate: int
@@ -208,7 +208,7 @@ def initFileInfo(path: str, ffmpeg: FFmpeg, log: Log, label: str = "") -> FileIn
 
  for i, v in enumerate(cont.streams.video):
  vdur = 0.0
- if hasattr(v, "duration") and v.duration is not None:
+ if v.duration is not None and v.time_base is not None:
  vdur = float(v.duration * v.time_base)
 
  fps = v.average_rate

diff --git a/auto_editor/subcommands/info.py b/auto_editor/subcommands/info.py
@@ -41,7 +41,7 @@ class VideoJson(TypedDict):
  aspect_ratio: list[int]
  pixel_aspect_ratio: str
  duration: float
- pix_fmt: str
+ pix_fmt: str | None
  color_range: str | None
  color_space: str | None
  color_primaries: str | None