deltacast-sdi-recorder/backend/app/models.py

81 lines
3.5 KiB
Python

from enum import Enum
from datetime import datetime
from pydantic import BaseModel, Field, ConfigDict
from typing import Optional, List
class CodecType(str, Enum):
"""Supported video codec types"""
PRORES = "prores"
DNXHD = "dnxhd"
UNCOMPRESSED = "uncompressed"
H264 = "h264"
class SRTDestination(BaseModel):
"""A single SRT output destination"""
model_config = ConfigDict(use_enum_values=False)
url: str = Field(..., description="SRT destination URL (srt://host:port)")
label: str = Field(default="", description="Human-readable label for this destination")
enabled: bool = Field(default=True, description="Whether this destination is active")
class RecorderConfig(BaseModel):
"""Configuration for a recorder port"""
model_config = ConfigDict(use_enum_values=False)
port_index: int = Field(..., description="Index of the recorder port (0-based)")
codec: CodecType = Field(..., description="Selected codec for recording")
bitrate: str = Field(default="185M", description="Target bitrate (e.g. 185M, 50M)")
quality_profile: str = Field(
default="hq",
description="Quality profile name (hq, mq, lq)"
)
recording_path: str = Field(
...,
description="Directory path for recording output files"
)
srt_enabled: bool = Field(
default=False, description="Enable SRT streaming output"
)
srt_destination: Optional[str] = Field(
default=None, description="SRT destination address — use srt_destinations for multi"
)
srt_destinations: List[SRTDestination] = Field(
default_factory=list, description="List of SRT output destinations"
)
preview_enabled: bool = Field(
default=True, description="Enable live HLS preview"
)
class PortStatus(BaseModel):
"""Real-time status of a recorder port"""
model_config = ConfigDict(use_enum_values=False)
port_index: int = Field(..., description="Index of the recorder port")
is_recording: bool = Field(..., description="Whether this port is currently recording")
frame_count: int = Field(..., description="Total frames recorded")
fps: float = Field(..., description="Frames per second")
bitrate_mbps: float = Field(..., description="Current bitrate in Mbps")
uptime_seconds: int = Field(..., description="Recording uptime in seconds")
current_file: str = Field(..., description="Current output file path")
codec: CodecType = Field(..., description="Codec being used")
srt_destinations: List[str] = Field(
default_factory=list, description="Active SRT destination URLs"
)
class SCTE35Marker(BaseModel):
"""SCTE-35/SCTE-104 advertisement marker"""
model_config = ConfigDict(use_enum_values=False)
event_id: str = Field(..., description="Unique identifier for this ad break event")
duration_seconds: int = Field(..., description="Duration of the ad break in seconds")
out_of_network: bool = Field(..., description="Whether this is an out-of-network ad")
splice_immediate: bool = Field(..., description="Whether splice should happen immediately")
timestamp: datetime = Field(..., description="Timestamp when marker was created/triggered")
webhook_url: Optional[str] = Field(default=None, description="Webhook URL to notify on ad break")
port_index: Optional[int] = Field(default=None, description="Port index (None = global)")
srt_destination_url: Optional[str] = Field(default=None, description="Target SRT destination (None = all)")