Initial commit: Video metadata registration service

- Add FastAPI server for video metadata registration - PostgreSQL database models for videos, video_streams, audio_streams, subtitle_streams - Batch registration script for .probe.json files - RESTful API endpoints for CRUD operations - Search functionality by title, artist, codec, resolution
2026-03-11 00:30:31 +08:00
commit b4aa7b96d3
16 changed files with 709 additions and 0 deletions
--- a/app/models/schemas.py
+++ b/app/models/schemas.py
@@ -0,0 +1,103 @@
+from datetime import datetime
+from uuid import UUID
+from typing import Optional, List
+from pydantic import BaseModel, ConfigDict
+
+
+class VideoStreamSchema(BaseModel):
+    model_config = ConfigDict(from_attributes=True)
+
+    stream_index: Optional[int] = None
+    codec_name: Optional[str] = None
+    codec_long_name: Optional[str] = None
+    profile: Optional[str] = None
+    level: Optional[int] = None
+    width: Optional[int] = None
+    height: Optional[int] = None
+    coded_width: Optional[int] = None
+    coded_height: Optional[int] = None
+    aspect_ratio: Optional[str] = None
+    pix_fmt: Optional[str] = None
+    field_order: Optional[str] = None
+    frame_rate: Optional[str] = None
+    start_time: Optional[float] = None
+    duration: Optional[float] = None
+    bit_rate: Optional[int] = None
+    nb_frames: Optional[int] = None
+    color_range: Optional[str] = None
+    color_space: Optional[str] = None
+    has_b_frames: Optional[int] = None
+    sample_aspect_ratio: Optional[str] = None
+
+
+class AudioStreamSchema(BaseModel):
+    model_config = ConfigDict(from_attributes=True)
+
+    stream_index: Optional[int] = None
+    codec_name: Optional[str] = None
+    codec_long_name: Optional[str] = None
+    profile: Optional[str] = None
+    channels: Optional[int] = None
+    channel_layout: Optional[str] = None
+    sample_rate: Optional[int] = None
+    sample_fmt: Optional[str] = None
+    bit_rate: Optional[int] = None
+    duration: Optional[float] = None
+    language: Optional[str] = None
+
+
+class SubtitleStreamSchema(BaseModel):
+    model_config = ConfigDict(from_attributes=True)
+
+    stream_index: Optional[int] = None
+    codec_name: Optional[str] = None
+    language: Optional[str] = None
+
+
+class VideoSchema(BaseModel):
+    model_config = ConfigDict(from_attributes=True)
+
+    id: UUID
+    file_path: str
+    file_name: str
+    file_extension: Optional[str] = None
+    file_size: Optional[int] = None
+    format_name: Optional[str] = None
+    format_long_name: Optional[str] = None
+    duration: Optional[float] = None
+    bit_rate: Optional[int] = None
+    nb_streams: Optional[int] = None
+    start_time: Optional[float] = None
+    title: Optional[str] = None
+    artist: Optional[str] = None
+    description: Optional[str] = None
+    probed_at: Optional[datetime] = None
+    registered_at: Optional[datetime] = None
+    created_at: Optional[datetime] = None
+    updated_at: Optional[datetime] = None
+    video_streams: List[VideoStreamSchema] = []
+    audio_streams: List[AudioStreamSchema] = []
+    subtitle_streams: List[SubtitleStreamSchema] = []
+
+
+class VideoListResponse(BaseModel):
+    total: int
+    videos: List[VideoSchema]
+
+
+class VideoSearchQuery(BaseModel):
+    title: Optional[str] = None
+    artist: Optional[str] = None
+    codec_name: Optional[str] = None
+    min_width: Optional[int] = None
+    max_width: Optional[int] = None
+    min_height: Optional[int] = None
+    max_height: Optional[int] = None
+    format_name: Optional[str] = None
+    skip: int = 0
+    limit: int = 20
+
+
+class RegisterRequest(BaseModel):
+    probe_json_path: str
+    absolute_file_path: str