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

infrahub/core/schema/definitions/core/resource_pool.py

@@ -2,179 +2,169 @@ from infrahub.core.constants import (
     BranchSupportType,
     InfrahubKind,
 )
+from infrahub.core.constants import RelationshipCardinality as Cardinality
+from infrahub.core.constants import RelationshipKind as RelKind
 
-core_resource_pool = {
-    "name": "ResourcePool",
-    "namespace": "Core",
-    "label": "Resource Pool",
-    "description": "The resource manager contains pools of resources to allow for automatic assignments.",
-    "include_in_menu": False,
-    "default_filter": "name__value",
-    "order_by": ["name__value"],
-    "display_labels": ["name__value"],
-    "human_friendly_id": ["name__value"],
-    "icon": "mdi:view-grid-outline",
-    "branch": BranchSupportType.AGNOSTIC.value,
-    "uniqueness_constraints": [["name__value"]],
-    "generate_profile": False,
-    "attributes": [
-        {
-            "name": "name",
-            "kind": "Text",
-            "order_weight": 1000,
-            "unique": True,
-        },
-        {
-            "name": "description",
-            "kind": "Text",
-            "optional": True,
-            "order_weight": 2000,
-        },
+from ...attribute_schema import AttributeSchema as Attr
+from ...generic_schema import GenericSchema
+from ...node_schema import NodeSchema
+from ...relationship_schema import (
+    RelationshipSchema as Rel,
+)
+
+core_resource_pool = GenericSchema(
+    name="ResourcePool",
+    namespace="Core",
+    label="Resource Pool",
+    description="The resource manager contains pools of resources to allow for automatic assignments.",
+    include_in_menu=False,
+    default_filter="name__value",
+    order_by=["name__value"],
+    display_labels=["name__value"],
+    human_friendly_id=["name__value"],
+    icon="mdi:view-grid-outline",
+    branch=BranchSupportType.AGNOSTIC,
+    uniqueness_constraints=[["name__value"]],
+    generate_profile=False,
+    attributes=[
+        Attr(name="name", kind="Text", order_weight=1000, unique=True),
+        Attr(name="description", kind="Text", optional=True, order_weight=2000),
     ],
-}
+)
 
-core_ip_prefix_pool = {
-    "name": "IPPrefixPool",
-    "namespace": "Core",
-    "description": "A pool of IP prefix resources",
-    "label": "IP Prefix Pool",
-    "include_in_menu": False,
-    "branch": BranchSupportType.AGNOSTIC.value,
-    "generate_profile": False,
-    "inherit_from": [InfrahubKind.RESOURCEPOOL, InfrahubKind.LINEAGESOURCE],
-    "attributes": [
-        {
-            "name": "default_prefix_length",
-            "kind": "Number",
-            "description": "The default prefix length as an integer for prefixes allocated from this pool.",
-            "optional": True,
-            "order_weight": 5000,
-        },
-        {
-            "name": "default_member_type",
-            "kind": "Text",
-            "enum": ["prefix", "address"],
-            "default_value": "prefix",
-            "optional": True,
-            "order_weight": 3000,
-        },
-        {
-            "name": "default_prefix_type",
-            "kind": "Text",
-            "optional": True,
-            "order_weight": 4000,
-        },
+core_ip_prefix_pool = NodeSchema(
+    name="IPPrefixPool",
+    namespace="Core",
+    description="A pool of IP prefix resources",
+    label="IP Prefix Pool",
+    include_in_menu=False,
+    branch=BranchSupportType.AGNOSTIC,
+    generate_profile=False,
+    inherit_from=[InfrahubKind.RESOURCEPOOL, InfrahubKind.LINEAGESOURCE],
+    attributes=[
+        Attr(
+            name="default_prefix_length",
+            kind="Number",
+            description="The default prefix length as an integer for prefixes allocated from this pool.",
+            optional=True,
+            order_weight=5000,
+        ),
+        Attr(
+            name="default_member_type",
+            kind="Text",
+            enum=["prefix", "address"],
+            default_value="prefix",
+            optional=True,
+            order_weight=3000,
+        ),
+        Attr(name="default_prefix_type", kind="Text", optional=True, order_weight=4000),
     ],
-    "relationships": [
-        {
-            "name": "resources",
-            "peer": "BuiltinIPPrefix",
-            "kind": "Attribute",
-            "identifier": "prefixpool__resource",
-            "cardinality": "many",
-            "branch": BranchSupportType.AGNOSTIC.value,
-            "optional": False,
-            "order_weight": 6000,
-        },
-        {
-            "name": "ip_namespace",
-            "peer": "BuiltinIPNamespace",
-            "kind": "Attribute",
-            "identifier": "prefixpool__ipnamespace",
-            "cardinality": "one",
-            "branch": BranchSupportType.AGNOSTIC.value,
-            "optional": False,
-            "order_weight": 7000,
-        },
+    relationships=[
+        Rel(
+            name="resources",
+            peer=InfrahubKind.IPPREFIX,
+            kind=RelKind.ATTRIBUTE,
+            identifier="prefixpool__resource",
+            cardinality=Cardinality.MANY,
+            branch=BranchSupportType.AGNOSTIC,
+            optional=False,
+            order_weight=6000,
+        ),
+        Rel(
+            name="ip_namespace",
+            peer=InfrahubKind.IPNAMESPACE,
+            kind=RelKind.ATTRIBUTE,
+            identifier="prefixpool__ipnamespace",
+            cardinality=Cardinality.ONE,
+            branch=BranchSupportType.AGNOSTIC,
+            optional=False,
+            order_weight=7000,
+        ),
     ],
-}
+)
 
-core_ip_address_pool = {
-    "name": "IPAddressPool",
-    "namespace": "Core",
-    "description": "A pool of IP address resources",
-    "label": "IP Address Pool",
-    "include_in_menu": False,
-    "branch": BranchSupportType.AGNOSTIC.value,
-    "generate_profile": False,
-    "inherit_from": [InfrahubKind.RESOURCEPOOL, InfrahubKind.LINEAGESOURCE],
-    "attributes": [
-        {
-            "name": "default_address_type",
-            "kind": "Text",
-            "optional": False,
-            "description": "The object type to create when reserving a resource in the pool",
-            "order_weight": 3000,
-        },
-        {
-            "name": "default_prefix_length",
-            "kind": "Number",
-            "description": "The default prefix length as an integer for addresses allocated from this pool.",
-            "optional": True,
-            "order_weight": 4000,
-        },
+core_ip_address_pool = NodeSchema(
+    name="IPAddressPool",
+    namespace="Core",
+    description="A pool of IP address resources",
+    label="IP Address Pool",
+    include_in_menu=False,
+    branch=BranchSupportType.AGNOSTIC,
+    generate_profile=False,
+    inherit_from=[InfrahubKind.RESOURCEPOOL, InfrahubKind.LINEAGESOURCE],
+    attributes=[
+        Attr(
+            name="default_address_type",
+            kind="Text",
+            optional=False,
+            description="The object type to create when reserving a resource in the pool",
+            order_weight=3000,
+        ),
+        Attr(
+            name="default_prefix_length",
+            kind="Number",
+            optional=True,
+            description="The default prefix length as an integer for addresses allocated from this pool.",
+            order_weight=4000,
+        ),
     ],
-    "relationships": [
-        {
-            "name": "resources",
-            "peer": "BuiltinIPPrefix",
-            "kind": "Attribute",
-            "identifier": "ipaddresspool__resource",
-            "cardinality": "many",
-            "branch": BranchSupportType.AGNOSTIC.value,
-            "optional": False,
-            "order_weight": 5000,
-        },
-        {
-            "name": "ip_namespace",
-            "peer": "BuiltinIPNamespace",
-            "kind": "Attribute",
-            "identifier": "ipaddresspool__ipnamespace",
-            "cardinality": "one",
-            "branch": BranchSupportType.AGNOSTIC.value,
-            "optional": False,
-            "order_weight": 6000,
-        },
+    relationships=[
+        Rel(
+            name="resources",
+            peer=InfrahubKind.IPPREFIX,
+            kind=RelKind.ATTRIBUTE,
+            identifier="ipaddresspool__resource",
+            cardinality=Cardinality.MANY,
+            branch=BranchSupportType.AGNOSTIC,
+            optional=False,
+            order_weight=5000,
+        ),
+        Rel(
+            name="ip_namespace",
+            peer=InfrahubKind.IPNAMESPACE,
+            kind=RelKind.ATTRIBUTE,
+            identifier="ipaddresspool__ipnamespace",
+            cardinality=Cardinality.ONE,
+            branch=BranchSupportType.AGNOSTIC,
+            optional=False,
+            order_weight=6000,
+        ),
     ],
-}
+)
 
-core_number_pool = {
-    "name": "NumberPool",
-    "namespace": "Core",
-    "description": "A pool of number resources",
-    "label": "Number Pool",
-    "include_in_menu": False,
-    "branch": BranchSupportType.AGNOSTIC.value,
-    "generate_profile": False,
-    "inherit_from": [InfrahubKind.RESOURCEPOOL, InfrahubKind.LINEAGESOURCE],
-    "attributes": [
-        {
-            "name": "node",
-            "kind": "Text",
-            "optional": False,
-            "description": "The model of the object that requires integers to be allocated",
-            "order_weight": 3000,
-        },
-        {
-            "name": "node_attribute",
-            "kind": "Text",
-            "description": "The attribute of the selected model",
-            "optional": False,
-            "order_weight": 4000,
-        },
-        {
-            "name": "start_range",
-            "kind": "Number",
-            "optional": False,
-            "description": "The start range for the pool",
-            "order_weight": 5000,
-        },
-        {
-            "name": "end_range",
-            "kind": "Number",
-            "optional": False,
-            "description": "The end range for the pool",
-            "order_weight": 6000,
-        },
+core_number_pool = NodeSchema(
+    name="NumberPool",
+    namespace="Core",
+    description="A pool of number resources",
+    label="Number Pool",
+    include_in_menu=False,
+    branch=BranchSupportType.AGNOSTIC,
+    generate_profile=False,
+    inherit_from=[InfrahubKind.RESOURCEPOOL, InfrahubKind.LINEAGESOURCE],
+    attributes=[
+        Attr(
+            name="node",
+            kind="Text",
+            optional=False,
+            description="The model of the object that requires integers to be allocated",
+            order_weight=3000,
+        ),
+        Attr(
+            name="node_attribute",
+            kind="Text",
+            description="The attribute of the selected model",
+            optional=False,
+            order_weight=4000,
+        ),
+        Attr(
+            name="start_range",
+            kind="Number",
+            optional=False,
+            description="The start range for the pool",
+            order_weight=5000,
+        ),
+        Attr(
+            name="end_range", kind="Number", optional=False, description="The end range for the pool", order_weight=6000
+        ),
     ],
-}
+)

infrahub/core/schema/definitions/core/template.py

@@ -1,12 +1,27 @@
-core_object_template = {
-    "name": "ObjectTemplate",
-    "namespace": "Core",
-    "include_in_menu": False,
-    "icon": "mdi:pencil-ruler",
-    "description": "Template to create pre-shaped objects.",
-    "label": "Object Templates",
-    "display_labels": ["template_name__value"],
-    "default_filter": "template_name__value",
-    "uniqueness_constraints": [["template_name__value"]],
-    "attributes": [{"name": "template_name", "kind": "Text", "optional": False, "unique": True, "order_weight": 1000}],
-}
+from ...attribute_schema import AttributeSchema as Attr
+from ...generic_schema import GenericSchema
+
+core_object_template = GenericSchema(
+    name="ObjectTemplate",
+    namespace="Core",
+    include_in_menu=False,
+    icon="mdi:pencil-ruler",
+    description="Template to create pre-shaped objects.",
+    label="Object Templates",
+    display_labels=["template_name__value"],
+    default_filter="template_name__value",
+    uniqueness_constraints=[["template_name__value"]],
+    attributes=[Attr(name="template_name", kind="Text", optional=False, unique=True, order_weight=1000)],
+)
+
+core_object_component_template = GenericSchema(
+    name="ObjectComponentTemplate",
+    namespace="Core",
+    include_in_menu=False,
+    icon="mdi:pencil-ruler",
+    description="Component template to create pre-shaped objects.",
+    label="Object Component Templates",
+    display_labels=["template_name__value"],
+    default_filter="template_name__value",
+    attributes=[Attr(name="template_name", kind="Text", optional=False, order_weight=1000)],
+)

infrahub/core/schema/definitions/core/transform.py

@@ -2,86 +2,95 @@ from infrahub.core.constants import (
     BranchSupportType,
     InfrahubKind,
 )
+from infrahub.core.constants import RelationshipCardinality as Cardinality
+from infrahub.core.constants import RelationshipKind as RelKind
 
-core_transform = {
-    "name": "Transformation",
-    "namespace": "Core",
-    "description": "Generic Transformation Object.",
-    "include_in_menu": False,
-    "icon": "mdi:cog-transfer",
-    "label": "Transformation",
-    "default_filter": "name__value",
-    "order_by": ["name__value"],
-    "display_labels": ["label__value"],
-    "branch": BranchSupportType.AWARE.value,
-    "documentation": "/topics/proposed-change",
-    "uniqueness_constraints": [["name__value"]],
-    "attributes": [
-        {"name": "name", "kind": "Text", "unique": True},
-        {"name": "label", "kind": "Text", "optional": True},
-        {"name": "description", "kind": "Text", "optional": True},
-        {"name": "timeout", "kind": "Number", "default_value": 10},
+from ...attribute_schema import AttributeSchema as Attr
+from ...generic_schema import GenericSchema
+from ...node_schema import NodeSchema
+from ...relationship_schema import (
+    RelationshipSchema as Rel,
+)
+
+core_transform = GenericSchema(
+    name="Transformation",
+    namespace="Core",
+    description="Generic Transformation Object.",
+    include_in_menu=False,
+    icon="mdi:cog-transfer",
+    label="Transformation",
+    default_filter="name__value",
+    order_by=["name__value"],
+    display_labels=["label__value"],
+    branch=BranchSupportType.AWARE,
+    documentation="/topics/proposed-change",
+    uniqueness_constraints=[["name__value"]],
+    attributes=[
+        Attr(name="name", kind="Text", unique=True),
+        Attr(name="label", kind="Text", optional=True),
+        Attr(name="description", kind="Text", optional=True),
+        Attr(name="timeout", kind="Number", default_value=10),
     ],
-    "relationships": [
-        {
-            "name": "query",
-            "peer": InfrahubKind.GRAPHQLQUERY,
-            "identifier": "graphql_query__transformation",
-            "kind": "Attribute",
-            "cardinality": "one",
-            "optional": False,
-        },
-        {
-            "name": "repository",
-            "peer": InfrahubKind.GENERICREPOSITORY,
-            "kind": "Attribute",
-            "cardinality": "one",
-            "identifier": "repository__transformation",
-            "optional": False,
-        },
-        {
-            "name": "tags",
-            "peer": InfrahubKind.TAG,
-            "kind": "Attribute",
-            "optional": True,
-            "cardinality": "many",
-        },
+    relationships=[
+        Rel(
+            name="query",
+            peer=InfrahubKind.GRAPHQLQUERY,
+            identifier="graphql_query__transformation",
+            kind=RelKind.ATTRIBUTE,
+            cardinality=Cardinality.ONE,
+            optional=False,
+        ),
+        Rel(
+            name="repository",
+            peer=InfrahubKind.GENERICREPOSITORY,
+            kind=RelKind.ATTRIBUTE,
+            cardinality=Cardinality.ONE,
+            identifier="repository__transformation",
+            optional=False,
+        ),
+        Rel(
+            name="tags",
+            peer=InfrahubKind.TAG,
+            kind=RelKind.ATTRIBUTE,
+            optional=True,
+            cardinality=Cardinality.MANY,
+        ),
     ],
-}
+)
 
-core_transform_jinja2 = {
-    "name": "TransformJinja2",
-    "namespace": "Core",
-    "description": "A file rendered from a Jinja2 template",
-    "include_in_menu": False,
-    "label": "Transform Jinja2",
-    "default_filter": "name__value",
-    "order_by": ["name__value"],
-    "display_labels": ["name__value"],
-    "inherit_from": [InfrahubKind.TRANSFORM],
-    "generate_profile": False,
-    "branch": BranchSupportType.AWARE.value,
-    "documentation": "/topics/transformation",
-    "attributes": [
-        {"name": "template_path", "kind": "Text"},
+core_transform_jinja2 = NodeSchema(
+    name="TransformJinja2",
+    namespace="Core",
+    description="A file rendered from a Jinja2 template",
+    include_in_menu=False,
+    label="Transform Jinja2",
+    default_filter="name__value",
+    order_by=["name__value"],
+    display_labels=["name__value"],
+    inherit_from=[InfrahubKind.TRANSFORM],
+    generate_profile=False,
+    branch=BranchSupportType.AWARE,
+    documentation="/topics/transformation",
+    attributes=[
+        Attr(name="template_path", kind="Text"),
     ],
-}
+)
 
-core_transform_python = {
-    "name": "TransformPython",
-    "namespace": "Core",
-    "description": "A transform function written in Python",
-    "include_in_menu": False,
-    "label": "Transform Python",
-    "default_filter": "name__value",
-    "order_by": ["name__value"],
-    "display_labels": ["name__value"],
-    "inherit_from": [InfrahubKind.TRANSFORM],
-    "generate_profile": False,
-    "branch": BranchSupportType.AWARE.value,
-    "documentation": "/topics/transformation",
-    "attributes": [
-        {"name": "file_path", "kind": "Text"},
-        {"name": "class_name", "kind": "Text"},
+core_transform_python = NodeSchema(
+    name="TransformPython",
+    namespace="Core",
+    description="A transform function written in Python",
+    include_in_menu=False,
+    label="Transform Python",
+    default_filter="name__value",
+    order_by=["name__value"],
+    display_labels=["name__value"],
+    inherit_from=[InfrahubKind.TRANSFORM],
+    generate_profile=False,
+    branch=BranchSupportType.AWARE,
+    documentation="/topics/transformation",
+    attributes=[
+        Attr(name="file_path", kind="Text"),
+        Attr(name="class_name", kind="Text"),
     ],
-}
+)