cartography 0.104.0rc2__py3-none-any.whl → 0.123.0__py3-none-any.whl

cartography/models/aws/sqs/queue.py

@@ -0,0 +1,89 @@
+from dataclasses import dataclass
+
+from cartography.models.core.common import PropertyRef
+from cartography.models.core.nodes import CartographyNodeProperties
+from cartography.models.core.nodes import CartographyNodeSchema
+from cartography.models.core.relationships import CartographyRelProperties
+from cartography.models.core.relationships import CartographyRelSchema
+from cartography.models.core.relationships import LinkDirection
+from cartography.models.core.relationships import make_target_node_matcher
+from cartography.models.core.relationships import OtherRelationships
+from cartography.models.core.relationships import TargetNodeMatcher
+
+
+@dataclass(frozen=True)
+class SQSQueueNodeProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("QueueArn")
+    arn: PropertyRef = PropertyRef("QueueArn", extra_index=True)
+    name: PropertyRef = PropertyRef("name")
+    url: PropertyRef = PropertyRef("url")
+    created_timestamp: PropertyRef = PropertyRef("CreatedTimestamp")
+    delay_seconds: PropertyRef = PropertyRef("DelaySeconds")
+    last_modified_timestamp: PropertyRef = PropertyRef("LastModifiedTimestamp")
+    maximum_message_size: PropertyRef = PropertyRef("MaximumMessageSize")
+    message_retention_period: PropertyRef = PropertyRef("MessageRetentionPeriod")
+    policy: PropertyRef = PropertyRef("Policy")
+    receive_message_wait_time_seconds: PropertyRef = PropertyRef(
+        "ReceiveMessageWaitTimeSeconds"
+    )
+    redrive_policy_dead_letter_target_arn: PropertyRef = PropertyRef(
+        "redrive_policy_dead_letter_target_arn"
+    )
+    redrive_policy_max_receive_count: PropertyRef = PropertyRef(
+        "redrive_policy_max_receive_count"
+    )
+    visibility_timeout: PropertyRef = PropertyRef("VisibilityTimeout")
+    kms_master_key_id: PropertyRef = PropertyRef("KmsMasterKeyId")
+    kms_data_key_reuse_period_seconds: PropertyRef = PropertyRef(
+        "KmsDataKeyReusePeriodSeconds"
+    )
+    fifo_queue: PropertyRef = PropertyRef("FifoQueue")
+    content_based_deduplication: PropertyRef = PropertyRef("ContentBasedDeduplication")
+    deduplication_scope: PropertyRef = PropertyRef("DeduplicationScope")
+    fifo_throughput_limit: PropertyRef = PropertyRef("FifoThroughputLimit")
+    region: PropertyRef = PropertyRef("Region", set_in_kwargs=True)
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class SQSQueueToAWSAccountRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class SQSQueueToAWSAccountRel(CartographyRelSchema):
+    target_node_label: str = "AWSAccount"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("AWS_ID", set_in_kwargs=True)}
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "RESOURCE"
+    properties: SQSQueueToAWSAccountRelProperties = SQSQueueToAWSAccountRelProperties()
+
+
+@dataclass(frozen=True)
+class SQSQueueToDeadLetterQueueRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class SQSQueueToDeadLetterQueueRel(CartographyRelSchema):
+    target_node_label: str = "SQSQueue"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("redrive_policy_dead_letter_target_arn")}
+    )
+    direction: LinkDirection = LinkDirection.OUTWARD
+    rel_label: str = "HAS_DEADLETTER_QUEUE"
+    properties: SQSQueueToDeadLetterQueueRelProperties = (
+        SQSQueueToDeadLetterQueueRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class SQSQueueSchema(CartographyNodeSchema):
+    label: str = "SQSQueue"
+    properties: SQSQueueNodeProperties = SQSQueueNodeProperties()
+    sub_resource_relationship: SQSQueueToAWSAccountRel = SQSQueueToAWSAccountRel()
+    other_relationships: OtherRelationships = OtherRelationships(
+        [SQSQueueToDeadLetterQueueRel()]
+    )

cartography/models/azure/aks_cluster.py

@@ -0,0 +1,54 @@
+import logging
+from dataclasses import dataclass
+
+from cartography.models.core.common import PropertyRef
+from cartography.models.core.nodes import CartographyNodeProperties
+from cartography.models.core.nodes import CartographyNodeSchema
+from cartography.models.core.relationships import CartographyRelProperties
+from cartography.models.core.relationships import CartographyRelSchema
+from cartography.models.core.relationships import LinkDirection
+from cartography.models.core.relationships import make_target_node_matcher
+from cartography.models.core.relationships import TargetNodeMatcher
+
+logger = logging.getLogger(__name__)
+
+
+# --- Node Definitions ---
+@dataclass(frozen=True)
+class AzureKubernetesClusterProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("id")
+    name: PropertyRef = PropertyRef("name")
+    location: PropertyRef = PropertyRef("location")
+    provisioning_state: PropertyRef = PropertyRef("provisioning_state")
+    kubernetes_version: PropertyRef = PropertyRef("kubernetes_version")
+    fqdn: PropertyRef = PropertyRef("fqdn")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+# --- Relationship Definitions ---
+@dataclass(frozen=True)
+class AzureKubernetesClusterToSubscriptionRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class AzureKubernetesClusterToSubscriptionRel(CartographyRelSchema):
+    target_node_label: str = "AzureSubscription"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("AZURE_SUBSCRIPTION_ID", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "RESOURCE"
+    properties: AzureKubernetesClusterToSubscriptionRelProperties = (
+        AzureKubernetesClusterToSubscriptionRelProperties()
+    )
+
+
+# --- Main Schema ---
+@dataclass(frozen=True)
+class AzureKubernetesClusterSchema(CartographyNodeSchema):
+    label: str = "AzureKubernetesCluster"
+    properties: AzureKubernetesClusterProperties = AzureKubernetesClusterProperties()
+    sub_resource_relationship: AzureKubernetesClusterToSubscriptionRel = (
+        AzureKubernetesClusterToSubscriptionRel()
+    )

cartography/models/azure/aks_nodepool.py

@@ -0,0 +1,54 @@
+import logging
+from dataclasses import dataclass
+
+from cartography.models.core.common import PropertyRef
+from cartography.models.core.nodes import CartographyNodeProperties
+from cartography.models.core.nodes import CartographyNodeSchema
+from cartography.models.core.relationships import CartographyRelProperties
+from cartography.models.core.relationships import CartographyRelSchema
+from cartography.models.core.relationships import LinkDirection
+from cartography.models.core.relationships import make_target_node_matcher
+from cartography.models.core.relationships import TargetNodeMatcher
+
+logger = logging.getLogger(__name__)
+
+
+# --- Node Definitions ---
+@dataclass(frozen=True)
+class AzureKubernetesNodePoolProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("id")
+    name: PropertyRef = PropertyRef("name")
+    provisioning_state: PropertyRef = PropertyRef("provisioning_state")
+    vm_size: PropertyRef = PropertyRef("vm_size")
+    os_type: PropertyRef = PropertyRef("os_type")
+    count: PropertyRef = PropertyRef("count")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+# --- Relationship Definitions ---
+@dataclass(frozen=True)
+class AzureKubernetesAgentPoolToClusterRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class AzureKubernetesAgentPoolToClusterRel(CartographyRelSchema):
+    target_node_label: str = "AzureKubernetesCluster"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("CLUSTER_ID", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "HAS_AGENT_POOL"
+    properties: AzureKubernetesAgentPoolToClusterRelProperties = (
+        AzureKubernetesAgentPoolToClusterRelProperties()
+    )
+
+
+# --- Main Schema ---
+@dataclass(frozen=True)
+class AzureKubernetesNodePoolSchema(CartographyNodeSchema):
+    label: str = "AzureKubernetesAgentPool"
+    properties: AzureKubernetesNodePoolProperties = AzureKubernetesNodePoolProperties()
+    sub_resource_relationship: AzureKubernetesAgentPoolToClusterRel = (
+        AzureKubernetesAgentPoolToClusterRel()
+    )

cartography/models/azure/app_service.py

@@ -0,0 +1,59 @@
+import logging
+from dataclasses import dataclass
+
+from cartography.models.core.common import PropertyRef
+from cartography.models.core.nodes import CartographyNodeProperties
+from cartography.models.core.nodes import CartographyNodeSchema
+from cartography.models.core.relationships import CartographyRelProperties
+from cartography.models.core.relationships import CartographyRelSchema
+from cartography.models.core.relationships import LinkDirection
+from cartography.models.core.relationships import make_target_node_matcher
+from cartography.models.core.relationships import TargetNodeMatcher
+
+logger = logging.getLogger(__name__)
+
+
+# --- Node Definitions ---
+@dataclass(frozen=True)
+class AzureAppServiceProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("id")
+    name: PropertyRef = PropertyRef("name")
+    kind: PropertyRef = PropertyRef("kind")
+    location: PropertyRef = PropertyRef("location")
+    state: PropertyRef = PropertyRef("state")
+    default_host_name: PropertyRef = PropertyRef("default_host_name")
+    https_only: PropertyRef = PropertyRef("https_only")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+# --- Relationship Definitions ---
+@dataclass(frozen=True)
+class AzureAppServiceToSubscriptionRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class AzureAppServiceToSubscriptionRel(CartographyRelSchema):
+    target_node_label: str = "AzureSubscription"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("AZURE_SUBSCRIPTION_ID", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "RESOURCE"
+    properties: AzureAppServiceToSubscriptionRelProperties = (
+        AzureAppServiceToSubscriptionRelProperties()
+    )
+
+
+# --- Main Schema ---
+@dataclass(frozen=True)
+class AzureAppServiceSchema(CartographyNodeSchema):
+    """
+    The schema for an Azure App Service.
+    """
+
+    label: str = "AzureAppService"
+    properties: AzureAppServiceProperties = AzureAppServiceProperties()
+    sub_resource_relationship: AzureAppServiceToSubscriptionRel = (
+        AzureAppServiceToSubscriptionRel()
+    )

cartography/models/azure/container_instance.py

@@ -0,0 +1,57 @@
+import logging
+from dataclasses import dataclass
+
+from cartography.models.core.common import PropertyRef
+from cartography.models.core.nodes import CartographyNodeProperties
+from cartography.models.core.nodes import CartographyNodeSchema
+from cartography.models.core.nodes import ExtraNodeLabels
+from cartography.models.core.relationships import CartographyRelProperties
+from cartography.models.core.relationships import CartographyRelSchema
+from cartography.models.core.relationships import LinkDirection
+from cartography.models.core.relationships import make_target_node_matcher
+from cartography.models.core.relationships import TargetNodeMatcher
+
+logger = logging.getLogger(__name__)
+
+
+# --- Node Definitions ---
+@dataclass(frozen=True)
+class AzureContainerInstanceProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("id")
+    name: PropertyRef = PropertyRef("name")
+    location: PropertyRef = PropertyRef("location")
+    type: PropertyRef = PropertyRef("type")
+    provisioning_state: PropertyRef = PropertyRef("provisioning_state")
+    ip_address: PropertyRef = PropertyRef("ip_address")
+    os_type: PropertyRef = PropertyRef("os_type")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+# --- Relationship Definitions ---
+@dataclass(frozen=True)
+class AzureContainerInstanceToSubscriptionRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class AzureContainerInstanceToSubscriptionRel(CartographyRelSchema):
+    target_node_label: str = "AzureSubscription"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("AZURE_SUBSCRIPTION_ID", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "RESOURCE"
+    properties: AzureContainerInstanceToSubscriptionRelProperties = (
+        AzureContainerInstanceToSubscriptionRelProperties()
+    )
+
+
+# --- Main Schema ---
+@dataclass(frozen=True)
+class AzureContainerInstanceSchema(CartographyNodeSchema):
+    label: str = "AzureContainerInstance"
+    extra_node_labels: ExtraNodeLabels = ExtraNodeLabels(["Container"])
+    properties: AzureContainerInstanceProperties = AzureContainerInstanceProperties()
+    sub_resource_relationship: AzureContainerInstanceToSubscriptionRel = (
+        AzureContainerInstanceToSubscriptionRel()
+    )

cartography/models/azure/cosmosdb/account.py

@@ -0,0 +1,77 @@
+from dataclasses import dataclass
+
+from cartography.models.core.common import PropertyRef
+from cartography.models.core.nodes import CartographyNodeProperties
+from cartography.models.core.nodes import CartographyNodeSchema
+from cartography.models.core.relationships import CartographyRelProperties
+from cartography.models.core.relationships import CartographyRelSchema
+from cartography.models.core.relationships import LinkDirection
+from cartography.models.core.relationships import make_target_node_matcher
+from cartography.models.core.relationships import TargetNodeMatcher
+
+
+@dataclass(frozen=True)
+class AzureCosmosDBAccountProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("id")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+    type: PropertyRef = PropertyRef("type")
+    resourcegroup: PropertyRef = PropertyRef("resourceGroup")
+    location: PropertyRef = PropertyRef("location")
+    kind: PropertyRef = PropertyRef("kind")
+    name: PropertyRef = PropertyRef("name")
+    ipranges: PropertyRef = PropertyRef("ipruleslist")
+    capabilities: PropertyRef = PropertyRef("capabilities")
+    documentendpoint: PropertyRef = PropertyRef("document_endpoint")
+    virtualnetworkfilterenabled: PropertyRef = PropertyRef(
+        "is_virtual_network_filter_enabled"
+    )
+    enableautomaticfailover: PropertyRef = PropertyRef("enable_automatic_failover")
+    provisioningstate: PropertyRef = PropertyRef("provisioning_state")
+    multiplewritelocations: PropertyRef = PropertyRef("enable_multiple_write_locations")
+    accountoffertype: PropertyRef = PropertyRef("database_account_offer_type")
+    publicnetworkaccess: PropertyRef = PropertyRef("public_network_access")
+    enablecassandraconnector: PropertyRef = PropertyRef("enable_cassandra_connector")
+    connectoroffer: PropertyRef = PropertyRef("connector_offer")
+    disablekeybasedmetadatawriteaccess: PropertyRef = PropertyRef(
+        "disable_key_based_metadata_write_access"
+    )
+    keyvaulturi: PropertyRef = PropertyRef("key_vault_key_uri")
+    enablefreetier: PropertyRef = PropertyRef("enable_free_tier")
+    enableanalyticalstorage: PropertyRef = PropertyRef("enable_analytical_storage")
+    defaultconsistencylevel: PropertyRef = PropertyRef(
+        "consistency_policy.default_consistency_level"
+    )
+    maxstalenessprefix: PropertyRef = PropertyRef(
+        "consistency_policy.max_staleness_prefix"
+    )
+    maxintervalinseconds: PropertyRef = PropertyRef(
+        "consistency_policy.max_interval_in_seconds"
+    )
+
+
+@dataclass(frozen=True)
+class AzureCosmosDBAccountToSubscriptionRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AzureSubscription)-[:RESOURCE]->(:AzureCosmosDBAccount)
+class AzureCosmosDBAccountToSubscriptionRel(CartographyRelSchema):
+    target_node_label: str = "AzureSubscription"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("AZURE_SUBSCRIPTION_ID", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "RESOURCE"
+    properties: AzureCosmosDBAccountToSubscriptionRelProperties = (
+        AzureCosmosDBAccountToSubscriptionRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AzureCosmosDBAccountSchema(CartographyNodeSchema):
+    label: str = "AzureCosmosDBAccount"
+    properties: AzureCosmosDBAccountProperties = AzureCosmosDBAccountProperties()
+    sub_resource_relationship: AzureCosmosDBAccountToSubscriptionRel = (
+        AzureCosmosDBAccountToSubscriptionRel()
+    )

cartography/models/azure/cosmosdb/accountfailoverpolicy.py

@@ -0,0 +1,77 @@
+from dataclasses import dataclass
+
+from cartography.models.core.common import PropertyRef
+from cartography.models.core.nodes import CartographyNodeProperties
+from cartography.models.core.nodes import CartographyNodeSchema
+from cartography.models.core.relationships import CartographyRelProperties
+from cartography.models.core.relationships import CartographyRelSchema
+from cartography.models.core.relationships import LinkDirection
+from cartography.models.core.relationships import make_target_node_matcher
+from cartography.models.core.relationships import OtherRelationships
+from cartography.models.core.relationships import TargetNodeMatcher
+
+
+@dataclass(frozen=True)
+class AzureCosmosDBAccountFailoverPolicyProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("id")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+    locationname: PropertyRef = PropertyRef("location_name")
+    failoverpriority: PropertyRef = PropertyRef("failover_priority")
+
+
+@dataclass(frozen=True)
+class AzureCosmosDBAccountFailoverPolicyToCosmosDBAccountRelProperties(
+    CartographyRelProperties
+):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AzureCosmosDBAccountFailoverPolicy)<-[:CONTAINS]-(:AzureCosmosDBAccount)
+class AzureCosmosDBAccountFailoverPolicyToCosmosDBAccountRel(CartographyRelSchema):
+    target_node_label: str = "AzureCosmosDBAccount"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("DatabaseAccountId", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "CONTAINS"
+    properties: AzureCosmosDBAccountFailoverPolicyToCosmosDBAccountRelProperties = (
+        AzureCosmosDBAccountFailoverPolicyToCosmosDBAccountRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AzureCosmosDBAccountFailoverPolicyToSubscriptionRelProperties(
+    CartographyRelProperties
+):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AzureSubscription)-[:RESOURCE]->(:AzureCosmosDBAccountFailoverPolicy)
+class AzureCosmosDBAccountFailoverPolicyToSubscriptionRel(CartographyRelSchema):
+    target_node_label: str = "AzureSubscription"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("AZURE_SUBSCRIPTION_ID", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "RESOURCE"
+    properties: AzureCosmosDBAccountFailoverPolicyToSubscriptionRelProperties = (
+        AzureCosmosDBAccountFailoverPolicyToSubscriptionRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AzureCosmosDBAccountFailoverPolicySchema(CartographyNodeSchema):
+    label: str = "AzureCosmosDBAccountFailoverPolicy"
+    properties: AzureCosmosDBAccountFailoverPolicyProperties = (
+        AzureCosmosDBAccountFailoverPolicyProperties()
+    )
+    sub_resource_relationship: AzureCosmosDBAccountFailoverPolicyToSubscriptionRel = (
+        AzureCosmosDBAccountFailoverPolicyToSubscriptionRel()
+    )
+    other_relationships: OtherRelationships = OtherRelationships(
+        [
+            AzureCosmosDBAccountFailoverPolicyToCosmosDBAccountRel(),
+        ]
+    )

cartography/models/azure/cosmosdb/cassandrakeyspace.py

@@ -0,0 +1,82 @@
+from dataclasses import dataclass
+
+from cartography.models.core.common import PropertyRef
+from cartography.models.core.nodes import CartographyNodeProperties
+from cartography.models.core.nodes import CartographyNodeSchema
+from cartography.models.core.nodes import ExtraNodeLabels
+from cartography.models.core.relationships import CartographyRelProperties
+from cartography.models.core.relationships import CartographyRelSchema
+from cartography.models.core.relationships import LinkDirection
+from cartography.models.core.relationships import make_target_node_matcher
+from cartography.models.core.relationships import OtherRelationships
+from cartography.models.core.relationships import TargetNodeMatcher
+
+
+@dataclass(frozen=True)
+class AzureCosmosDBCassandraKeyspaceProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("id")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+    location: PropertyRef = PropertyRef("location")
+    name: PropertyRef = PropertyRef("name")
+    type: PropertyRef = PropertyRef("type")
+    throughput: PropertyRef = PropertyRef("options.throughput")
+    maxthroughput: PropertyRef = PropertyRef("options.autoscale_setting.max_throughput")
+
+
+@dataclass(frozen=True)
+class AzureCosmosDBCassandraKeyspaceToCosmosDBAccountRelProperties(
+    CartographyRelProperties
+):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AzureCosmosDBAccount)-[:CONTAINS]->(:AzureCosmosDBCassandraKeyspace)
+class AzureCosmosDBCassandraKeyspaceToCosmosDBAccountRel(CartographyRelSchema):
+    target_node_label: str = "AzureCosmosDBAccount"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("database_account_id")},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "CONTAINS"
+    properties: AzureCosmosDBCassandraKeyspaceToCosmosDBAccountRelProperties = (
+        AzureCosmosDBCassandraKeyspaceToCosmosDBAccountRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AzureCosmosDBCassandraKeyspaceToSubscriptionRelProperties(
+    CartographyRelProperties
+):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AzureSubscription)-[:RESOURCE]->(:AzureCosmosDBCassandraKeyspace)
+class AzureCosmosDBCassandraKeyspaceToSubscriptionRel(CartographyRelSchema):
+    target_node_label: str = "AzureSubscription"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("AZURE_SUBSCRIPTION_ID", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "RESOURCE"
+    properties: AzureCosmosDBCassandraKeyspaceToSubscriptionRelProperties = (
+        AzureCosmosDBCassandraKeyspaceToSubscriptionRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AzureCosmosDBCassandraKeyspaceSchema(CartographyNodeSchema):
+    label: str = "AzureCosmosDBCassandraKeyspace"
+    extra_node_labels: ExtraNodeLabels = ExtraNodeLabels(["Database"])
+    properties: AzureCosmosDBCassandraKeyspaceProperties = (
+        AzureCosmosDBCassandraKeyspaceProperties()
+    )
+    sub_resource_relationship: AzureCosmosDBCassandraKeyspaceToSubscriptionRel = (
+        AzureCosmosDBCassandraKeyspaceToSubscriptionRel()
+    )
+    other_relationships: OtherRelationships = OtherRelationships(
+        [
+            AzureCosmosDBCassandraKeyspaceToCosmosDBAccountRel(),
+        ],
+    )

cartography/models/azure/cosmosdb/cassandratable.py

@@ -0,0 +1,81 @@
+from dataclasses import dataclass
+
+from cartography.models.core.common import PropertyRef
+from cartography.models.core.nodes import CartographyNodeProperties
+from cartography.models.core.nodes import CartographyNodeSchema
+from cartography.models.core.relationships import CartographyRelProperties
+from cartography.models.core.relationships import CartographyRelSchema
+from cartography.models.core.relationships import LinkDirection
+from cartography.models.core.relationships import make_target_node_matcher
+from cartography.models.core.relationships import OtherRelationships
+from cartography.models.core.relationships import TargetNodeMatcher
+
+
+@dataclass(frozen=True)
+class AzureCosmosDBCassandraTableProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("id")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+    type: PropertyRef = PropertyRef("type")
+    location: PropertyRef = PropertyRef("location")
+    name: PropertyRef = PropertyRef("name")
+    throughput: PropertyRef = PropertyRef("options.throughput")
+    maxthroughput: PropertyRef = PropertyRef("options.autoscale_setting.max_throughput")
+    container: PropertyRef = PropertyRef("resource.id")
+    defaultttl: PropertyRef = PropertyRef("resource.default_ttl")
+    analyticalttl: PropertyRef = PropertyRef("resource.analytical_storage_ttl")
+
+
+@dataclass(frozen=True)
+class AzureCosmosDBCassandraTableToCosmosDBCassandraKeyspaceProperties(
+    CartographyRelProperties
+):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AzureCosmosDBCassandraKeyspace)-[:CONTAINS]->(:AzureCosmosDBCassandraTable)
+class AzureCosmosDBCassandraTableToCosmosDBCassandraKeyspaceRel(CartographyRelSchema):
+    target_node_label: str = "AzureCosmosDBCassandraKeyspace"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("keyspace_id")},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "CONTAINS"
+    properties: AzureCosmosDBCassandraTableToCosmosDBCassandraKeyspaceProperties = (
+        AzureCosmosDBCassandraTableToCosmosDBCassandraKeyspaceProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AzureCosmosDBCassandraTableToSubscriptionRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AzureSubscription)-[:RESOURCE]->(:AzureCosmosDBCassandraTable)
+class AzureCosmosDBCassandraTableToSubscriptionRel(CartographyRelSchema):
+    target_node_label: str = "AzureSubscription"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("AZURE_SUBSCRIPTION_ID", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "RESOURCE"
+    properties: AzureCosmosDBCassandraTableToSubscriptionRelProperties = (
+        AzureCosmosDBCassandraTableToSubscriptionRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AzureCosmosDBCassandraTableSchema(CartographyNodeSchema):
+    label: str = "AzureCosmosDBCassandraTable"
+    properties: AzureCosmosDBCassandraTableProperties = (
+        AzureCosmosDBCassandraTableProperties()
+    )
+    sub_resource_relationship: AzureCosmosDBCassandraTableToSubscriptionRel = (
+        AzureCosmosDBCassandraTableToSubscriptionRel()
+    )
+    other_relationships: OtherRelationships = OtherRelationships(
+        [
+            AzureCosmosDBCassandraTableToCosmosDBCassandraKeyspaceRel(),
+        ]
+    )