diff --git a/gns3server/db/tasks.py b/gns3server/db/tasks.py index cb4850c3..33f06389 100644 --- a/gns3server/db/tasks.py +++ b/gns3server/db/tasks.py @@ -31,6 +31,7 @@ from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession from alembic import command, config from alembic.script import ScriptDirectory from alembic.runtime.migration import MigrationContext +from alembic.util.exc import CommandError from gns3server.db.repositories.computes import ComputesRepository from gns3server.db.repositories.images import ImagesRepository @@ -48,13 +49,19 @@ log = logging.getLogger(__name__) def run_upgrade(connection, cfg): cfg.attributes["connection"] = connection - command.upgrade(cfg, "head") + try: + command.upgrade(cfg, "head") + except CommandError as e: + log.error(f"Could not upgrade database: {e}") def run_stamp(connection, cfg): cfg.attributes["connection"] = connection - command.stamp(cfg, 'head') + try: + command.stamp(cfg, "head") + except CommandError as e: + log.error(f"Could not stamp database: {e}") def check_revision(connection, cfg):