mirror of
https://github.com/morpheus65535/bazarr
synced 2024-12-22 15:54:26 +00:00
280 lines
16 KiB
Python
280 lines
16 KiB
Python
"""empty message
|
|
|
|
Revision ID: dc09994b7e65
|
|
Revises:
|
|
Create Date: 2023-04-12 14:50:25.281288
|
|
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy import exc as sa_exc
|
|
import warnings
|
|
|
|
try:
|
|
from psycopg2.errors import UndefinedObject
|
|
except ImportError:
|
|
pass
|
|
|
|
from app.database import TableHistory, TableHistoryMovie, TableBlacklist, TableBlacklistMovie, TableEpisodes, \
|
|
TableShows, TableMovies, TableLanguagesProfiles, TableShowsRootfolder, TableMoviesRootfolder
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = 'dc09994b7e65'
|
|
down_revision = None
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
bind = op.get_context().bind
|
|
insp = sa.inspect(bind)
|
|
tables = insp.get_table_names()
|
|
|
|
should_recreate = 'always' if bind.engine.name == 'sqlite' else 'auto'
|
|
|
|
|
|
def column_exists(table_name, column_name):
|
|
columns = insp.get_columns(table_name)
|
|
return any(c["name"] == column_name for c in columns)
|
|
|
|
|
|
def constraint_exists(table_name, constraint_name):
|
|
constraints = insp.get_unique_constraints(table_name)
|
|
return any(c["name"] == constraint_name for c in constraints)
|
|
|
|
|
|
def column_type(table_name, column_name):
|
|
_type = [x['type'].python_type for x in insp.get_columns(table_name) if x['name'] == column_name]
|
|
return _type[0] if _type else None
|
|
|
|
|
|
def upgrade():
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
warnings.filterwarnings("ignore", category=sa_exc.SAWarning)
|
|
|
|
# Update announcements table
|
|
with op.batch_alter_table('table_announcements') as batch_op:
|
|
batch_op.execute('DROP INDEX IF EXISTS tableannouncements_hash')
|
|
if not column_exists('table_announcements', 'id'):
|
|
batch_op.add_column(sa.Column('id', sa.Integer, primary_key=True))
|
|
|
|
# Update system table
|
|
with op.batch_alter_table('system', recreate=should_recreate) as batch_op:
|
|
if not column_exists('system', 'id'):
|
|
batch_op.add_column(sa.Column('id', sa.Integer, primary_key=True))
|
|
|
|
# Update custom_score_profile_conditions table
|
|
if 'table_custom_score_profile_conditions' in tables:
|
|
with op.batch_alter_table('table_custom_score_profile_conditions') as batch_op:
|
|
batch_op.execute('DROP INDEX IF EXISTS tablecustomscoreprofileconditions_profile_id')
|
|
batch_op.alter_column('profile_id', index=False)
|
|
batch_op.execute('DROP INDEX IF EXISTS ix_table_custom_score_profile_conditions_profile_id;')
|
|
|
|
# Update notifier table
|
|
with op.batch_alter_table('table_settings_notifier') as batch_op:
|
|
batch_op.alter_column('name', existing_type=sa.TEXT(), nullable=False)
|
|
|
|
# Update series table
|
|
with op.batch_alter_table('table_shows', recreate=should_recreate) as batch_op:
|
|
if bind.engine.name == 'postgresql':
|
|
batch_op.execute('ALTER TABLE table_shows DROP CONSTRAINT IF EXISTS table_shows_pkey CASCADE;')
|
|
batch_op.execute(sa.update(TableShows)
|
|
.values({TableShows.profileId: None})
|
|
.where(TableShows.profileId.not_in(sa.select(TableLanguagesProfiles.profileId))))
|
|
batch_op.create_primary_key(constraint_name='pk_table_shows', columns=['sonarrSeriesId'])
|
|
batch_op.create_foreign_key(constraint_name='fk_series_profileId_languages_profiles',
|
|
referent_table='table_languages_profiles',
|
|
local_cols=['profileId'],
|
|
remote_cols=['profileId'],
|
|
ondelete='SET NULL')
|
|
if column_exists(table_name='table_shows', column_name='hearing_impaired'):
|
|
batch_op.drop_column(column_name='hearing_impaired')
|
|
if column_exists(table_name='table_shows', column_name='forced'):
|
|
batch_op.drop_column(column_name='forced')
|
|
batch_op.alter_column(column_name='imdbId', server_default=None)
|
|
batch_op.alter_column(column_name='tags', server_default=None)
|
|
batch_op.alter_column(column_name='seriesType', server_default=None)
|
|
batch_op.alter_column(column_name='tvdbId', existing_type=sa.INTEGER(), nullable=True)
|
|
if column_exists('table_shows', 'alternateTitles') and not column_exists('table_shows', 'alternativeTitles'):
|
|
batch_op.alter_column(column_name='alternateTitles', new_column_name='alternativeTitles')
|
|
batch_op.execute('DROP INDEX IF EXISTS tableshows_path')
|
|
batch_op.execute('DROP INDEX IF EXISTS tableshows_profileId')
|
|
batch_op.execute('DROP INDEX IF EXISTS tableshows_sonarrSeriesId')
|
|
if not constraint_exists('table_shows', 'unique_table_shows_path'):
|
|
batch_op.create_unique_constraint(constraint_name='unique_table_shows_path', columns=['path'])
|
|
|
|
# Update episodes table
|
|
with op.batch_alter_table('table_episodes') as batch_op:
|
|
if bind.engine.name == 'postgresql':
|
|
batch_op.execute('ALTER TABLE table_episodes DROP CONSTRAINT IF EXISTS table_episodes_pkey CASCADE;')
|
|
batch_op.execute(sa.delete(TableEpisodes).where(TableEpisodes.sonarrSeriesId.not_in(
|
|
sa.select(TableShows.sonarrSeriesId))))
|
|
batch_op.alter_column(column_name='sonarrSeriesId', existing_type=sa.INTEGER(), nullable=True)
|
|
batch_op.create_primary_key(constraint_name='pk_table_episodes', columns=['sonarrEpisodeId'])
|
|
batch_op.create_foreign_key(constraint_name='fk_sonarrSeriesId_episodes',
|
|
referent_table='table_shows',
|
|
local_cols=['sonarrSeriesId'],
|
|
remote_cols=['sonarrSeriesId'],
|
|
ondelete='CASCADE')
|
|
batch_op.alter_column(column_name='file_size', server_default='0')
|
|
if column_exists('table_episodes', 'scene_name'):
|
|
batch_op.alter_column(column_name='scene_name', new_column_name='sceneName')
|
|
batch_op.execute('DROP INDEX IF EXISTS tableepisodes_sonarrEpisodeId')
|
|
|
|
# Update series history table
|
|
table_history_timestamp_altered = False
|
|
with op.batch_alter_table('table_history') as batch_op:
|
|
batch_op.execute(sa.delete(TableHistory).where(TableHistory.sonarrEpisodeId.not_in(
|
|
sa.select(TableEpisodes.sonarrEpisodeId))))
|
|
batch_op.execute(sa.delete(TableHistory).where(TableHistory.sonarrSeriesId.not_in(
|
|
sa.select(TableShows.sonarrSeriesId))))
|
|
batch_op.alter_column(column_name='sonarrEpisodeId', existing_type=sa.INTEGER(), nullable=True)
|
|
batch_op.alter_column(column_name='sonarrSeriesId', existing_type=sa.INTEGER(), nullable=True)
|
|
batch_op.create_foreign_key(constraint_name='fk_sonarrEpisodeId_history',
|
|
referent_table='table_episodes',
|
|
local_cols=['sonarrEpisodeId'],
|
|
remote_cols=['sonarrEpisodeId'],
|
|
ondelete='CASCADE')
|
|
batch_op.create_foreign_key(constraint_name='fk_sonarrSeriesId_history',
|
|
referent_table='table_shows',
|
|
local_cols=['sonarrSeriesId'],
|
|
remote_cols=['sonarrSeriesId'],
|
|
ondelete='CASCADE')
|
|
if not column_exists('table_history', 'id'):
|
|
batch_op.add_column(sa.Column('id', sa.Integer, primary_key=True))
|
|
if column_type('table_history', 'score') == str:
|
|
batch_op.alter_column(column_name='score', existing_type=sa.Text, type_=sa.Integer)
|
|
if column_type('table_history', 'timestamp') == int:
|
|
table_history_timestamp_altered = True
|
|
batch_op.alter_column(column_name='timestamp', existing_type=sa.Integer, type_=sa.DateTime)
|
|
with op.batch_alter_table('table_history') as batch_op:
|
|
# must be run after alter_column as been committed
|
|
if table_history_timestamp_altered:
|
|
batch_op.execute(sa.text("UPDATE table_history SET timestamp = datetime(timestamp, 'unixepoch')"))
|
|
|
|
# Update series blacklist table
|
|
table_blacklist_timestamp_altered = False
|
|
with op.batch_alter_table('table_blacklist') as batch_op:
|
|
batch_op.execute(sa.delete(TableBlacklist).where(TableBlacklist.sonarr_episode_id.not_in(
|
|
sa.select(TableEpisodes.sonarrEpisodeId))))
|
|
batch_op.execute(sa.delete(TableBlacklist).where(TableBlacklist.sonarr_series_id.not_in(
|
|
sa.select(TableShows.sonarrSeriesId))))
|
|
batch_op.create_foreign_key(constraint_name='fk_sonarrEpisodeId_blacklist',
|
|
referent_table='table_episodes',
|
|
local_cols=['sonarr_episode_id'],
|
|
remote_cols=['sonarrEpisodeId'],
|
|
ondelete='CASCADE')
|
|
batch_op.create_foreign_key(constraint_name='fk_sonarrSeriesId_blacklist',
|
|
referent_table='table_shows',
|
|
local_cols=['sonarr_series_id'],
|
|
remote_cols=['sonarrSeriesId'],
|
|
ondelete='CASCADE')
|
|
if not column_exists('table_blacklist', 'id'):
|
|
batch_op.add_column(sa.Column('id', sa.Integer, primary_key=True))
|
|
if column_type('table_blacklist', 'timestamp') == int:
|
|
table_blacklist_timestamp_altered = True
|
|
batch_op.alter_column(column_name='timestamp', existing_type=sa.Integer, type_=sa.DateTime, nullable=True)
|
|
else:
|
|
batch_op.alter_column(column_name='timestamp', existing_type=sa.DATETIME(), nullable=True)
|
|
with op.batch_alter_table('table_blacklist') as batch_op:
|
|
# must be run after alter_column as been committed
|
|
if table_blacklist_timestamp_altered:
|
|
batch_op.execute(sa.text("UPDATE table_blacklist SET timestamp = datetime(timestamp, 'unixepoch')"))
|
|
|
|
# Update series rootfolder table
|
|
with op.batch_alter_table('table_shows_rootfolder') as batch_op:
|
|
batch_op.execute(sa.delete(TableShowsRootfolder))
|
|
if bind.engine.name == 'postgresql':
|
|
batch_op.execute('ALTER TABLE table_shows_rootfolder DROP CONSTRAINT IF EXISTS '
|
|
'table_shows_rootfolder_pkey CASCADE;')
|
|
batch_op.alter_column(column_name='id', existing_type=sa.INTEGER(), nullable=False, autoincrement=True)
|
|
batch_op.create_primary_key(constraint_name='pk_table_shows_rootfolder', columns=['id'])
|
|
|
|
# Update movies table
|
|
with op.batch_alter_table('table_movies', recreate=should_recreate) as batch_op:
|
|
if bind.engine.name == 'postgresql':
|
|
batch_op.execute('ALTER TABLE table_movies DROP CONSTRAINT IF EXISTS table_movies_pkey CASCADE;')
|
|
batch_op.execute(sa.update(TableMovies)
|
|
.values({TableMovies.profileId: None})
|
|
.where(TableMovies.profileId.not_in(sa.select(TableLanguagesProfiles.profileId))))
|
|
batch_op.create_primary_key(constraint_name='pk_table_movies', columns=['radarrId'])
|
|
batch_op.create_foreign_key(constraint_name='fk_movies_profileId_languages_profiles',
|
|
referent_table='table_languages_profiles',
|
|
local_cols=['profileId'],
|
|
remote_cols=['profileId'],
|
|
ondelete='SET NULL')
|
|
if column_exists(table_name='table_shows', column_name='hearing_impaired'):
|
|
batch_op.drop_column(column_name='hearing_impaired')
|
|
if column_exists(table_name='table_shows', column_name='forced'):
|
|
batch_op.drop_column(column_name='forced')
|
|
batch_op.alter_column(column_name='file_size', server_default='0')
|
|
batch_op.alter_column(column_name='tags', server_default=None)
|
|
batch_op.execute('DROP INDEX IF EXISTS tablemovies_path')
|
|
batch_op.execute('DROP INDEX IF EXISTS tablemovies_profileId')
|
|
batch_op.execute('DROP INDEX IF EXISTS tablemovies_radarrId')
|
|
batch_op.execute('DROP INDEX IF EXISTS tablemovies_tmdbId')
|
|
if not constraint_exists('table_movies', 'unique_table_movies_path'):
|
|
batch_op.create_unique_constraint(constraint_name='unique_table_movies_path', columns=['path'])
|
|
if not constraint_exists('table_movies', 'unique_table_movies_tmdbId'):
|
|
batch_op.create_unique_constraint(constraint_name='unique_table_movies_tmdbId', columns=['tmdbId'])
|
|
|
|
# Update movies history table
|
|
table_history_movie_timestamp_altered = False
|
|
with op.batch_alter_table('table_history_movie') as batch_op:
|
|
batch_op.execute(sa.delete(TableHistoryMovie).where(TableHistoryMovie.radarrId.not_in(
|
|
sa.select(TableMovies.radarrId))))
|
|
batch_op.alter_column(column_name='radarrId', existing_type=sa.INTEGER(), nullable=True)
|
|
batch_op.create_foreign_key(constraint_name='fk_radarrId_history_movie',
|
|
referent_table='table_movies',
|
|
local_cols=['radarrId'],
|
|
remote_cols=['radarrId'],
|
|
ondelete='CASCADE')
|
|
if not column_exists('table_history_movie', 'id'):
|
|
batch_op.add_column(sa.Column('id', sa.Integer, primary_key=True))
|
|
if column_type('table_history_movie', 'score') == str:
|
|
batch_op.alter_column(column_name='score', existing_type=sa.Text, type_=sa.Integer)
|
|
if column_type('table_history_movie', 'timestamp') == int:
|
|
table_history_movie_timestamp_altered = True
|
|
batch_op.alter_column(column_name='timestamp', existing_type=sa.Integer, type_=sa.DateTime)
|
|
with op.batch_alter_table('table_history_movie') as batch_op:
|
|
# must be run after alter_column as been committed
|
|
if table_history_movie_timestamp_altered:
|
|
batch_op.execute(sa.text("UPDATE table_history_movie SET timestamp = datetime(timestamp, 'unixepoch')"))
|
|
|
|
# Update movies blacklist table
|
|
table_blacklist_movie_timestamp_altered = False
|
|
with op.batch_alter_table('table_blacklist_movie') as batch_op:
|
|
batch_op.execute(sa.delete(TableBlacklistMovie).where(TableBlacklistMovie.radarr_id.not_in(
|
|
sa.select(TableMovies.radarrId))))
|
|
batch_op.create_foreign_key(constraint_name='fk_radarrId_blacklist_movie',
|
|
referent_table='table_movies',
|
|
local_cols=['radarr_id'],
|
|
remote_cols=['radarrId'],
|
|
ondelete='CASCADE')
|
|
if not column_exists('table_blacklist_movie', 'id'):
|
|
batch_op.add_column(sa.Column('id', sa.Integer, primary_key=True))
|
|
if column_type('table_blacklist_movie', 'timestamp') == int:
|
|
table_blacklist_movie_timestamp_altered = True
|
|
batch_op.alter_column(column_name='timestamp', existing_type=sa.Integer, type_=sa.DateTime, nullable=True)
|
|
else:
|
|
batch_op.alter_column(column_name='timestamp', existing_type=sa.DATETIME(), nullable=True)
|
|
with op.batch_alter_table('table_blacklist_movie') as batch_op:
|
|
# must be run after alter_column as been committed
|
|
if table_blacklist_movie_timestamp_altered:
|
|
batch_op.execute(sa.text("UPDATE table_blacklist_movie SET timestamp = datetime(timestamp, 'unixepoch')"))
|
|
|
|
# Update movies rootfolder table
|
|
with op.batch_alter_table('table_movies_rootfolder') as batch_op:
|
|
batch_op.execute(sa.delete(TableMoviesRootfolder))
|
|
if bind.engine.name == 'postgresql':
|
|
batch_op.execute('ALTER TABLE table_movies_rootfolder DROP CONSTRAINT IF EXISTS '
|
|
'table_movies_rootfolder_pkey CASCADE;')
|
|
batch_op.alter_column(column_name='id', existing_type=sa.INTEGER(), nullable=False, autoincrement=True)
|
|
batch_op.create_primary_key(constraint_name='pk_table_movies_rootfolder', columns=['id'])
|
|
|
|
warnings.filterwarnings("default", category=sa_exc.SAWarning)
|
|
# ### end Alembic commands ###
|
|
|
|
|
|
def downgrade():
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
pass
|
|
# ### end Alembic commands ###
|