Add graph number in branch and set it

Author: gmazoyer

backend/infrahub/core/branch/models.py

@@ -46,6 +46,7 @@ class Branch(StandardNode):
     is_isolated: bool = True
     schema_changed_at: Optional[str] = None
     schema_hash: Optional[SchemaBranchHash] = None
+    graph_version: int | None = None
 
     _exclude_attrs: list[str] = ["id", "uuid", "owner"]
 

backend/infrahub/core/migrations/shared.py

@@ -7,6 +7,7 @@
 from typing_extensions import Self
 
 from infrahub.core import registry
+from infrahub.core.branch.enums import BranchStatus
 from infrahub.core.path import SchemaPath  # noqa: TC001
 from infrahub.core.query import Query  # noqa: TC001
 from infrahub.core.schema import (
@@ -257,7 +258,7 @@ async def rebase_branch(self, db: InfrahubDatabase, branch: Branch) -> bool:
         try:
             await branch.rebase(db=db)
             console.print("done")
-        except Exception:
+            branch.graph_version = self.minimum_version + 1
             # NOTE: Narrow to more accurate exception
             console.print("failed")
             branch.status = BranchStatus.NEED_UPGRADE_REBASE