def upgrade(ver, session):
if ver is None:
columns = table_columns('remember_rejected_entry', session)
if 'uid' in columns:
# Drop the old table
log.info('Dropping old version of remember_rejected_entry table from db')
drop_tables(['remember_rejected_entry'], session)
# Create new table from the current model
Base.metadata.create_all(bind=session.bind)
# We go directly to version 2, as remember_rejected_entries table has just been made from current model
# TODO: Fix this somehow. Just avoid dropping tables?
ver = 3
else:
ver = 0
if ver == 0:
log.info('Adding reason column to remember_rejected_entry table.')
table_add_column('remember_rejected_entry', 'reason', String, session)
ver = 1
if ver == 1:
log.info('Adding `added` column to remember_rejected_entry table.')
table_add_column('remember_rejected_entry', 'added', DateTime, session, default=datetime.now)
ver = 2
if ver == 2:
log.info('Adding expires column to remember_rejected_entry table.')
table_add_column('remember_rejected_entry', 'expires', DateTime, session)
ver = 3
return ver
def upgrade(ver, session):
if ver is None:
columns = table_columns('remember_rejected_entry', session)
if 'uid' in columns:
# Drop the old table
log.info(
'Dropping old version of remember_rejected_entry table from db'
)
drop_tables(['remember_rejected_entry'], session)
# Create new table from the current model
Base.metadata.create_all(bind=session.bind)
# We go directly to version 2, as remember_rejected_entries table has just been made from current model
# TODO: Fix this somehow. Just avoid dropping tables?
ver = 3
else:
ver = 0
if ver == 0:
log.info('Adding reason column to remember_rejected_entry table.')
table_add_column('remember_rejected_entry', 'reason', String, session)
ver = 1
if ver == 1:
log.info('Adding `added` column to remember_rejected_entry table.')
table_add_column('remember_rejected_entry',
'added',
DateTime,
session,
default=datetime.now)
ver = 2
if ver == 2:
log.info('Adding expires column to remember_rejected_entry table.')
table_add_column('remember_rejected_entry', 'expires', DateTime,
session)
ver = 3
return ver
def upgrade(ver, session):
if ver is None:
columns = table_columns("thetvdb_favorites", session)
if not "series_ids" in columns:
# Drop the old table
log.info("Dropping old version of thetvdb_favorites table from db")
drop_tables(["thetvdb_favorites"], session)
# Create new table from the current model
Base.metadata.create_all(bind=session.bind)
ver = 0
return ver
def upgrade(ver, session):
if ver is None:
columns = table_columns('thetvdb_favorites', session)
if not 'series_ids' in columns:
# Drop the old table
log.info('Dropping old version of thetvdb_favorites table from db')
drop_tables(['thetvdb_favorites'], session)
# Create new table from the current model
Base.metadata.create_all(bind=session.bind)
ver = 0
return ver
def upgrade(ver, session):
if ver is None:
if table_exists('episode_qualities', session):
log.info('Series database format is too old to upgrade, dropping and recreating tables.')
# Drop the deprecated data
drop_tables(['series', 'series_episodes', 'episode_qualities'], session)
# Create new tables from the current models
Base.metadata.create_all(bind=session.bind)
# Upgrade episode_releases table to have a proper count and seed it with appropriate numbers
columns = table_columns('episode_releases', session)
if not 'proper_count' in columns:
log.info('Upgrading episode_releases table to have proper_count column')
table_add_column('episode_releases', 'proper_count', Integer, session)
release_table = table_schema('episode_releases', session)
for row in session.execute(select([release_table.c.id, release_table.c.title])):
# Recalculate the proper_count from title for old episodes
proper_count = len([part for part in re.split('[\W_]+', row['title'].lower())
if part in SeriesParser.propers])
session.execute(update(release_table, release_table.c.id == row['id'], {'proper_count': proper_count}))
ver = 0
if ver == 0:
log.info('Migrating first_seen column from series_episodes to episode_releases table.')
# Create the column in episode_releases
table_add_column('episode_releases', 'first_seen', DateTime, session)
# Seed the first_seen value for all the past releases with the first_seen of their episode.
episode_table = table_schema('series_episodes', session)
release_table = table_schema('episode_releases', session)
for row in session.execute(select([episode_table.c.id, episode_table.c.first_seen])):
session.execute(update(release_table, release_table.c.episode_id == row['id'],
{'first_seen': row['first_seen']}))
ver = 1
if ver == 1:
log.info('Adding `identified_by` column to series table.')
table_add_column('series', 'identified_by', String, session)
ver = 2
if ver == 2:
release_table = table_schema('episode_releases', session)
log.info('Creating index on episode_releases table.')
Index('ix_episode_releases_episode_id', release_table.c.episode_id).create(bind=session.bind)
ver = 3
return ver
