infrahub-server 1.1.7__py3-none-any.whl → 1.2.0b1__py3-none-any.whl

infrahub/core/migrations/graph/m008_add_human_friendly_id.py

@@ -43,6 +43,6 @@ class Migration008(InternalSchemaMigration):
         ]
         return cls(migrations=migrations, **kwargs)  # type: ignore[arg-type]
 
-    async def validate_migration(self, db: InfrahubDatabase) -> MigrationResult:
+    async def validate_migration(self, db: InfrahubDatabase) -> MigrationResult:  # noqa: ARG002
         result = MigrationResult()
         return result

infrahub/core/migrations/graph/m009_add_generate_profile_attr.py

@@ -35,6 +35,6 @@ class Migration009(InternalSchemaMigration):
         ]
         return cls(migrations=migrations, **kwargs)  # type: ignore[arg-type]
 
-    async def validate_migration(self, db: InfrahubDatabase) -> MigrationResult:
+    async def validate_migration(self, db: InfrahubDatabase) -> MigrationResult:  # noqa: ARG002
         result = MigrationResult()
         return result

infrahub/core/migrations/graph/m010_add_generate_profile_attr_generic.py

@@ -35,6 +35,6 @@ class Migration010(InternalSchemaMigration):
         ]
         return cls(migrations=migrations, **kwargs)  # type: ignore[arg-type]
 
-    async def validate_migration(self, db: InfrahubDatabase) -> MigrationResult:
+    async def validate_migration(self, db: InfrahubDatabase) -> MigrationResult:  # noqa: ARG002
         result = MigrationResult()
         return result

infrahub/core/migrations/graph/m011_remove_profile_relationship_schema.py

@@ -16,7 +16,7 @@ class Migration011Query01(Query):
     type: QueryType = QueryType.WRITE
     insert_return = False
 
