Add need upgrade rebase branch status

Author: gmazoyer

backend/infrahub/cli/db.py

@@ -427,6 +427,7 @@ async def rebase_and_migrate_branches(
 
     for migration in rebase_migrations:
         execution_result = await migration.execute_against_branches(db=db, branches=branches)
+        validation_result = None
 
         if execution_result.success:
             validation_result = await migration.validate_migration(db=db)

backend/infrahub/core/branch/enums.py

@@ -4,5 +4,6 @@
 class BranchStatus(InfrahubStringEnum):
     OPEN = "OPEN"
     NEED_REBASE = "NEED_REBASE"
+    NEED_UPGRADE_REBASE = "NEED_UPGRADE_REBASE"
     CLOSED = "CLOSED"
     DELETING = "DELETING"

backend/infrahub/core/migrations/shared.py

@@ -260,7 +260,10 @@ async def rebase_branch(self, db: InfrahubDatabase, branch: Branch) -> bool:
         except Exception:
             # NOTE: Narrow to more accurate exception
             console.print("failed")
+            branch.status = BranchStatus.NEED_UPGRADE_REBASE
             return False
+        finally:
+            await branch.save(db=db)
 
         return True