mirror of
https://github.com/maxdorninger/MediaManager.git
synced 2026-04-17 15:43:28 +02:00
274 lines
10 KiB
Python
274 lines
10 KiB
Python
"""Initial migration
|
|
|
|
Revision ID: 93fb07842385
|
|
Revises:
|
|
Create Date: 2025-05-27 21:36:18.532068
|
|
|
|
"""
|
|
|
|
from typing import Sequence, Union
|
|
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.dialects import postgresql
|
|
|
|
from alembic import op
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = "93fb07842385"
|
|
down_revision: Union[str, None] = None
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
"""Upgrade schema."""
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
# Create user table
|
|
op.create_table(
|
|
"user",
|
|
sa.Column("id", sa.UUID(), nullable=False),
|
|
sa.Column("email", sa.String(length=320), nullable=False),
|
|
sa.Column("hashed_password", sa.String(length=1024), nullable=False),
|
|
sa.Column("is_active", sa.Boolean(), nullable=False),
|
|
sa.Column("is_superuser", sa.Boolean(), nullable=False),
|
|
sa.Column("is_verified", sa.Boolean(), nullable=False),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_index(op.f("ix_user_email"), "user", ["email"], unique=True)
|
|
|
|
# Create oauth account table
|
|
op.create_table(
|
|
"oauth_account",
|
|
sa.Column("id", sa.UUID(), nullable=False),
|
|
sa.Column("oauth_name", sa.String(length=100), nullable=False),
|
|
sa.Column("access_token", sa.String(length=1024), nullable=False),
|
|
sa.Column("expires_at", sa.Integer(), nullable=True),
|
|
sa.Column("refresh_token", sa.String(length=1024), nullable=True),
|
|
sa.Column("account_id", sa.String(length=320), nullable=False),
|
|
sa.Column("account_email", sa.String(length=320), nullable=False),
|
|
sa.Column("user_id", sa.UUID(), nullable=False),
|
|
sa.ForeignKeyConstraint(["user_id"], ["user.id"], ondelete="cascade"),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_index(
|
|
op.f("ix_oauth_account_account_id"),
|
|
"oauth_account",
|
|
["account_id"],
|
|
unique=False,
|
|
)
|
|
op.create_index(
|
|
op.f("ix_oauth_account_oauth_name"),
|
|
"oauth_account",
|
|
["oauth_name"],
|
|
unique=False,
|
|
)
|
|
|
|
# Create torrent table
|
|
op.create_table(
|
|
"torrent",
|
|
sa.Column("id", sa.UUID(), nullable=False),
|
|
sa.Column(
|
|
"status",
|
|
sa.Enum(
|
|
"finished", "downloading", "error", "unknown", name="torrentstatus"
|
|
),
|
|
nullable=False,
|
|
),
|
|
sa.Column("title", sa.String(), nullable=False),
|
|
sa.Column(
|
|
"quality",
|
|
sa.Enum("uhd", "fullhd", "hd", "sd", "unknown", name="quality"),
|
|
nullable=False,
|
|
),
|
|
sa.Column("imported", sa.Boolean(), nullable=False),
|
|
sa.Column("hash", sa.String(), nullable=False),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
|
|
# Create indexer query result table
|
|
op.create_table(
|
|
"indexer_query_result",
|
|
sa.Column("id", sa.UUID(), nullable=False),
|
|
sa.Column("title", sa.String(), nullable=False),
|
|
sa.Column("download_url", sa.String(), nullable=False),
|
|
sa.Column("seeders", sa.Integer(), nullable=False),
|
|
sa.Column("flags", postgresql.ARRAY(sa.String()), nullable=True),
|
|
sa.Column(
|
|
"quality",
|
|
sa.Enum("uhd", "fullhd", "hd", "sd", "unknown", name="quality"),
|
|
nullable=False,
|
|
),
|
|
sa.Column("season", postgresql.ARRAY(sa.Integer()), nullable=True),
|
|
sa.Column("size", sa.BigInteger(), nullable=True),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
|
|
# Create notification table
|
|
op.create_table(
|
|
"notification",
|
|
sa.Column("id", sa.UUID(), nullable=False),
|
|
sa.Column("message", sa.String(), nullable=False),
|
|
sa.Column("read", sa.Boolean(), nullable=False),
|
|
sa.Column("timestamp", sa.DateTime(), nullable=False),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
|
|
# Create show table
|
|
op.create_table(
|
|
"show",
|
|
sa.Column("id", sa.UUID(), nullable=False),
|
|
sa.Column("external_id", sa.Integer(), nullable=False),
|
|
sa.Column("metadata_provider", sa.String(), nullable=False),
|
|
sa.Column("name", sa.String(), nullable=False),
|
|
sa.Column("overview", sa.String(), nullable=False),
|
|
sa.Column("year", sa.Integer(), nullable=True),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
sa.UniqueConstraint("external_id", "metadata_provider"),
|
|
)
|
|
|
|
# Create movie table
|
|
op.create_table(
|
|
"movie",
|
|
sa.Column("id", sa.UUID(), nullable=False),
|
|
sa.Column("external_id", sa.Integer(), nullable=False),
|
|
sa.Column("metadata_provider", sa.String(), nullable=False),
|
|
sa.Column("name", sa.String(), nullable=False),
|
|
sa.Column("overview", sa.String(), nullable=False),
|
|
sa.Column("year", sa.Integer(), nullable=True),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
sa.UniqueConstraint("external_id", "metadata_provider"),
|
|
)
|
|
|
|
# Create season table
|
|
op.create_table(
|
|
"season",
|
|
sa.Column("id", sa.UUID(), nullable=False),
|
|
sa.Column("show_id", sa.UUID(), nullable=False),
|
|
sa.Column("number", sa.Integer(), nullable=False),
|
|
sa.Column("external_id", sa.Integer(), nullable=False),
|
|
sa.Column("name", sa.String(), nullable=False),
|
|
sa.Column("overview", sa.String(), nullable=False),
|
|
sa.ForeignKeyConstraint(["show_id"], ["show.id"], ondelete="CASCADE"),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
sa.UniqueConstraint("show_id", "number"),
|
|
)
|
|
|
|
# Create movie file table
|
|
op.create_table(
|
|
"movie_file",
|
|
sa.Column("movie_id", sa.UUID(), nullable=False),
|
|
sa.Column("file_path_suffix", sa.String(), nullable=False),
|
|
sa.Column(
|
|
"quality",
|
|
sa.Enum("uhd", "fullhd", "hd", "sd", "unknown", name="quality"),
|
|
nullable=False,
|
|
),
|
|
sa.Column("torrent_id", sa.UUID(), nullable=True),
|
|
sa.ForeignKeyConstraint(["movie_id"], ["movie.id"], ondelete="CASCADE"),
|
|
sa.ForeignKeyConstraint(["torrent_id"], ["torrent.id"], ondelete="SET NULL"),
|
|
sa.PrimaryKeyConstraint("movie_id", "file_path_suffix"),
|
|
)
|
|
|
|
# Create movie request table
|
|
op.create_table(
|
|
"movie_request",
|
|
sa.Column("id", sa.UUID(), nullable=False),
|
|
sa.Column("movie_id", sa.UUID(), nullable=False),
|
|
sa.Column(
|
|
"wanted_quality",
|
|
sa.Enum("uhd", "fullhd", "hd", "sd", "unknown", name="quality"),
|
|
nullable=False,
|
|
),
|
|
sa.Column(
|
|
"min_quality",
|
|
sa.Enum("uhd", "fullhd", "hd", "sd", "unknown", name="quality"),
|
|
nullable=False,
|
|
),
|
|
sa.Column("authorized", sa.Boolean(), nullable=False),
|
|
sa.Column("requested_by_id", sa.UUID(), nullable=True),
|
|
sa.Column("authorized_by_id", sa.UUID(), nullable=True),
|
|
sa.ForeignKeyConstraint(["authorized_by_id"], ["user.id"], ondelete="SET NULL"),
|
|
sa.ForeignKeyConstraint(["movie_id"], ["movie.id"], ondelete="CASCADE"),
|
|
sa.ForeignKeyConstraint(["requested_by_id"], ["user.id"], ondelete="SET NULL"),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
sa.UniqueConstraint("movie_id", "wanted_quality"),
|
|
)
|
|
|
|
# Create episode table
|
|
op.create_table(
|
|
"episode",
|
|
sa.Column("id", sa.UUID(), nullable=False),
|
|
sa.Column("season_id", sa.UUID(), nullable=False),
|
|
sa.Column("number", sa.Integer(), nullable=False),
|
|
sa.Column("external_id", sa.Integer(), nullable=False),
|
|
sa.Column("title", sa.String(), nullable=False),
|
|
sa.ForeignKeyConstraint(["season_id"], ["season.id"], ondelete="CASCADE"),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
sa.UniqueConstraint("season_id", "number"),
|
|
)
|
|
|
|
# Create season file table
|
|
op.create_table(
|
|
"season_file",
|
|
sa.Column("season_id", sa.UUID(), nullable=False),
|
|
sa.Column("torrent_id", sa.UUID(), nullable=True),
|
|
sa.Column("file_path_suffix", sa.String(), nullable=False),
|
|
sa.Column(
|
|
"quality",
|
|
sa.Enum("uhd", "fullhd", "hd", "sd", "unknown", name="quality"),
|
|
nullable=False,
|
|
),
|
|
sa.ForeignKeyConstraint(["season_id"], ["season.id"], ondelete="CASCADE"),
|
|
sa.ForeignKeyConstraint(["torrent_id"], ["torrent.id"], ondelete="SET NULL"),
|
|
sa.PrimaryKeyConstraint("season_id", "file_path_suffix"),
|
|
)
|
|
|
|
# Create season request table
|
|
op.create_table(
|
|
"season_request",
|
|
sa.Column("id", sa.UUID(), nullable=False),
|
|
sa.Column("season_id", sa.UUID(), nullable=False),
|
|
sa.Column(
|
|
"wanted_quality",
|
|
sa.Enum("uhd", "fullhd", "hd", "sd", "unknown", name="quality"),
|
|
nullable=False,
|
|
),
|
|
sa.Column(
|
|
"min_quality",
|
|
sa.Enum("uhd", "fullhd", "hd", "sd", "unknown", name="quality"),
|
|
nullable=False,
|
|
),
|
|
sa.Column("requested_by_id", sa.UUID(), nullable=True),
|
|
sa.Column("authorized", sa.Boolean(), nullable=False),
|
|
sa.Column("authorized_by_id", sa.UUID(), nullable=True),
|
|
sa.ForeignKeyConstraint(["authorized_by_id"], ["user.id"], ondelete="SET NULL"),
|
|
sa.ForeignKeyConstraint(["requested_by_id"], ["user.id"], ondelete="SET NULL"),
|
|
sa.ForeignKeyConstraint(["season_id"], ["season.id"], ondelete="CASCADE"),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
sa.UniqueConstraint("season_id", "wanted_quality"),
|
|
)
|
|
|
|
# ### end Alembic commands ###
|
|
|
|
|
|
def downgrade() -> None:
|
|
"""Downgrade schema."""
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
op.drop_table("season_request")
|
|
op.drop_table("season_file")
|
|
op.drop_table("episode")
|
|
op.drop_table("movie_request")
|
|
op.drop_table("movie_file")
|
|
op.drop_table("season")
|
|
op.drop_table("movie")
|
|
op.drop_table("show")
|
|
op.drop_table("notification")
|
|
op.drop_table("indexer_query_result")
|
|
op.drop_table("torrent")
|
|
op.drop_index(op.f("ix_oauth_account_oauth_name"), table_name="oauth_account")
|
|
op.drop_index(op.f("ix_oauth_account_account_id"), table_name="oauth_account")
|
|
op.drop_table("oauth_account")
|
|
op.drop_index(op.f("ix_user_email"), table_name="user")
|
|
op.drop_table("user")
|
|
# ### end Alembic commands ###
|