34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
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 | None = Field(default=None)
|
|
service_id: str | None = Field(default=None)
|
|
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 - only enforce uniqueness when both service and service_id are not null
|
|
__table_args__ = ()
|
|
|
|
# relationships
|
|
sound: "Sound" = Relationship(back_populates="extractions")
|
|
user: "User" = Relationship(back_populates="extractions")
|