MediaManager/media_manager/tv/models.py at 5e4c858bfad353de234d009c462be7b6584227ab · maxdorninger/MediaManager · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
from uuid import UUID
from datetime import date

from sqlalchemy import ForeignKey, PrimaryKeyConstraint, UniqueConstraint
from sqlalchemy.orm import Mapped, mapped_column, relationship

from media_manager.auth.db import User
from media_manager.database import Base
from media_manager.torrent.models import Quality


class Show(Base):
    __tablename__ = "show"
    __table_args__ = (UniqueConstraint("external_id", "metadata_provider"),)

    id: Mapped[UUID] = mapped_column(primary_key=True)
    external_id: Mapped[int]
    metadata_provider: Mapped[str]
    name: Mapped[str]
    overview: Mapped[str]
    year: Mapped[int | None]
    ended: Mapped[bool] = mapped_column(default=False)
    continuous_download: Mapped[bool] = mapped_column(default=False)
    library: Mapped[str] = mapped_column(default="")

    seasons: Mapped[list["Season"]] = relationship(
        back_populates="show", cascade="all, delete"
    )


class Season(Base):
    __tablename__ = "season"
    __table_args__ = (UniqueConstraint("show_id", "number"),)

    id: Mapped[UUID] = mapped_column(primary_key=True)
    show_id: Mapped[UUID] = mapped_column(
        ForeignKey(column="show.id", ondelete="CASCADE"),
    )
    number: Mapped[int]
    external_id: Mapped[int]
    name: Mapped[str]
    overview: Mapped[str]
    air_date: Mapped[date | None]

    show: Mapped["Show"] = relationship(back_populates="seasons")
    episodes: Mapped[list["Episode"]] = relationship(
        back_populates="season", cascade="all, delete"
    )

    season_files = relationship(
        "SeasonFile", back_populates="season", cascade="all, delete"
    )
    season_requests = relationship(
        "SeasonRequest", back_populates="season", cascade="all, delete"
    )


class Episode(Base):
    __tablename__ = "episode"
    __table_args__ = (UniqueConstraint("season_id", "number"),)
    id: Mapped[UUID] = mapped_column(primary_key=True)
    season_id: Mapped[UUID] = mapped_column(
        ForeignKey("season.id", ondelete="CASCADE"),
    )
    number: Mapped[int]
    external_id: Mapped[int]
    title: Mapped[str]

    season: Mapped["Season"] = relationship(back_populates="episodes")


class SeasonFile(Base):
    __tablename__ = "season_file"
    __table_args__ = (PrimaryKeyConstraint("season_id", "file_path_suffix"),)
    season_id: Mapped[UUID] = mapped_column(
        ForeignKey(column="season.id", ondelete="CASCADE"),
    )
    torrent_id: Mapped[UUID | None] = mapped_column(
        ForeignKey(column="torrent.id", ondelete="SET NULL"),
    )
    file_path_suffix: Mapped[str]
    quality: Mapped[Quality]

    torrent = relationship("Torrent", back_populates="season_files", uselist=False)
    season = relationship("Season", back_populates="season_files", uselist=False)


class SeasonRequest(Base):
    __tablename__ = "season_request"
    __table_args__ = (UniqueConstraint("season_id", "wanted_quality"),)
    id: Mapped[UUID] = mapped_column(primary_key=True)
    season_id: Mapped[UUID] = mapped_column(
        ForeignKey(column="season.id", ondelete="CASCADE"),
    )
    wanted_quality: Mapped[Quality]
    min_quality: Mapped[Quality]
    requested_by_id: Mapped[UUID | None] = mapped_column(
        ForeignKey(column="user.id", ondelete="SET NULL"),
    )
    authorized: Mapped[bool] = mapped_column(default=False)
    authorized_by_id: Mapped[UUID | None] = mapped_column(
        ForeignKey(column="user.id", ondelete="SET NULL"),
    )

    requested_by: Mapped["User|None"] = relationship(
        foreign_keys=[requested_by_id], uselist=False
    )
    authorized_by: Mapped["User|None"] = relationship(
        foreign_keys=[authorized_by_id], uselist=False
    )
    season = relationship("Season", back_populates="season_requests", uselist=False)