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