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

infrahub/core/schema/definitions/core/account.py

@@ -4,165 +4,168 @@ from infrahub.core.constants import (
     BranchSupportType,
     InfrahubKind,
 )
+from infrahub.core.constants import RelationshipCardinality as Cardinality
 
-core_account = {
-    "name": "Account",
-    "namespace": "Core",
-    "description": "User Account for Infrahub",
-    "include_in_menu": False,
-    "label": "Account",
-    "icon": "mdi:account",
-    "default_filter": "name__value",
-    "order_by": ["name__value"],
-    "display_labels": ["label__value"],
-    "generate_profile": False,
-    "branch": BranchSupportType.AGNOSTIC.value,
-    "inherit_from": [InfrahubKind.LINEAGEOWNER, InfrahubKind.LINEAGESOURCE, InfrahubKind.GENERICACCOUNT],
-}
+from ...attribute_schema import AttributeSchema as Attr
+from ...dropdown import DropdownChoice
+from ...generic_schema import GenericSchema
+from ...node_schema import NodeSchema
+from ...relationship_schema import (
+    RelationshipSchema as Rel,
+)
+
+core_account = NodeSchema(
+    name="Account",
+    namespace="Core",
+    description="User Account for Infrahub",
+    include_in_menu=False,
+    label="Account",
+    icon="mdi:account",
+    default_filter="name__value",
+    order_by=["name__value"],
+    display_labels=["label__value"],
+    generate_profile=False,
+    branch=BranchSupportType.AGNOSTIC,
+    inherit_from=[InfrahubKind.LINEAGEOWNER, InfrahubKind.LINEAGESOURCE, InfrahubKind.GENERICACCOUNT],
+)
 
