cartography 0.93.0rc1__py3-none-any.whl → 0.123.0__py3-none-any.whl

cartography/models/azure/storage/queue.py

@@ -0,0 +1,71 @@
+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 AzureStorageQueueProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("id")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+    type: PropertyRef = PropertyRef("type")
+    name: PropertyRef = PropertyRef("name")
+
+
+@dataclass(frozen=True)
+class AzureStorageQueueToStorageQueueServiceProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AzureStorageQueueService)-[:CONTAINS]->(:AzureStorageQueue)
+class AzureStorageQueueToStorageQueueServiceRel(CartographyRelSchema):
+    target_node_label: str = "AzureStorageQueueService"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("service_id")},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "CONTAINS"
+    properties: AzureStorageQueueToStorageQueueServiceProperties = (
+        AzureStorageQueueToStorageQueueServiceProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AzureStorageQueueToSubscriptionRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AzureSubscription)-[:RESOURCE]->(:AzureStorageQueue)
+class AzureStorageQueueToSubscriptionRel(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: AzureStorageQueueToSubscriptionRelProperties = (
+        AzureStorageQueueToSubscriptionRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AzureStorageQueueSchema(CartographyNodeSchema):
+    label: str = "AzureStorageQueue"
+    properties: AzureStorageQueueProperties = AzureStorageQueueProperties()
+    sub_resource_relationship: AzureStorageQueueToSubscriptionRel = (
+        AzureStorageQueueToSubscriptionRel()
+    )
+    other_relationships: OtherRelationships = OtherRelationships(
+        [
+            AzureStorageQueueToStorageQueueServiceRel(),
+        ]
+    )

cartography/models/azure/storage/queueservice.py

@@ -0,0 +1,73 @@
+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 AzureStorageQueueServiceProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("id")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+    type: PropertyRef = PropertyRef("type")
+    name: PropertyRef = PropertyRef("name")
+
+
+@dataclass(frozen=True)
+class AzureStorageQueueServiceToStorageAccountProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AzureStorageAccount)-[:USES]->(:AzureStorageQueueService)
+class AzureStorageQueueServiceToStorageAccountRel(CartographyRelSchema):
+    target_node_label: str = "AzureStorageAccount"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("storage_account_id")},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "USES"
+    properties: AzureStorageQueueServiceToStorageAccountProperties = (
+        AzureStorageQueueServiceToStorageAccountProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AzureStorageQueueServiceToSubscriptionRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AzureSubscription)-[:RESOURCE]->(:AzureStorageQueueService)
+class AzureStorageQueueServiceToSubscriptionRel(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: AzureStorageQueueServiceToSubscriptionRelProperties = (
+        AzureStorageQueueServiceToSubscriptionRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AzureStorageQueueServiceSchema(CartographyNodeSchema):
+    label: str = "AzureStorageQueueService"
+    properties: AzureStorageQueueServiceProperties = (
+        AzureStorageQueueServiceProperties()
+    )
+    sub_resource_relationship: AzureStorageQueueServiceToSubscriptionRel = (
+        AzureStorageQueueServiceToSubscriptionRel()
+    )
+    other_relationships: OtherRelationships = OtherRelationships(
+        [
+            AzureStorageQueueServiceToStorageAccountRel(),
+        ]
+    )

cartography/models/azure/storage/table.py

@@ -0,0 +1,72 @@
+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 AzureStorageTableProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("id")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+    type: PropertyRef = PropertyRef("type")
+    name: PropertyRef = PropertyRef("name")
+    tablename: PropertyRef = PropertyRef("table_name")
+
+
+@dataclass(frozen=True)
+class AzureStorageTableToStorageTableServiceProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AzureStorageTableService)-[:CONTAINS]->(:AzureStorageTable)
+class AzureStorageTableToStorageTableServiceRel(CartographyRelSchema):
+    target_node_label: str = "AzureStorageTableService"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("service_id")},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "CONTAINS"
+    properties: AzureStorageTableToStorageTableServiceProperties = (
+        AzureStorageTableToStorageTableServiceProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AzureStorageTableToSubscriptionRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AzureSubscription)-[:RESOURCE]->(:AzureStorageTable)
+class AzureStorageTableToSubscriptionRel(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: AzureStorageTableToSubscriptionRelProperties = (
+        AzureStorageTableToSubscriptionRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AzureStorageTableSchema(CartographyNodeSchema):
+    label: str = "AzureStorageTable"
+    properties: AzureStorageTableProperties = AzureStorageTableProperties()
+    sub_resource_relationship: AzureStorageTableToSubscriptionRel = (
+        AzureStorageTableToSubscriptionRel()
+    )
+    other_relationships: OtherRelationships = OtherRelationships(
+        [
+            AzureStorageTableToStorageTableServiceRel(),
+        ]
+    )

cartography/models/azure/storage/tableservice.py

@@ -0,0 +1,73 @@
+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 AzureStorageTableServiceProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("id")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+    type: PropertyRef = PropertyRef("type")
+    name: PropertyRef = PropertyRef("name")
+
+
+@dataclass(frozen=True)
+class AzureStorageTableServiceToStorageAccountProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AzureStorageAccount)-[:USES]->(:AzureStorageTableService)
+class AzureStorageTableServiceToStorageAccountRel(CartographyRelSchema):
+    target_node_label: str = "AzureStorageAccount"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("storage_account_id")},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "USES"
+    properties: AzureStorageTableServiceToStorageAccountProperties = (
+        AzureStorageTableServiceToStorageAccountProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AzureStorageTableServiceToSubscriptionRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AzureSubscription)-[:RESOURCE]->(:AzureStorageTableService)
+class AzureStorageTableServiceToSubscriptionRel(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: AzureStorageTableServiceToSubscriptionRelProperties = (
+        AzureStorageTableServiceToSubscriptionRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AzureStorageTableServiceSchema(CartographyNodeSchema):
+    label: str = "AzureStorageTableService"
+    properties: AzureStorageTableServiceProperties = (
+        AzureStorageTableServiceProperties()
+    )
+    sub_resource_relationship: AzureStorageTableServiceToSubscriptionRel = (
+        AzureStorageTableServiceToSubscriptionRel()
+    )
+    other_relationships: OtherRelationships = OtherRelationships(
+        [
+            AzureStorageTableServiceToStorageAccountRel(),
+        ]
+    )

cartography/models/azure/subnet.py

@@ -0,0 +1,101 @@
+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_source_node_matcher
+from cartography.models.core.relationships import make_target_node_matcher
+from cartography.models.core.relationships import OtherRelationships
+from cartography.models.core.relationships import SourceNodeMatcher
+from cartography.models.core.relationships import TargetNodeMatcher
+
+logger = logging.getLogger(__name__)
+
+
+@dataclass(frozen=True)
+class AzureSubnetProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("id")
+    name: PropertyRef = PropertyRef("name")
+    address_prefix: PropertyRef = PropertyRef("address_prefix")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class AzureSubnetToVNetRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class AzureSubnetToVNetRel(CartographyRelSchema):
+    target_node_label: str = "AzureVirtualNetwork"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("VNET_ID", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "CONTAINS"
+    properties: AzureSubnetToVNetRelProperties = AzureSubnetToVNetRelProperties()
+
+
+@dataclass(frozen=True)
+class AzureSubnetToNSGRelProperties(CartographyRelProperties):
+    """
+    The properties for the relationship from an AzureSubnet to an AzureNetworkSecurityGroup.
+    These must be defined so the loader can save them on the relationship.
+    """
+
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+    _sub_resource_id: PropertyRef = PropertyRef("_sub_resource_id", set_in_kwargs=True)
+    _sub_resource_label: PropertyRef = PropertyRef(
+        "_sub_resource_label", set_in_kwargs=True
+    )
+
+
+@dataclass(frozen=True)
+class AzureSubnetToNSGRel(CartographyRelSchema):
+    source_node_label: str = "AzureSubnet"
+    source_node_matcher: SourceNodeMatcher = make_source_node_matcher(
+        {"id": PropertyRef("NODE_ID")},
+    )
+    target_node_label: str = "AzureNetworkSecurityGroup"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("NSG_ID")},
+    )
+    direction: LinkDirection = LinkDirection.OUTWARD
+    rel_label: str = "ASSOCIATED_WITH"
+    properties: AzureSubnetToNSGRelProperties = AzureSubnetToNSGRelProperties()
+
+
+@dataclass(frozen=True)
+class AzureSubnetToSubscriptionRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class AzureSubnetToSubscriptionRel(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: AzureSubnetToSubscriptionRelProperties = (
+        AzureSubnetToSubscriptionRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AzureSubnetSchema(CartographyNodeSchema):
+    label: str = "AzureSubnet"
+    properties: AzureSubnetProperties = AzureSubnetProperties()
+    sub_resource_relationship: AzureSubnetToSubscriptionRel = (
+        AzureSubnetToSubscriptionRel()
+    )
+    other_relationships: OtherRelationships = OtherRelationships(
+        [
+            AzureSubnetToVNetRel(),
+        ],
+    )

cartography/models/azure/subscription.py

@@ -0,0 +1,47 @@
+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 AzureSubscriptionProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("subscriptionId")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+    path: PropertyRef = PropertyRef("id")
+    name: PropertyRef = PropertyRef("displayName")
+    state: PropertyRef = PropertyRef("state")
+
+
+@dataclass(frozen=True)
+class AzureSubscriptionToTenantProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AzureTenant)-[:RESOURCE]->(:AzureSubscription)
+class AzureSubscriptionToTenantRel(CartographyRelSchema):
+    target_node_label: str = "AzureTenant"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("TENANT_ID", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "RESOURCE"
+    properties: AzureSubscriptionToTenantProperties = (
+        AzureSubscriptionToTenantProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AzureSubscriptionSchema(CartographyNodeSchema):
+    label: str = "AzureSubscription"
+    properties: AzureSubscriptionProperties = AzureSubscriptionProperties()
+    sub_resource_relationship: AzureSubscriptionToTenantRel = (
+        AzureSubscriptionToTenantRel()
+    )

cartography/models/azure/tags/storage_tag.py

@@ -0,0 +1,40 @@
+from dataclasses import dataclass
+
+from cartography.models.azure.tags.tag import AzureTagProperties
+from cartography.models.azure.tags.tag import AzureTagToSubscriptionRel
+from cartography.models.core.common import PropertyRef
+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 StorageAccountToTagRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class StorageAccountToTagRel(CartographyRelSchema):
+    target_node_label: str = "AzureStorageAccount"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("resource_id")},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "TAGGED"
+    properties: StorageAccountToTagRelProperties = StorageAccountToTagRelProperties()
+
+
+@dataclass(frozen=True)
+class AzureStorageTagsSchema(CartographyNodeSchema):
+    label: str = "AzureTag"
+    properties: AzureTagProperties = AzureTagProperties()
+    sub_resource_relationship: AzureTagToSubscriptionRel = AzureTagToSubscriptionRel()
+    other_relationships: OtherRelationships = OtherRelationships(
+        [
+            StorageAccountToTagRel(),
+        ],
+    )

cartography/models/azure/tags/tag.py

@@ -0,0 +1,37 @@
+from dataclasses import dataclass
+
+from cartography.models.core.common import PropertyRef
+from cartography.models.core.nodes import CartographyNodeProperties
+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 AzureTagProperties(CartographyNodeProperties):
+    # The ID is a string: "{subscription_id}|{key}:{value}"
+    id: PropertyRef = PropertyRef("id")
+    key: PropertyRef = PropertyRef("key")
+    value: PropertyRef = PropertyRef("value")
+    subscription_id: PropertyRef = PropertyRef("subscription_id")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class AzureTagToSubscriptionRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class AzureTagToSubscriptionRel(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: AzureTagToSubscriptionRelProperties = (
+        AzureTagToSubscriptionRelProperties()
+    )

cartography/models/azure/tenant.py

@@ -0,0 +1,17 @@
+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
+
+
+@dataclass(frozen=True)
+class AzureTenantProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("id")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class AzureTenantSchema(CartographyNodeSchema):
+    label: str = "AzureTenant"
+    properties: AzureTenantProperties = AzureTenantProperties()

cartography/models/azure/virtual_network.py

@@ -0,0 +1,49 @@
+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__)
+
+
+@dataclass(frozen=True)
+class AzureVirtualNetworkProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("id")
+    name: PropertyRef = PropertyRef("name")
+    location: PropertyRef = PropertyRef("location")
+    provisioning_state: PropertyRef = PropertyRef("provisioning_state")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class AzureVirtualNetworkToSubscriptionRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class AzureVirtualNetworkToSubscriptionRel(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: AzureVirtualNetworkToSubscriptionRelProperties = (
+        AzureVirtualNetworkToSubscriptionRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AzureVirtualNetworkSchema(CartographyNodeSchema):
+    label: str = "AzureVirtualNetwork"
+    properties: AzureVirtualNetworkProperties = AzureVirtualNetworkProperties()
+    sub_resource_relationship: AzureVirtualNetworkToSubscriptionRel = (
+        AzureVirtualNetworkToSubscriptionRel()
+    )

cartography/models/azure/vm/datadisk.py

@@ -0,0 +1,80 @@
+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 AzureDataDiskProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("managed_disk.id")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+    name: PropertyRef = PropertyRef("name")
+    lun: PropertyRef = PropertyRef("lun")
+    vhd: PropertyRef = PropertyRef("vhd.uri")
+    image: PropertyRef = PropertyRef("image.uri")
+    size: PropertyRef = PropertyRef("disk_size_gb")
+    caching: PropertyRef = PropertyRef("caching")
+    createoption: PropertyRef = PropertyRef("create_option")
+    write_accelerator_enabled: PropertyRef = PropertyRef("write_accelerator_enabled")
+    managed_disk_storage_type: PropertyRef = PropertyRef(
+        "managed_disk.storage_account_type"
+    )
+
+
+@dataclass(frozen=True)
+class AzureDataDiskToVirtualMachineProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AzureVirtualMachine)-[:ATTACHED_TO]->(:AzureDataDisk)
+class AzureDataDiskToVirtualMachineRel(CartographyRelSchema):
+    target_node_label: str = "AzureVirtualMachine"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("vm_id")},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "ATTACHED_TO"
+    properties: AzureDataDiskToVirtualMachineProperties = (
+        AzureDataDiskToVirtualMachineProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AzureDataDiskToSubscriptionRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AzureSubscription)-[:RESOURCE]->(:AzureDataDisk)
+class AzureDataDiskToSubscriptionRel(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: AzureDataDiskToSubscriptionRelProperties = (
+        AzureDataDiskToSubscriptionRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AzureDataDiskSchema(CartographyNodeSchema):
+    label: str = "AzureDataDisk"
+    properties: AzureDataDiskProperties = AzureDataDiskProperties()
+    sub_resource_relationship: AzureDataDiskToSubscriptionRel = (
+        AzureDataDiskToSubscriptionRel()
+    )
+    other_relationships: OtherRelationships = OtherRelationships(
+        [
+            AzureDataDiskToVirtualMachineRel(),
+        ]
+    )