-    async def query_init(self, db: InfrahubDatabase, **kwargs: dict[str, Any]) -> None:
+    async def query_init(self, db: InfrahubDatabase, **kwargs: dict[str, Any]) -> None:  # noqa: ARG002
         query = """
         // get all the SchemaRelationship nodes for 'profiles' relationships
         MATCH (sr_to_delete:SchemaRelationship)-[HAS_ATTRIBUTE]->(:Attribute {name: "name"})-[HAS_VALUE]->(av:AttributeValue {value: "profiles"})
@@ -42,7 +42,7 @@ class Migration011(GraphMigration):
     queries: Sequence[type[Query]] = [Migration011Query01]
     minimum_version: int = 10
 
-    async def validate_migration(self, db: InfrahubDatabase) -> MigrationResult:
+    async def validate_migration(self, db: InfrahubDatabase) -> MigrationResult:  # noqa: ARG002
         result = MigrationResult()
 
         return result

infrahub/core/migrations/graph/m012_convert_account_generic.py

@@ -54,8 +54,7 @@ class Migration012RenameTypeAttributeData(AttributeRenameQuery):
             branch_support=BranchSupportType.AGNOSTIC.value,
         )
 
-        if "branch" in kwargs:
-            del kwargs["branch"]
+        kwargs.pop("branch", None)
 
         super().__init__(new_attr=new_attr, previous_attr=previous_attr, branch=global_branch, **kwargs)
 
@@ -113,8 +112,7 @@ class Migration012AddLabelData(NodeDuplicateQuery):
             sync_with_git=False,
         )
 
-        if "branch" in kwargs:
-            del kwargs["branch"]
+        kwargs.pop("branch", None)
 
         super().__init__(new_node=new_node, previous_node=previous_node, branch=branch, **kwargs)
 
@@ -165,8 +163,7 @@ class Migration012RenameRelationshipAccountTokenData(RelationshipDuplicateQuery)
             dst_peer=InfrahubKind.ACCOUNTTOKEN,
         )
 
-        if "branch" in kwargs:
-            del kwargs["branch"]
+        kwargs.pop("branch", None)
 
         super().__init__(new_rel=new_rel, previous_rel=previous_rel, branch=global_branch, **kwargs)
 
@@ -189,8 +186,7 @@ class Migration012RenameRelationshipRefreshTokenData(RelationshipDuplicateQuery)
             dst_peer=InfrahubKind.REFRESHTOKEN,
         )
 
-        if "branch" in kwargs:
-            del kwargs["branch"]
+        kwargs.pop("branch", None)
 
         super().__init__(new_rel=new_rel, previous_rel=previous_rel, branch=global_branch, **kwargs)
 
@@ -213,8 +209,7 @@ class Migration012RenameRelationshipThreadData(RelationshipDuplicateQuery):
             dst_peer=InfrahubKind.THREAD,
         )
 
-        if "branch" in kwargs:
-            del kwargs["branch"]
+        kwargs.pop("branch", None)
 
         super().__init__(new_rel=new_rel, previous_rel=previous_rel, branch=global_branch, **kwargs)
 
@@ -237,8 +232,7 @@ class Migration012RenameRelationshipCommentData(RelationshipDuplicateQuery):
             dst_peer=InfrahubKind.COMMENT,
         )
 
-        if "branch" in kwargs:
-            del kwargs["branch"]
+        kwargs.pop("branch", None)
 
         super().__init__(new_rel=new_rel, previous_rel=previous_rel, branch=default_branch, **kwargs)
 
@@ -249,8 +243,7 @@ class Migration012DeleteOldElementsSchema(DeleteElementInSchemaQuery):
     insert_return = False
 
     def __init__(self, **kwargs: Any):
-        if "branch" in kwargs:
-            del kwargs["branch"]
+        kwargs.pop("branch", None)
 
         super().__init__(
             element_names=["name", "password", "label", "description", "type", "role", "tokens"],
@@ -267,8 +260,7 @@ class Migration012UpdateDisplayLabels(SchemaAttributeUpdateQuery):
     insert_return = False
 
     def __init__(self, **kwargs: Any):
-        if "branch" in kwargs:
-            del kwargs["branch"]
+        kwargs.pop("branch", None)
 
         super().__init__(
             attribute_name="display_labels",
@@ -285,8 +277,7 @@ class Migration012UpdateOrderBy(SchemaAttributeUpdateQuery):
     insert_return = False
 
     def __init__(self, **kwargs: Any):
-        if "branch" in kwargs:
-            del kwargs["branch"]
+        kwargs.pop("branch", None)
 
         super().__init__(
             attribute_name="order_by",
@@ -303,8 +294,7 @@ class Migration012UpdateDefaultFilter(SchemaAttributeUpdateQuery):
     insert_return = False
 
     def __init__(self, **kwargs: Any):
-        if "branch" in kwargs:
-            del kwargs["branch"]
+        kwargs.pop("branch", None)
 
         super().__init__(
             attribute_name="default_filter",
@@ -321,8 +311,7 @@ class Migration012UpdateHFID(SchemaAttributeUpdateQuery):
     insert_return = False
 
     def __init__(self, **kwargs: Any):
-        if "branch" in kwargs:
-            del kwargs["branch"]
+        kwargs.pop("branch", None)
 
         super().__init__(
             attribute_name="human_friendly_id",
@@ -350,7 +339,7 @@ class Migration012(GraphMigration):
     ]
     minimum_version: int = 11
 
-    async def validate_migration(self, db: InfrahubDatabase) -> MigrationResult:
+    async def validate_migration(self, db: InfrahubDatabase) -> MigrationResult:  # noqa: ARG002
         result = MigrationResult()
 
         return result

infrahub/core/migrations/graph/m013_convert_git_password_credential.py

@@ -36,7 +36,7 @@ class Migration013ConvertCoreRepositoryWithCred(Query):
     name = "migration_013_convert_repository_with_cred"
     type = QueryType.WRITE
 
-    async def query_init(self, db: InfrahubDatabase, **kwargs: dict[str, Any]) -> None:
+    async def query_init(self, db: InfrahubDatabase, **kwargs: dict[str, Any]) -> None:  # noqa: ARG002
         at = Timestamp()
         filters, params = at.get_query_filter_path()
 
@@ -172,7 +172,7 @@ class Migration013ConvertCoreRepositoryWithoutCred(Query):
     name = "migration_013_convert_repository_without_cred"
     type = QueryType.WRITE
 
-    async def query_init(self, db: InfrahubDatabase, **kwargs: dict[str, Any]) -> None:
+    async def query_init(self, db: InfrahubDatabase, **kwargs: dict[str, Any]) -> None:  # noqa: ARG002
         at = Timestamp()
         filters, params = at.get_query_filter_path()
 
@@ -237,8 +237,7 @@ class Migration013DeleteUsernamePasswordGenericSchema(DeleteElementInSchemaQuery
     insert_return = False
 
     def __init__(self, **kwargs: Any):
-        if "branch" in kwargs:
-            del kwargs["branch"]
+        kwargs.pop("branch", None)
 
         super().__init__(
             element_names=["username", "password"],
@@ -255,8 +254,7 @@ class Migration013DeleteUsernamePasswordReadWriteSchema(DeleteElementInSchemaQue
     insert_return = False
 
     def __init__(self, **kwargs: Any):
-        if "branch" in kwargs:
-            del kwargs["branch"]
+        kwargs.pop("branch", None)
 
         super().__init__(
             element_names=["username", "password"],
@@ -273,8 +271,7 @@ class Migration013DeleteUsernamePasswordReadOnlySchema(DeleteElementInSchemaQuer
     insert_return = False
 
     def __init__(self, **kwargs: Any):
-        if "branch" in kwargs:
-            del kwargs["branch"]
+        kwargs.pop("branch", None)
 
         super().__init__(
             element_names=["username", "password"],
@@ -289,8 +286,7 @@ class Migration013AddInternalStatusData(AttributeAddQuery):
     type = QueryType.WRITE
 
     def __init__(self, **kwargs: Any):
-        if "branch" in kwargs:
-            del kwargs["branch"]
+        kwargs.pop("branch", None)
 
         super().__init__(
             node_kind="CoreGenericRepository",
@@ -315,7 +311,7 @@ class Migration013(GraphMigration):
     ]
     minimum_version: int = 12
 
-    async def validate_migration(self, db: InfrahubDatabase) -> MigrationResult:
+    async def validate_migration(self, db: InfrahubDatabase) -> MigrationResult:  # noqa: ARG002
         result = MigrationResult()
 
         return result

infrahub/core/migrations/graph/m014_remove_index_attr_value.py

@@ -33,12 +33,12 @@ class Migration014(GraphMigration):
             try:
                 ts.manager.index.init(nodes=[INDEX_TO_DELETE], rels=[])
                 await ts.manager.index.drop()
-            except Exception as exc:  # pylint: disable=broad-exception-caught
+            except Exception as exc:
                 result.errors.append(str(exc))
                 return result
 
         return result
 
-    async def validate_migration(self, db: InfrahubDatabase) -> MigrationResult:
+    async def validate_migration(self, db: InfrahubDatabase) -> MigrationResult:  # noqa: ARG002
         result = MigrationResult()
         return result

infrahub/core/migrations/graph/m015_diff_format_update.py

@@ -20,7 +20,7 @@ class Migration015(ArbitraryMigration):
     name: str = "015_diff_format_update"
     minimum_version: int = 14
 
-    async def validate_migration(self, db: InfrahubDatabase) -> MigrationResult:
+    async def validate_migration(self, db: InfrahubDatabase) -> MigrationResult:  # noqa: ARG002
         result = MigrationResult()
 
         return result

infrahub/core/migrations/graph/m016_diff_delete_bug_fix.py

@@ -20,7 +20,7 @@ class Migration016(ArbitraryMigration):
     name: str = "016_diff_delete_bug_fix_update"
     minimum_version: int = 15
 
-    async def validate_migration(self, db: InfrahubDatabase) -> MigrationResult:
+    async def validate_migration(self, db: InfrahubDatabase) -> MigrationResult:  # noqa: ARG002
         result = MigrationResult()
 
         return result

infrahub/core/migrations/graph/m017_add_core_profile.py

@@ -22,7 +22,7 @@ class Migration017(InternalSchemaMigration):
     minimum_version: int = 16
     migrations: Sequence[SchemaMigration] = []
 
-    async def validate_migration(self, db: InfrahubDatabase) -> MigrationResult:
+    async def validate_migration(self, db: InfrahubDatabase) -> MigrationResult:  # noqa: ARG002
         result = MigrationResult()
 
         return result

infrahub/core/migrations/graph/m018_uniqueness_nulls.py

@@ -26,7 +26,7 @@ class Migration018(InternalSchemaMigration):
     minimum_version: int = 17
     migrations: Sequence[SchemaMigration] = []
 
-    async def validate_migration(self, db: InfrahubDatabase) -> MigrationResult:
+    async def validate_migration(self, db: InfrahubDatabase) -> MigrationResult:  # noqa: ARG002
         result = MigrationResult()
 
         return result
@@ -52,7 +52,7 @@ class Migration018(InternalSchemaMigration):
 
         for schema_kind in schema_branch.node_names + schema_branch.generic_names:
             schema = schema_branch.get(name=schema_kind, duplicate=False)
-            if not isinstance(schema, (NodeSchema, GenericSchema)):
+            if not isinstance(schema, NodeSchema | GenericSchema):
                 continue
 
             schema_constraint_path_groups = schema.get_unique_constraint_schema_attribute_paths(

infrahub/core/migrations/graph/m019_restore_rels_to_time.py

@@ -0,0 +1,256 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any, Sequence
+
+from infrahub.core.migrations.shared import GraphMigration, MigrationResult
+from infrahub.log import get_logger
+
+from ...constants import GLOBAL_BRANCH_NAME, BranchSupportType
+from ...query import Query, QueryType
+
+if TYPE_CHECKING:
+    from infrahub.database import InfrahubDatabase
+
+log = get_logger()
+
+
+class FixBranchAwareEdgesQuery(Query):
+    name = "replace_global_edges"
+    type = QueryType.WRITE
+    insert_return = False
+
+    async def query_init(self, db: InfrahubDatabase, **kwargs: dict[str, Any]) -> None:  # noqa: ARG002
+        """
+        Between a Node and a Relationship, if Relationship.branch_support=aware, replace any global edge
+        to the branch of a non-global edge leaving out of the Relationship node. Note that there can't
+        be multiple non-global branches on these edges, as a dedicated Relationship node would exist for that.
+        """
+
+        query = """
+        MATCH (node:Node)-[global_edge:IS_RELATED {branch: $global_branch}]-(rel:Relationship)
+        WHERE rel.branch_support=$branch_aware
+        MATCH (rel)-[non_global_edge:IS_RELATED]-(node_2: Node)
+        WHERE non_global_edge.branch <> $global_branch
+        SET global_edge.branch = non_global_edge.branch
+        """
+
+        params = {
+            "global_branch": GLOBAL_BRANCH_NAME,
+            "branch_aware": BranchSupportType.AWARE.value,
+            "branch_agnostic": BranchSupportType.AGNOSTIC.value,
+        }
+
+        self.params.update(params)
+        self.add_to_query(query)
+
+
+class SetMissingToTimeQuery(Query):
+    name = "set_missing_to_time"
+    type = QueryType.WRITE
+    insert_return = False
+
+    async def query_init(self, db: InfrahubDatabase, **kwargs: dict[str, Any]) -> None:  # noqa: ARG002
+        """
+        If both a deleted edge and an active edge with no time exist between 2 nodes on the same branch,
+        set `to` time of active edge using `from` time of the deleted one. This would typically happen after having
+        replaced a deleted edge on global branch by correct branch with above query.
+        """
+
+        query = """
+        MATCH (node:Node)-[deleted_edge:IS_RELATED {status: "deleted"}]-(rel:Relationship)
+        MATCH (rel)-[active_edge:IS_RELATED {status: "active"}]-(node)
+        WHERE active_edge.to IS NULL AND deleted_edge.branch = active_edge.branch
+        SET active_edge.to = deleted_edge.from
+        """
+
+        self.add_to_query(query)
+
+
+class DeleteNodesRelsQuery(Query):
+    name = "delete_relationships_of_deleted_nodes"
+    type = QueryType.WRITE
+    insert_return = False
+
+    async def query_init(self, db: InfrahubDatabase, **kwargs: dict[str, Any]) -> None:  # noqa: ARG002
+        """
+        Some nodes may have been incorrectly deleted, typically, while these nodes edges connected to Root
+        are correctly deleted, edges connected to other `Node` through a `Relationship` node may still be active.
+        Following query correctly deletes these edges by both setting correct to time and creating corresponding deleted edge.
+        """
+
+        query = """
+        MATCH (deleted_node: Node)-[deleted_edge:IS_PART_OF {status: "deleted"}]->(:Root)
+        MATCH (deleted_node)-[:IS_RELATED]-(rel:Relationship)
+
+        // exclude nodes having been deleted through migration. find those with same uuid and exclude the one with earlier
+        // timestamp on active branch
+        WHERE NOT EXISTS {
+          MATCH (deleted_node)-[e1:IS_RELATED]-(rel)-[e2:IS_RELATED]-(other_node)
+          WITH deleted_node, other_node, MIN(e1.from) AS min_e1_from, MIN(e2.from) AS min_e2_from
+          WHERE deleted_node <> other_node AND deleted_node.uuid = other_node.uuid AND min_e1_from < min_e2_from
+        }
+
+        // Note that if an AWARE node has been deleted on a branch and relationship is AGNOSTIC, we do not "delete" this relationship
+        // right now as this aware node might exist on another branch.
+
+        // Set to time if there is an active edge:
+        // - on deleted edge branch
+        // - or on any branch and deleted node is agnostic
+        // - or deleted node is aware and rel is agnostic
+        CALL {
+            WITH rel, deleted_edge
+            OPTIONAL MATCH (rel)-[peer_active_edge {status: "active"}]-(peer_1)
+            WHERE (peer_active_edge.branch = deleted_edge.branch OR (rel.branch_support <> $branch_agnostic AND deleted_edge.branch = $global_branch))
+            AND peer_active_edge.to IS NULL
+            SET peer_active_edge.to = deleted_edge.from
+        }
+
+        // Get distinct rel nodes linked to a deleted node, with the time at which we should delete rel edges.
+        // Take the MAX time so if it does not take the deleted time of a node deleted through a duplication migration.
+        WITH DISTINCT rel,
+            deleted_edge.branch AS deleted_edge_branch,
+            deleted_edge.branch_level AS branch_level,
+            MAX(deleted_edge.from) as deleted_time,
+            deleted_node.branch_support as deleted_node_branch_support
+
+
+        // No need to check deleted edge branch because
+        // If deleted_node has different branch support type (agnostic/aware) than rel type,
+        // there might already be a deleted edge that we would not match if we filter on deleted_edge_branch.
+        // If both are aware, it still works, as we would have one Relationship node for each branch on which this relationship exists.
+        MATCH (rel)-[]-(peer_2)
+        WHERE NOT exists((rel)-[{status: "deleted"}]-(peer_2))
+
+
+        // If res is agnostic and delete node is agnostic, we should delete on global branch
+        // If rel is aware and deleted node is aware, we should use deleted edge branch
+        // If rel is aware and delete node is agnostic, we need to create deleted edges for every distinct branch on which this relationship exists.
+        WITH DISTINCT
+            CASE
+                // Branch on which `deleted` edge should be created depends on rel.branch_support.
+                WHEN rel.branch_support = $branch_agnostic
+                THEN CASE
+                    WHEN deleted_node_branch_support = $branch_agnostic THEN [$global_branch]
+                    ELSE []
+                END
+                ELSE
+                CASE
+                    WHEN deleted_node_branch_support = $branch_agnostic
+                    THEN COLLECT {
+                        WITH rel
+                        MATCH (rel)-[active_edge {status: "active"}]-(peer_2)
+                        RETURN DISTINCT active_edge.branch
+                    }
+                    ELSE
+                    CASE
+                        // if no active edge on this branch exists it means this relationship node is dedicated for another branch
+                        WHEN exists((rel)-[{status: "active", branch: deleted_edge_branch}]-(peer_2)) THEN [deleted_edge_branch]
+                        ELSE []
+                    END
+                END
+            END AS branches,
+          branch_level,
+          deleted_time,
+          peer_2,
+          rel
+
+        UNWIND branches as branch
+
+        // Then creates `deleted` edge.
+        // Below CALL subqueries are called once for each rel-peer_2 pair for which we want to create a deleted edge.
+        // Note that with current infrahub relationships edges design, only one of this CALL should be matched per pair.
+
+        CALL {
+            WITH rel, peer_2, branch, branch_level, deleted_time
+            MATCH (rel)-[:IS_RELATED]->(peer_2)
+            MERGE (rel)-[:IS_RELATED {status: "deleted", branch: branch, branch_level: branch_level, from: deleted_time}]->(peer_2)
+        }
+
+        CALL {
+            WITH rel, peer_2, branch, branch_level, deleted_time
+            MATCH (rel)-[:IS_PROTECTED]->(peer_2)
+            MERGE (rel)-[:IS_PROTECTED {status: "deleted", branch: branch, branch_level: branch_level, from: deleted_time}]->(peer_2)
+        }
+
+        CALL {
+            WITH rel, peer_2, branch, branch_level, deleted_time
+            MATCH (rel)-[:IS_VISIBLE]->(peer_2)
+            MERGE (rel)-[:IS_VISIBLE {status: "deleted", branch: branch, branch_level: branch_level, from: deleted_time}]->(peer_2)
+        }
+
+        CALL {
+            WITH rel, peer_2, branch, branch_level, deleted_time
+            MATCH (rel)-[:HAS_OWNER]->(peer_2)
+            MERGE (rel)-[:HAS_OWNER {status: "deleted", branch: branch, branch_level: branch_level, from: deleted_time}]->(peer_2)
+        }
+
+        CALL {
+            WITH rel, peer_2, branch, branch_level, deleted_time
+            MATCH (rel)-[:HAS_SOURCE]->(peer_2)
+            MERGE (rel)-[:HAS_SOURCE {status: "deleted", branch: branch, branch_level: branch_level, from: deleted_time}]->(peer_2)
+        }
+
+        CALL {
+            WITH rel, peer_2, branch, branch_level, deleted_time
+            MATCH (rel)<-[:IS_RELATED]-(peer_2)
+            MERGE (rel)<-[:IS_RELATED {status: "deleted", branch: branch, branch_level: branch_level, from: deleted_time}]-(peer_2)
+        }
+
+        CALL {
+            WITH rel, peer_2, branch, branch_level, deleted_time
+            MATCH (rel)<-[:IS_PROTECTED]-(peer_2)
+            MERGE (rel)<-[:IS_PROTECTED {status: "deleted", branch: branch, branch_level: branch_level, from: deleted_time}]-(peer_2)
+        }
+
+        CALL {
+            WITH rel, peer_2, branch, branch_level, deleted_time
+            MATCH (rel)<-[:IS_VISIBLE]-(peer_2)
+            MERGE (rel)<-[:IS_VISIBLE {status: "deleted", branch: branch, branch_level: branch_level, from: deleted_time}]-(peer_2)
+        }
+
+        CALL {
+            WITH rel, peer_2, branch, branch_level, deleted_time
+            MATCH (rel)<-[:HAS_OWNER]-(peer_2)
+            MERGE (rel)<-[:HAS_OWNER {status: "deleted", branch: branch, branch_level: branch_level, from: deleted_time}]-(peer_2)
+        }
+
+        CALL {
+            WITH rel, peer_2, branch, branch_level, deleted_time
+            MATCH (rel)<-[:HAS_SOURCE]-(peer_2)
+            MERGE (rel)<-[:HAS_SOURCE {status: "deleted", branch: branch, branch_level: branch_level, from: deleted_time}]-(peer_2)
+        }
+        """
+
+        params = {
+            "global_branch": GLOBAL_BRANCH_NAME,
+            "branch_aware": BranchSupportType.AWARE.value,
+            "branch_agnostic": BranchSupportType.AGNOSTIC.value,
+        }
+
+        self.params.update(params)
+        self.add_to_query(query)
+
+
+class Migration019(GraphMigration):
+    """
+    Fix corrupted state introduced by Migration012 when duplicating a CoreAccount (branch Aware)
+    being part of a CoreStandardGroup (branch Agnostic). Database is corrupted at multiple points:
+    - Old CoreAccount node <> group_member node `active` edge has no `to` time (possibly because of #5590).
+    - Old CoreAccount node <> group_member node `deleted` edge is on `$global_branch` branch instead of `main`.
+    - New CoreAccount node <> group_member node `active` edge is on `$global_branch` branch instead of `main`.
+
+    Also, users having deleted corresponding CoreStandardGroup will also have the following data corruption,
+    as deletion did not happen correctly due to above issues:
+    - Both CoreAccount <> group_member and CoreStandardGroup <> group_member edges
+      have not been deleted (ie status is `active` without `to` time and no additional `deleted` edge).
+
+    This migration fixes all above issues to have consistent edges, and fixes IFC-1204.
+    """
+
+    name: str = "019_fix_edges_state"
+    minimum_version: int = 18
+    queries: Sequence[type[Query]] = [FixBranchAwareEdgesQuery, SetMissingToTimeQuery, DeleteNodesRelsQuery]
+
+    async def validate_migration(self, db: InfrahubDatabase) -> MigrationResult:  # noqa: ARG002
+        result = MigrationResult()
+        return result

infrahub/core/migrations/graph/m020_add_generate_template_attr.py

@@ -0,0 +1,48 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any
+
+from typing_extensions import Self
+
+from infrahub.core.constants import SchemaPathType
+from infrahub.core.migrations.shared import MigrationResult
+from infrahub.core.path import SchemaPath
+
+from ..schema.node_attribute_add import NodeAttributeAddMigration
+from ..shared import InternalSchemaMigration
+
+if TYPE_CHECKING:
+    from infrahub.database import InfrahubDatabase
+
+
+class Migration020(InternalSchemaMigration):
+    name: str = "020_add_generate_template_attr"
+    minimum_version: int = 19
+
+    @classmethod
+    def init(cls, **kwargs: dict[str, Any]) -> Self:
+        internal_schema = cls.get_internal_schema()
+        schema_node = internal_schema.get_node(name="SchemaNode")
+        schema_generic = internal_schema.get_node(name="SchemaGeneric")
+
+        migrations = [
+            NodeAttributeAddMigration(
+                new_node_schema=schema_node,
+                previous_node_schema=schema_node,
+                schema_path=SchemaPath(
+                    schema_kind="SchemaNode", path_type=SchemaPathType.ATTRIBUTE, field_name="generate_template"
+                ),
+            ),
+            NodeAttributeAddMigration(
+                new_node_schema=schema_generic,
+                previous_node_schema=schema_generic,
+                schema_path=SchemaPath(
+                    schema_kind="SchemaNode", path_type=SchemaPathType.ATTRIBUTE, field_name="generate_template"
+                ),
+            ),
+        ]
+        return cls(migrations=migrations, **kwargs)  # type: ignore[arg-type]
+
+    async def validate_migration(self, db: InfrahubDatabase) -> MigrationResult:  # noqa: ARG002
+        result = MigrationResult()
+        return result

infrahub/core/migrations/query/attribute_add.py

@@ -30,7 +30,7 @@ class AttributeAddQuery(Query):
 
         super().__init__(**kwargs)
 
-    async def query_init(self, db: InfrahubDatabase, **kwargs: dict[str, Any]) -> None:
+    async def query_init(self, db: InfrahubDatabase, **kwargs: dict[str, Any]) -> None:  # noqa: ARG002
         branch_filter, branch_params = self.branch.get_query_filter_path(at=self.at.to_string())
         self.params.update(branch_params)
 

infrahub/core/migrations/query/attribute_rename.py

@@ -86,7 +86,7 @@ class AttributeRenameQuery(Query):
         subquery.append("RETURN peer_node as p2")
         return "\n".join(subquery)
 
-    async def query_init(self, db: InfrahubDatabase, **kwargs: dict[str, Any]) -> None:
+    async def query_init(self, db: InfrahubDatabase, **kwargs: dict[str, Any]) -> None:  # noqa: ARG002
         branch_filter, branch_params = self.branch.get_query_filter_path(at=self.at.to_string())
         self.params.update(branch_params)
 

infrahub/core/migrations/query/delete_element_in_schema.py

@@ -90,7 +90,7 @@ class DeleteElementInSchemaQuery(Query):
         sub_query_in = "\nUNION\n".join(sub_queries_in)
         return sub_query_in
 
-    async def query_init(self, db: InfrahubDatabase, **kwargs: dict[str, Any]) -> None:
+    async def query_init(self, db: InfrahubDatabase, **kwargs: dict[str, Any]) -> None:  # noqa: ARG002
         branch_filter, branch_params = self.branch.get_query_filter_path(at=self.at.to_string())
         self.params.update(branch_params)