feat: Add Extraction model and seed main playlist functionality

This commit is contained in:
JSC
2025-07-28 19:39:32 +02:00
parent 34e6289f92
commit c993230f98
8 changed files with 121 additions and 49 deletions

39
app/models/extraction.py Normal file
View File

@@ -0,0 +1,39 @@
from typing import TYPE_CHECKING
from sqlmodel import Field, Relationship, UniqueConstraint
from app.models.base import BaseModel
if TYPE_CHECKING:
from app.models.sound import Sound
from app.models.user import User
class Extraction(BaseModel, table=True):
"""Database model for a stream."""
service: str = Field(nullable=False)
service_id: str = Field(nullable=False)
user_id: int = Field(foreign_key="user.id", nullable=False)
sound_id: int | None = Field(foreign_key="sound.id", default=None)
url: str = Field(nullable=False)
title: str | None = Field(default=None)
track: str | None = Field(default=None)
artist: str | None = Field(default=None)
album: str | None = Field(default=None)
genre: str | None = Field(default=None)
status: str = Field(nullable=False, default="pending")
error: str | None = Field(default=None)
# constraints
__table_args__ = (
UniqueConstraint(
"service",
"service_id",
name="uq_extraction_service_service_id",
),
)
# relationships
sound: "Sound" = Relationship(back_populates="extractions")
user: "User" = Relationship(back_populates="extractions")