Flatten Bolt-Compress to repo root

bolt_compress/models.py

@@ -0,0 +1,103 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+from enum import Enum
+from pathlib import Path
+from typing import Optional
+
+
+class MediaKind(str, Enum):
+    IMAGE = "image"
+    VIDEO = "video"
+
+
+IMAGE_EXTENSIONS = {
+    ".jpg",
+    ".jpeg",
+    ".png",
+    ".webp",
+    ".bmp",
+    ".tif",
+    ".tiff",
+}
+
+VIDEO_EXTENSIONS = {
+    ".mp4",
+    ".mov",
+    ".mkv",
+    ".avi",
+    ".webm",
+    ".m4v",
+    ".mpg",
+    ".mpeg",
+    ".wmv",
+    ".flv",
+    ".ts",
+    ".3gp",
+    ".mts",
+    ".m2ts",
+}
+
+SUPPORTED_EXTENSIONS = IMAGE_EXTENSIONS | VIDEO_EXTENSIONS
+
+
+@dataclass(frozen=True)
+class MediaInfo:
+    path: Path
+    kind: MediaKind
+    size: int
+    width: int
+    height: int
+    format_name: str
+    duration: float = 0.0
+    codec: str = ""
+
+
+@dataclass
+class MediaItem:
+    item_id: str
+    info: MediaInfo
+    output_path: Optional[Path] = None
+    output_size: int = 0
+    message: str = ""
+
+    @property
+    def path(self) -> Path:
+        return self.info.path
+
+    @property
+    def kind(self) -> MediaKind:
+        return self.info.kind
+
+    @property
+    def size(self) -> int:
+        return self.info.size
+
+
+@dataclass
+class ImageOptions:
+    output_format: str = "original"  # original | jpeg | webp | png
+    quality: int = 82
+    resize_mode: str = "original"  # original | fit | percent
+    max_width: int = 1920
+    max_height: int = 1080
+    scale_percent: int = 80
+    keep_metadata: bool = False
+
+
+@dataclass
+class VideoOptions:
+    codec: str = "h264"  # h264 | h265 | vp9 | av1
+    quality: int = 68
+    speed: str = "balanced"  # fast | balanced | best
+    resolution: int = 0  # 0 = original, otherwise target short edge
+    audio_bitrate: int = 128  # 0 = mute
+    keep_metadata: bool = False
+
+
+def default_image_options() -> ImageOptions:
+    return ImageOptions()
+
+
+def default_video_options() -> VideoOptions:
+    return VideoOptions()