Catch migration failure exception to update CLI feedback

Author: gmazoyer

backend/infrahub/cli/db.py

@@ -36,6 +36,7 @@
     GraphRelationshipProperties,
 )
 from infrahub.core.initialization import get_root_node, initialize_registry
+from infrahub.core.migrations.exceptions import MigrationFailureError
 from infrahub.core.migrations.graph import get_graph_migrations, get_migration_by_number
 from infrahub.core.migrations.schema.models import SchemaApplyMigrationData
 from infrahub.core.migrations.schema.tasks import schema_apply_migrations
@@ -384,7 +385,7 @@ async def trigger_rebase_branches(db: InfrahubDatabase) -> None:
                 send_events=False,
             )
             rprint(SUCCESS_BADGE)
-        except ValidationError:
+        except (ValidationError, MigrationFailureError):
             rprint(FAILED_BADGE)
 
 

backend/infrahub/core/branch/tasks.py

@@ -89,7 +89,7 @@ async def migrate_branch(branch: str, context: InfrahubContext, send_events: boo
             await migration_runner.run(db=db)
         except MigrationFailureError as exc:
             log.error(f"Failed to run migrations for branch '{obj.name}': {exc.errors}")
-            return
+            raise
 
         if obj.status == BranchStatus.NEED_UPGRADE_REBASE:
             obj.status = BranchStatus.OPEN