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/video.py
+++ b/app/models/video.py
@@ -0,0 +1,116 @@
+import uuid
+from datetime import datetime
+from sqlalchemy import (
+    Column,
+    String,
+    Integer,
+    BigInteger,
+    Float,
+    Text,
+    ForeignKey,
+    DateTime,
+)
+from sqlalchemy.dialects.postgresql import UUID
+from sqlalchemy.orm import relationship
+from app.database import Base
+
+
+class Video(Base):
+    __tablename__ = "videos"
+
+    id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
+    file_path = Column(String(512), unique=True, nullable=False)
+    file_name = Column(String(255), nullable=False)
+    file_extension = Column(String(10))
+    file_size = Column(BigInteger)
+    format_name = Column(String(50))
+    format_long_name = Column(String(100))
+    duration = Column(Float)
+    bit_rate = Column(BigInteger)
+    nb_streams = Column(Integer)
+    start_time = Column(Float, default=0)
+    title = Column(String(255))
+    artist = Column(String(255))
+    description = Column(Text)
+    probed_at = Column(DateTime)
+    registered_at = Column(DateTime, default=datetime.utcnow)
+    created_at = Column(DateTime, default=datetime.utcnow)
+    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+
+    video_streams = relationship(
+        "VideoStream", back_populates="video", cascade="all, delete-orphan"
+    )
+    audio_streams = relationship(
+        "AudioStream", back_populates="video", cascade="all, delete-orphan"
+    )
+    subtitle_streams = relationship(
+        "SubtitleStream", back_populates="video", cascade="all, delete-orphan"
+    )
+
+
+class VideoStream(Base):
+    __tablename__ = "video_streams"
+
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    video_id = Column(
+        UUID(as_uuid=True), ForeignKey("videos.id", ondelete="CASCADE"), nullable=False
+    )
+    stream_index = Column(Integer)
+    codec_name = Column(String(30))
+    codec_long_name = Column(String(100))
+    profile = Column(String(30))
+    level = Column(Integer)
+    width = Column(Integer)
+    height = Column(Integer)
+    coded_width = Column(Integer)
+    coded_height = Column(Integer)
+    aspect_ratio = Column(String(20))
+    pix_fmt = Column(String(30))
+    field_order = Column(String(20))
+    frame_rate = Column(String(20))
+    start_time = Column(Float)
+    duration = Column(Float)
+    bit_rate = Column(BigInteger)
+    nb_frames = Column(BigInteger)
+    color_range = Column(String(10))
+    color_space = Column(String(20))
+    has_b_frames = Column(Integer)
+    sample_aspect_ratio = Column(String(20))
+
+    video = relationship("Video", back_populates="video_streams")
+
+
+class AudioStream(Base):
+    __tablename__ = "audio_streams"
+
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    video_id = Column(
+        UUID(as_uuid=True), ForeignKey("videos.id", ondelete="CASCADE"), nullable=False
+    )
+    stream_index = Column(Integer)
+    codec_name = Column(String(30))
+    codec_long_name = Column(String(100))
+    profile = Column(String(30))
+    channels = Column(Integer)
+    channel_layout = Column(String(30))
+    sample_rate = Column(Integer)
+    sample_fmt = Column(String(20))
+    bit_rate = Column(BigInteger)
+    duration = Column(Float)
+    language = Column(String(10))
+
+    video = relationship("Video", back_populates="audio_streams")
+
+
+class SubtitleStream(Base):
+    __tablename__ = "subtitle_streams"
+
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    video_id = Column(
+        UUID(as_uuid=True), ForeignKey("videos.id", ondelete="CASCADE"), nullable=False
+    )
+    stream_index = Column(Integer)
+    codec_name = Column(String(30))
+    language = Column(String(10))
+
+    video = relationship("Video", back_populates="subtitle_streams")