mirror of
https://github.com/maxdorninger/MediaManager.git
synced 2026-04-17 15:43:28 +02:00
fix initial migration
This commit is contained in:
maxDorninger
@@ -8,6 +8,9 @@ Create Date: 2025-05-27 21:36:18.532068
|
||||
|
||||
from typing import Sequence, Union
|
||||
|
||||
from alembic import op
|
||||
import sqlalchemy as sa
|
||||
from sqlalchemy.dialects import postgresql
|
||||
|
||||
# revision identifiers, used by Alembic.
|
||||
revision: str = "93fb07842385"
|
||||
@@ -20,11 +23,190 @@ 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! ###
|
||||
|
||||
# ### end Alembic commands ###
|
||||
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 ###
|
||||
Reference in New Issue
Block a user