lbry-sdk/lbry/extras/daemon/migrator/dbmigrator.py

# pylint: skip-file
import os
import sys
import logging

log = logging.getLogger(__name__)


def migrate_db(conf, start, end):
    current = start
    while current < end:
        if current == 1:
            from .migrate1to2 import do_migration
        elif current == 2:
            from .migrate2to3 import do_migration
        elif current == 3:
            from .migrate3to4 import do_migration
        elif current == 4:
            from .migrate4to5 import do_migration
        elif current == 5:
            from .migrate5to6 import do_migration
        elif current == 6:
            from .migrate6to7 import do_migration
        elif current == 7:
            from .migrate7to8 import do_migration
        elif current == 8:
            from .migrate8to9 import do_migration
        elif current == 9:
            from .migrate9to10 import do_migration
        elif current == 10:
            from .migrate10to11 import do_migration
        elif current == 11:
            from .migrate11to12 import do_migration
        elif current == 12:
            from .migrate12to13 import do_migration
        elif current == 13:
            from .migrate13to14 import do_migration
        elif current == 14:
            from .migrate14to15 import do_migration
        else:
            raise Exception(f"DB migration of version {current} to {current+1} is not available")
        try:
            do_migration(conf)
        except Exception:
            log.exception("failed to migrate database")
            if os.path.exists(os.path.join(conf.data_dir, "lbrynet.sqlite")):
                backup_name = f"rev_{current}_unmigrated_database"
                count = 0
                while os.path.exists(os.path.join(conf.data_dir, backup_name + ".sqlite")):
                    count += 1
                    backup_name = f"rev_{current}_unmigrated_database_{count}"
                backup_path = os.path.join(conf.data_dir, backup_name + ".sqlite")
                os.rename(os.path.join(conf.data_dir, "lbrynet.sqlite"), backup_path)
                log.info("made a backup of the unmigrated database: %s", backup_path)
            if os.path.isfile(os.path.join(conf.data_dir, "db_revision")):
                os.remove(os.path.join(conf.data_dir, "db_revision"))
            return None
        current += 1
        log.info("successfully migrated the database from revision %i to %i", current - 1, current)
    return None


def run_migration_script():
    log_format = "(%(asctime)s)[%(filename)s:%(lineno)s] %(funcName)s(): %(message)s"
    logging.basicConfig(level=logging.DEBUG, format=log_format, filename="migrator.log")
    sys.stdout = open("migrator.out.log", 'w')
    sys.stderr = open("migrator.err.log", 'w')
    migrate_db(sys.argv[1], int(sys.argv[2]), int(sys.argv[3]))


if __name__ == "__main__":
    run_migration_script()