-core_account_token = {
-    "name": "AccountToken",
-    "namespace": "Internal",
-    "description": "Token for User Account",
-    "include_in_menu": False,
-    "label": "Account Token",
-    "default_filter": "token__value",
-    "display_labels": ["token__value"],
-    "generate_profile": False,
-    "branch": BranchSupportType.AGNOSTIC.value,
-    "uniqueness_constraints": [["token__value"]],
-    "documentation": "/topics/auth",
-    "attributes": [
-        {"name": "name", "kind": "Text", "optional": True},
-        {"name": "token", "kind": "Text", "unique": True},
-        {"name": "expiration", "kind": "DateTime", "optional": True},
+core_account_token = NodeSchema(
+    name="AccountToken",
+    namespace="Internal",
+    description="Token for User Account",
+    include_in_menu=False,
+    label="Account Token",
+    default_filter="token__value",
+    display_labels=["token__value"],
+    generate_profile=False,
+    branch=BranchSupportType.AGNOSTIC,
+    uniqueness_constraints=[["token__value"]],
+    documentation="/topics/auth",
+    attributes=[
+        Attr(name="name", kind="Text", optional=True),
+        Attr(name="token", kind="Text", unique=True),
+        Attr(name="expiration", kind="DateTime", optional=True),
     ],
-    "relationships": [
-        {
-            "name": "account",
-            "peer": InfrahubKind.GENERICACCOUNT,
-            "optional": False,
-            "cardinality": "one",
-            "identifier": "account__token",
-        },
+    relationships=[
+        Rel(
+            name="account",
+            peer=InfrahubKind.GENERICACCOUNT,
+            optional=False,
+            cardinality=Cardinality.ONE,
+            identifier="account__token",
+        ),
     ],
-}
+)
 
-core_password_credential = {
-    "name": "PasswordCredential",
-    "namespace": "Core",
-    "description": "Username/Password based credential",
-    "include_in_menu": False,
-    "label": "Username / Password",
-    "generate_profile": False,
-    "branch": BranchSupportType.AGNOSTIC.value,
-    "inherit_from": [InfrahubKind.CREDENTIAL],
-    "attributes": [
-        {
-            "name": "username",
-            "kind": "Text",
-            "optional": True,
-            "branch": BranchSupportType.AGNOSTIC.value,
-            "order_weight": 6000,
-        },
-        {
-            "name": "password",
-            "kind": "Password",
-            "optional": True,
-            "branch": BranchSupportType.AGNOSTIC.value,
-            "order_weight": 7000,
-        },
+core_password_credential = NodeSchema(
+    name="PasswordCredential",
+    namespace="Core",
+    description="Username/Password based credential",
+    include_in_menu=False,
+    label="Username / Password",
+    generate_profile=False,
+    branch=BranchSupportType.AGNOSTIC,
+    inherit_from=[InfrahubKind.CREDENTIAL],
+    attributes=[
+        Attr(name="username", kind="Text", optional=True, branch=BranchSupportType.AGNOSTIC, order_weight=6000),
+        Attr(
+            name="password",
+            kind="Password",
+            optional=True,
+            branch=BranchSupportType.AGNOSTIC,
+            order_weight=7000,
+        ),
     ],
-}
+)
 
-core_refresh_token = {
-    "name": "RefreshToken",
-    "namespace": "Internal",
-    "description": "Refresh Token",
-    "include_in_menu": False,
-    "label": "Refresh Token",
-    "display_labels": [],
-    "generate_profile": False,
-    "branch": BranchSupportType.AGNOSTIC.value,
-    "attributes": [
-        {"name": "expiration", "kind": "DateTime", "optional": False},
+core_refresh_token = NodeSchema(
+    name="RefreshToken",
+    namespace="Internal",
+    description="Refresh Token",
+    include_in_menu=False,
+    label="Refresh Token",
+    display_labels=[],
+    generate_profile=False,
+    branch=BranchSupportType.AGNOSTIC,
+    attributes=[
+        Attr(name="expiration", kind="DateTime", optional=False),
     ],
-    "relationships": [
-        {
-            "name": "account",
-            "peer": InfrahubKind.GENERICACCOUNT,
-            "optional": False,
-            "cardinality": "one",
-            "identifier": "account__refreshtoken",
-        },
+    relationships=[
+        Rel(
+            name="account",
+            peer=InfrahubKind.GENERICACCOUNT,
+            optional=False,
+            cardinality=Cardinality.ONE,
+            identifier="account__refreshtoken",
+        ),
     ],
-}
+)
 
-core_credential = {
-    "name": "Credential",
-    "namespace": "Core",
-    "description": "A credential that could be referenced to access external services.",
-    "include_in_menu": False,
-    "label": "Credential",
-    "default_filter": "name__value",
-    "order_by": ["name__value"],
-    "display_labels": ["label__value"],
-    "icon": "mdi:key-variant",
-    "human_friendly_id": ["name__value"],
-    "branch": BranchSupportType.AGNOSTIC.value,
-    "uniqueness_constraints": [["name__value"]],
-    "documentation": "/topics/auth",
-    "attributes": [
-        {"name": "name", "kind": "Text", "unique": True, "order_weight": 1000},
-        {"name": "label", "kind": "Text", "optional": True, "order_weight": 2000},
-        {"name": "description", "kind": "Text", "optional": True, "order_weight": 3000},
+core_credential = GenericSchema(
+    name="Credential",
+    namespace="Core",
+    description="A credential that could be referenced to access external services.",
+    include_in_menu=False,
+    label="Credential",
+    default_filter="name__value",
+    order_by=["name__value"],
+    display_labels=["label__value"],
+    icon="mdi:key-variant",
+    human_friendly_id=["name__value"],
+    branch=BranchSupportType.AGNOSTIC,
+    uniqueness_constraints=[["name__value"]],
+    documentation="/topics/auth",
+    attributes=[
+        Attr(name="name", kind="Text", unique=True, order_weight=1000),
+        Attr(name="label", kind="Text", optional=True, order_weight=2000),
+        Attr(name="description", kind="Text", optional=True, order_weight=3000),
     ],
-}
+)
 
-core_generic_account = {
-    "name": "GenericAccount",
-    "namespace": "Core",
-    "description": "User Account for Infrahub",
-    "include_in_menu": False,
-    "label": "Account",
-    "icon": "mdi:account",
-    "default_filter": "name__value",
-    "order_by": ["name__value"],
-    "display_labels": ["label__value"],
-    "human_friendly_id": ["name__value"],
-    "branch": BranchSupportType.AGNOSTIC.value,
-    "documentation": "/topics/auth",
-    "uniqueness_constraints": [["name__value"]],
-    "attributes": [
-        {"name": "name", "kind": "Text", "unique": True},
-        {"name": "password", "kind": "HashedPassword", "unique": False},
-        {"name": "label", "kind": "Text", "optional": True},
-        {"name": "description", "kind": "Text", "optional": True},
-        {
-            "name": "account_type",
-            "kind": "Text",
-            "default_value": AccountType.USER.value,
-            "enum": AccountType.available_types(),
-        },
-        {
-            "name": "status",
-            "kind": "Dropdown",
-            "choices": [
-                {
-                    "name": AccountStatus.ACTIVE.value,
-                    "label": "Active",
-                    "description": "Account is allowed to login",
-                    "color": "#52be80",
-                },
-                {
-                    "name": AccountStatus.INACTIVE.value,
-                    "label": "Inactive",
-                    "description": "Account is not allowed to login",
-                    "color": "#e74c3c",
-                },
+core_generic_account = GenericSchema(
+    name="GenericAccount",
+    namespace="Core",
+    description="User Account for Infrahub",
+    include_in_menu=False,
+    label="Account",
+    icon="mdi:account",
+    default_filter="name__value",
+    order_by=["name__value"],
+    display_labels=["label__value"],
+    human_friendly_id=["name__value"],
+    branch=BranchSupportType.AGNOSTIC,
+    documentation="/topics/auth",
+    uniqueness_constraints=[["name__value"]],
+    attributes=[
+        Attr(name="name", kind="Text", unique=True),
+        Attr(name="password", kind="HashedPassword", unique=False),
+        Attr(name="label", kind="Text", optional=True),
+        Attr(name="description", kind="Text", optional=True),
+        Attr(
+            name="account_type",
+            kind="Text",
+            default_value=AccountType.USER.value,
+            enum=AccountType.available_types(),
+        ),
+        Attr(
+            name="status",
+            kind="Dropdown",
+            choices=[
+                DropdownChoice(
+                    name=AccountStatus.ACTIVE.value,
+                    label="Active",
+                    description="Account is allowed to login",
+                    color="#52be80",
+                ),
+                DropdownChoice(
+                    name=AccountStatus.INACTIVE.value,
+                    label="Inactive",
+                    description="Account is not allowed to login",
+                    color="#e74c3c",
+                ),
             ],
-            "default_value": AccountStatus.ACTIVE.value,
-        },
+            default_value=AccountStatus.ACTIVE.value,
+        ),
     ],
-    "relationships": [{"name": "tokens", "peer": InfrahubKind.ACCOUNTTOKEN, "optional": True, "cardinality": "many"}],
-}
+    relationships=[Rel(name="tokens", peer=InfrahubKind.ACCOUNTTOKEN, optional=True, cardinality=Cardinality.MANY)],
+)

infrahub/core/schema/definitions/core/artifact.py

@@ -4,124 +4,133 @@ from infrahub.core.constants import (
     ContentType,
     InfrahubKind,
 )
+from infrahub.core.constants import RelationshipCardinality as Cardinality
+from infrahub.core.constants import RelationshipKind as RelKind
 
-core_artifact_target = {
-    "name": "ArtifactTarget",
-    "include_in_menu": False,
-    "namespace": "Core",
-    "description": "Extend a node to be associated with artifacts",
-    "label": "Artifact Target",
-    "relationships": [
-        {
-            "name": "artifacts",
-            "peer": InfrahubKind.ARTIFACT,
-            "optional": True,
-            "cardinality": "many",
-            "kind": "Generic",
-            "identifier": "artifact__node",
-        },
+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_artifact_target = GenericSchema(
+    name="ArtifactTarget",
+    include_in_menu=False,
+    namespace="Core",
+    description="Extend a node to be associated with artifacts",
+    label="Artifact Target",
+    relationships=[
+        Rel(
+            name="artifacts",
+            peer=InfrahubKind.ARTIFACT,
+            optional=True,
+            cardinality=Cardinality.MANY,
+            kind=RelKind.GENERIC,
+            identifier="artifact__node",
+        ),
     ],
-}
+)
 
-core_artifact = {
-    "name": "Artifact",
-    "namespace": "Core",
-    "label": "Artifact",
-    "include_in_menu": False,
-    "icon": "mdi:file-document-outline",
-    "default_filter": "name__value",
-    "order_by": ["name__value"],
-    "display_labels": ["name__value"],
-    "branch": BranchSupportType.LOCAL.value,
-    "generate_profile": False,
-    "inherit_from": [InfrahubKind.TASKTARGET],
-    "documentation": "/topics/artifact",
-    "attributes": [
-        {"name": "name", "kind": "Text"},
-        {
-            "name": "status",
-            "kind": "Text",
-            "enum": ArtifactStatus.available_types(),
-        },
-        {
-            "name": "content_type",
-            "kind": "Text",
-            "enum": ContentType.available_types(),
-        },
-        {
-            "name": "checksum",
-            "kind": "Text",
-            "optional": True,
-        },
-        {
-            "name": "storage_id",
-            "kind": "Text",
-            "optional": True,
-            "description": "ID of the file in the object store",
-        },
-        {"name": "parameters", "kind": "JSON", "optional": True},
+core_artifact = NodeSchema(
+    name="Artifact",
+    namespace="Core",
+    label="Artifact",
+    include_in_menu=False,
+    icon="mdi:file-document-outline",
+    default_filter="name__value",
+    order_by=["name__value"],
+    display_labels=["name__value"],
+    branch=BranchSupportType.LOCAL,
+    generate_profile=False,
+    inherit_from=[InfrahubKind.TASKTARGET],
+    documentation="/topics/artifact",
+    attributes=[
+        Attr(name="name", kind="Text"),
+        Attr(
+            name="status",
+            kind="Text",
+            enum=ArtifactStatus.available_types(),
+        ),
+        Attr(
+            name="content_type",
+            kind="Text",
+            enum=ContentType.available_types(),
+        ),
+        Attr(
+            name="checksum",
+            kind="Text",
+            optional=True,
+        ),
+        Attr(
+            name="storage_id",
+            kind="Text",
+            optional=True,
+            description="ID of the file in the object store",
+        ),
+        Attr(
+            name="parameters",
+            kind="JSON",
+            optional=True,
+        ),
     ],
-    "relationships": [
-        {
-            "name": "object",
-            "peer": InfrahubKind.ARTIFACTTARGET,
-            "kind": "Attribute",
-            "identifier": "artifact__node",
-            "cardinality": "one",
-            "optional": False,
-        },
-        {
-            "name": "definition",
-            "peer": InfrahubKind.ARTIFACTDEFINITION,
-            "kind": "Attribute",
-            "identifier": "artifact__artifact_definition",
-            "cardinality": "one",
-            "optional": False,
-        },
+    relationships=[
+        Rel(
+            name="object",
+            peer=InfrahubKind.ARTIFACTTARGET,
+            kind=RelKind.ATTRIBUTE,
+            identifier="artifact__node",
+            cardinality=Cardinality.ONE,
+            optional=False,
+        ),
+        Rel(
+            name="definition",
+            peer=InfrahubKind.ARTIFACTDEFINITION,
+            kind=RelKind.ATTRIBUTE,
+            identifier="artifact__artifact_definition",
+            cardinality=Cardinality.ONE,
+            optional=False,
+        ),
     ],
-}
+)
 
-core_artifact_definition = {
-    "name": "ArtifactDefinition",
-    "namespace": "Core",
-    "include_in_menu": False,
-    "icon": "mdi:file-document-multiple-outline",
-    "label": "Artifact Definition",
-    "default_filter": "name__value",
-    "order_by": ["name__value"],
-    "display_labels": ["name__value"],
-    "branch": BranchSupportType.AWARE.value,
-    "generate_profile": False,
-    "uniqueness_constraints": [["name__value"]],
-    "inherit_from": [InfrahubKind.TASKTARGET],
-    "documentation": "/topics/artifact",
-    "attributes": [
-        {"name": "name", "kind": "Text", "unique": True},
-        {"name": "artifact_name", "kind": "Text"},
-        {"name": "description", "kind": "Text", "optional": True},
-        {"name": "parameters", "kind": "JSON"},
-        {
-            "name": "content_type",
-            "kind": "Text",
-            "enum": ContentType.available_types(),
-        },
+core_artifact_definition = NodeSchema(
+    name="ArtifactDefinition",
+    namespace="Core",
+    include_in_menu=False,
+    icon="mdi:file-document-multiple-outline",
+    label="Artifact Definition",
+    default_filter="name__value",
+    order_by=["name__value"],
+    display_labels=["name__value"],
+    branch=BranchSupportType.AWARE,
+    generate_profile=False,
+    uniqueness_constraints=[["name__value"]],
+    inherit_from=[InfrahubKind.TASKTARGET],
+    documentation="/topics/artifact",
+    attributes=[
+        Attr(name="name", kind="Text", unique=True),
+        Attr(name="artifact_name", kind="Text"),
+        Attr(name="description", kind="Text", optional=True),
+        Attr(name="parameters", kind="JSON"),
+        Attr(name="content_type", kind="Text", enum=ContentType.available_types()),
     ],
-    "relationships": [
-        {
-            "name": "targets",
-            "peer": InfrahubKind.GENERICGROUP,
-            "kind": "Attribute",
-            "identifier": "artifact_definition___group",
-            "cardinality": "one",
-            "optional": False,
-        },
-        {
-            "name": "transformation",
-            "peer": InfrahubKind.TRANSFORM,
-            "kind": "Attribute",
-            "identifier": "artifact_definition___transformation",
-            "cardinality": "one",
-            "optional": False,
-        },
+    relationships=[
+        Rel(
+            name="targets",
+            peer=InfrahubKind.GENERICGROUP,
+            kind=RelKind.ATTRIBUTE,
+            identifier="artifact_definition___group",
+            cardinality=Cardinality.ONE,
+            optional=False,
+        ),
+        Rel(
+            name="transformation",
+            peer=InfrahubKind.TRANSFORM,
+            kind=RelKind.ATTRIBUTE,
+            identifier="artifact_definition___transformation",
+            cardinality=Cardinality.ONE,
+            optional=False,
+        ),
     ],
-}
+)

infrahub/core/schema/definitions/core/builtin.py

@@ -2,20 +2,23 @@ from infrahub.core.constants import (
     BranchSupportType,
 )
 
-builtin_tag = {
-    "name": "Tag",
-    "namespace": "Builtin",
-    "description": "Standard Tag object to attached to other objects to provide some context.",
-    "include_in_menu": True,
-    "icon": "mdi:tag-multiple",
-    "label": "Tag",
-    "default_filter": "name__value",
-    "order_by": ["name__value"],
-    "display_labels": ["name__value"],
-    "branch": BranchSupportType.AWARE.value,
-    "uniqueness_constraints": [["name__value"]],
-    "attributes": [
-        {"name": "name", "kind": "Text", "unique": True},
-        {"name": "description", "kind": "Text", "optional": True},
+from ...attribute_schema import AttributeSchema as Attr
+from ...node_schema import NodeSchema
+
+builtin_tag = NodeSchema(
+    name="Tag",
+    namespace="Builtin",
+    description="Standard Tag object to attached to other objects to provide some context.",
+    include_in_menu=True,
+    icon="mdi:tag-multiple",
+    label="Tag",
+    default_filter="name__value",
+    order_by=["name__value"],
+    display_labels=["name__value"],
+    branch=BranchSupportType.AWARE,
+    uniqueness_constraints=[["name__value"]],
+    attributes=[
+        Attr(name="name", kind="Text", unique=True),
+        Attr(name="description", kind="Text", optional=True),
     ],
-}
+)