Add generate_template to node and generic schema (#5910)

Author: gmazoyer

backend/infrahub/core/graph/__init__.py

@@ -1 +1 @@
-GRAPH_VERSION = 19
+GRAPH_VERSION = 20

backend/infrahub/core/migrations/graph/__init__.py

@@ -21,6 +21,7 @@
 from .m017_add_core_profile import Migration017
 from .m018_uniqueness_nulls import Migration018
 from .m019_restore_rels_to_time import Migration019
+from .m020_add_generate_template_attr import Migration020
 
 if TYPE_CHECKING:
     from infrahub.core.root import Root
@@ -47,6 +48,7 @@
     Migration017,
     Migration018,
     Migration019,
+    Migration020,
 ]
 
 

backend/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