cartography 0.103.0rc1__py3-none-any.whl → 0.104.0__py3-none-any.whl

cartography/models/aws/ssm/parameters.py

@@ -0,0 +1,84 @@
+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 SSMParameterNodeProperties(CartographyNodeProperties):
+
+    arn: PropertyRef = PropertyRef("ARN", extra_index=True)
+    id: PropertyRef = PropertyRef("ARN")
+    name: PropertyRef = PropertyRef("Name")
+    description: PropertyRef = PropertyRef("Description")
+    type: PropertyRef = PropertyRef("Type")
+    keyid: PropertyRef = PropertyRef("KeyId")
+    kms_key_id_short: PropertyRef = PropertyRef("KMSKeyIdShort")
+    version: PropertyRef = PropertyRef("Version")
+    lastmodifieddate: PropertyRef = PropertyRef("LastModifiedDate")
+    tier: PropertyRef = PropertyRef("Tier")
+    lastmodifieduser: PropertyRef = PropertyRef("LastModifiedUser")
+    datatype: PropertyRef = PropertyRef("DataType")
+    allowedpattern: PropertyRef = PropertyRef("AllowedPattern")
+    policies_json: PropertyRef = PropertyRef("PoliciesJson")
+    region: PropertyRef = PropertyRef("Region", set_in_kwargs=True)
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class SSMParameterToAWSAccountRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class SSMParameterToAWSAccountRel(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: SSMParameterToAWSAccountRelProperties = (
+        SSMParameterToAWSAccountRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class SSMParameterToKMSKeyRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class SSMParameterToKMSKeyRel(CartographyRelSchema):
+    target_node_label: str = "KMSKey"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {
+            "id": PropertyRef("KMSKeyIdShort"),
+        }
+    )
+    direction: LinkDirection = LinkDirection.OUTWARD
+    rel_label: str = "ENCRYPTED_BY"
+    properties: SSMParameterToKMSKeyRelProperties = SSMParameterToKMSKeyRelProperties()
+
+
+@dataclass(frozen=True)
+class SSMParameterSchema(CartographyNodeSchema):
+
+    label: str = "SSMParameter"
+    properties: SSMParameterNodeProperties = SSMParameterNodeProperties()
+    sub_resource_relationship: SSMParameterToAWSAccountRel = (
+        SSMParameterToAWSAccountRel()
+    )
+
+    other_relationships: OtherRelationships = OtherRelationships(
+        [
+            SSMParameterToKMSKeyRel(),
+        ],
+    )

cartography/models/cloudflare/account.py

@@ -0,0 +1,25 @@
+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 CloudflareAccountNodeProperties(CartographyNodeProperties):
+    created_on: PropertyRef = PropertyRef("created_on")
+    name: PropertyRef = PropertyRef("name")
+    abuse_contact_email: PropertyRef = PropertyRef("settings.abuse_contact_email")
+    default_nameservers: PropertyRef = PropertyRef("settings.default_nameservers")
+    enforce_twofactor: PropertyRef = PropertyRef("settings.enforce_twofactor")
+    use_account_custom_ns_by_default: PropertyRef = PropertyRef(
+        "settings.use_account_custom_ns_by_default"
+    )
+    id: PropertyRef = PropertyRef("id")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class CloudflareAccountSchema(CartographyNodeSchema):
+    label: str = "CloudflareAccount"
+    properties: CloudflareAccountNodeProperties = CloudflareAccountNodeProperties()

cartography/models/cloudflare/dnsrecord.py

@@ -0,0 +1,55 @@
+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
+
+
+@dataclass(frozen=True)
+class CloudflareDNSRecordNodeProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("id")
+    name: PropertyRef = PropertyRef("name", extra_index=True)
+    value: PropertyRef = PropertyRef("content")
+    type: PropertyRef = PropertyRef("type")
+    comment: PropertyRef = PropertyRef("comment")
+    proxied: PropertyRef = PropertyRef("proxied")
+    ttl: PropertyRef = PropertyRef("ttl")
+    created_on: PropertyRef = PropertyRef("created_on")
+    modified_on: PropertyRef = PropertyRef("modified_on")
+    proxiable: PropertyRef = PropertyRef("proxiable")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class CloudflareDNSRecordToZoneRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:CloudflareDNSRecord)<-[:RESOURCE]-(:CloudflareZone)
+class CloudflareDNSRecordToZoneRel(CartographyRelSchema):
+    target_node_label: str = "CloudflareZone"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("zone_id", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "RESOURCE"
+    properties: CloudflareDNSRecordToZoneRelProperties = (
+        CloudflareDNSRecordToZoneRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class CloudflareDNSRecordSchema(CartographyNodeSchema):
+    label: str = "CloudflareDNSRecord"
+    properties: CloudflareDNSRecordNodeProperties = CloudflareDNSRecordNodeProperties()
+    extra_node_labels: ExtraNodeLabels = ExtraNodeLabels(["DNSRecord"])
+    sub_resource_relationship: CloudflareDNSRecordToZoneRel = (
+        CloudflareDNSRecordToZoneRel()
+    )

cartography/models/cloudflare/member.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.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 CloudflareMemberNodeProperties(CartographyNodeProperties):
+    status: PropertyRef = PropertyRef("status")
+    email: PropertyRef = PropertyRef("user.email")
+    firstname: PropertyRef = PropertyRef("user.first_name")
+    user_id: PropertyRef = PropertyRef("user.id")
+    lastname: PropertyRef = PropertyRef("user.last_name")
+    two_factor_authentication_enabled: PropertyRef = PropertyRef(
+        "user.two_factor_authentication_enabled"
+    )
+    id: PropertyRef = PropertyRef("id")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class CloudflareMemberToAccountRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:CloudflareMember)<-[:RESOURCE]-(:CloudflareAccount)
+class CloudflareMemberToAccountRel(CartographyRelSchema):
+    target_node_label: str = "CloudflareAccount"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("account_id", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "RESOURCE"
+    properties: CloudflareMemberToAccountRelProperties = (
+        CloudflareMemberToAccountRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class CloudflareMemberToCloudflareRoleProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:CloudflareRole)<-[:HAS_ROLE]-(:CloudflareMember)
+class CloudflareMemberToCloudflareRoleRel(CartographyRelSchema):
+    target_node_label: str = "CloudflareRole"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {
+            "id": PropertyRef(
+                "roles_ids",
+                one_to_many=True,
+            )
+        },
+    )
+    direction: LinkDirection = LinkDirection.OUTWARD
+    rel_label: str = "HAS_ROLE"
+    properties: CloudflareMemberToCloudflareRoleProperties = (
+        CloudflareMemberToCloudflareRoleProperties()
+    )
+
+
+@dataclass(frozen=True)
+class CloudflareMemberSchema(CartographyNodeSchema):
+    label: str = "CloudflareMember"
+    properties: CloudflareMemberNodeProperties = CloudflareMemberNodeProperties()
+    sub_resource_relationship: CloudflareMemberToAccountRel = (
+        CloudflareMemberToAccountRel()
+    )
+    other_relationships: OtherRelationships = OtherRelationships(
+        [
+            CloudflareMemberToCloudflareRoleRel(),
+        ],
+    )

cartography/models/cloudflare/role.py

@@ -0,0 +1,44 @@
+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 CloudflareRoleNodeProperties(CartographyNodeProperties):
+    description: PropertyRef = PropertyRef("description")
+    name: PropertyRef = PropertyRef("name")
+    id: PropertyRef = PropertyRef("id")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class CloudflareRoleToAccountRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:CloudflareRole)<-[:RESOURCE]-(:CloudflareAccount)
+class CloudflareRoleToAccountRel(CartographyRelSchema):
+    target_node_label: str = "CloudflareAccount"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("account_id", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "RESOURCE"
+    properties: CloudflareRoleToAccountRelProperties = (
+        CloudflareRoleToAccountRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class CloudflareRoleSchema(CartographyNodeSchema):
+    label: str = "CloudflareRole"
+    properties: CloudflareRoleNodeProperties = CloudflareRoleNodeProperties()
+    sub_resource_relationship: CloudflareRoleToAccountRel = CloudflareRoleToAccountRel()

cartography/models/cloudflare/zone.py

@@ -0,0 +1,59 @@
+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 CloudflareZoneNodeProperties(CartographyNodeProperties):
+    activated_on: PropertyRef = PropertyRef("activated_on")
+    created_on: PropertyRef = PropertyRef("created_on")
+    development_mode: PropertyRef = PropertyRef("development_mode")
+    cdn_only: PropertyRef = PropertyRef("meta.cdn_only")
+    custom_certificate_quota: PropertyRef = PropertyRef("meta.custom_certificate_quota")
+    dns_only: PropertyRef = PropertyRef("meta.dns_only")
+    foundation_dns: PropertyRef = PropertyRef("meta.foundation_dns")
+    page_rule_quota: PropertyRef = PropertyRef("meta.page_rule_quota")
+    phishing_detected: PropertyRef = PropertyRef("meta.phishing_detected")
+    modified_on: PropertyRef = PropertyRef("modified_on")
+    name: PropertyRef = PropertyRef("name")
+    original_dnshost: PropertyRef = PropertyRef("original_dnshost")
+    original_registrar: PropertyRef = PropertyRef("original_registrar")
+    status: PropertyRef = PropertyRef("status")
+    verification_key: PropertyRef = PropertyRef("verification_key")
+    id: PropertyRef = PropertyRef("id")
+    paused: PropertyRef = PropertyRef("paused")
+    type: PropertyRef = PropertyRef("type")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class CloudflareZoneToAccountRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:CloudflareZone)<-[:RESOURCE]-(:CloudflareAccount)
+class CloudflareZoneToAccountRel(CartographyRelSchema):
+    target_node_label: str = "CloudflareAccount"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("account_id", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "RESOURCE"
+    properties: CloudflareZoneToAccountRelProperties = (
+        CloudflareZoneToAccountRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class CloudflareZoneSchema(CartographyNodeSchema):
+    label: str = "CloudflareZone"
+    properties: CloudflareZoneNodeProperties = CloudflareZoneNodeProperties()
+    sub_resource_relationship: CloudflareZoneToAccountRel = CloudflareZoneToAccountRel()

cartography/models/core/nodes.py

@@ -91,7 +91,7 @@ class CartographyNodeSchema(abc.ABC):
         """
         Optional.
         Allows subclasses to specify additional cartography relationships on the node.
-        :return: None if not overriden. Else return the node's OtherRelationships.
+        :return: None if not overridden. Else return the node's OtherRelationships.
         """
         return None
 
@@ -100,6 +100,19 @@ class CartographyNodeSchema(abc.ABC):
         """
         Optional.
         Allows specifying extra labels on the node.
-        :return: None if not overriden. Else return the ExtraNodeLabels specified on the node.
+        :return: None if not overridden. Else return the ExtraNodeLabels specified on the node.
         """
         return None
+
+    @property
+    def scoped_cleanup(self) -> bool:
+        """
+        Optional.
+        Allows specifying whether cleanups of this node must be scoped to the sub resource relationship.
+        If True (default), when we clean up nodes of this type, we will only delete stale nodes in the current sub
+        resource. This is how our AWS sync behaves.
+        If False, when we clean up node of this type, we will delete all stale nodes. This is designed for resource
+        types that don't have a "tenant"-like entity.
+        :return: True if not overridden. Else return the boolean value specified on the node.
+        """
+        return True

cartography/models/openai/adminapikey.py

@@ -0,0 +1,90 @@
+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 OpenAIAdminApiKeyNodeProperties(CartographyNodeProperties):
+    object: PropertyRef = PropertyRef("object")
+    id: PropertyRef = PropertyRef("id")
+    name: PropertyRef = PropertyRef("name")
+    created_at: PropertyRef = PropertyRef("created_at")
+    last_used_at: PropertyRef = PropertyRef("last_used_at")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class OpenAIAdminApiKeyToOrganizationRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:OpenAIOrganization)-[:RESOURCE]->(:OpenAIAdminApiKey)
+class OpenAIAdminApiKeyToOrganizationRel(CartographyRelSchema):
+    target_node_label: str = "OpenAIOrganization"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("ORG_ID", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "RESOURCE"
+    properties: OpenAIAdminApiKeyToOrganizationRelProperties = (
+        OpenAIAdminApiKeyToOrganizationRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class OpenAIAdminApiKeyToUserRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:OpenAIUser)-[:OWNS]->(:OpenAIAdminApiKey)
+class OpenAIAdminApiKeyToUserRel(CartographyRelSchema):
+    target_node_label: str = "OpenAIUser"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("owner_user_id")},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "OWNS"
+    properties: OpenAIAdminApiKeyToUserRelProperties = (
+        OpenAIAdminApiKeyToUserRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class OpenAIAdminApiKeyToSARelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:OpenAIServiceAccount)-[:OWNS]->(:OpenAIAdminApiKey)
+class OpenAIAdminApiKeyToSARel(CartographyRelSchema):
+    target_node_label: str = "OpenAIServiceAccount"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("owner_sa_id")},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "OWNS"
+    properties: OpenAIAdminApiKeyToSARelProperties = (
+        OpenAIAdminApiKeyToSARelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class OpenAIAdminApiKeySchema(CartographyNodeSchema):
+    label: str = "OpenAIAdminApiKey"
+    properties: OpenAIAdminApiKeyNodeProperties = OpenAIAdminApiKeyNodeProperties()
+    sub_resource_relationship: OpenAIAdminApiKeyToOrganizationRel = (
+        OpenAIAdminApiKeyToOrganizationRel()
+    )
+    other_relationships: OtherRelationships = OtherRelationships(
+        [OpenAIAdminApiKeyToUserRel(), OpenAIAdminApiKeyToSARel()],
+    )

cartography/models/openai/apikey.py

@@ -0,0 +1,84 @@
+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 OpenAIApiKeyNodeProperties(CartographyNodeProperties):
+    object: PropertyRef = PropertyRef("object")
+    name: PropertyRef = PropertyRef("name")
+    created_at: PropertyRef = PropertyRef("created_at")
+    last_used_at: PropertyRef = PropertyRef("last_used_at")
+    id: PropertyRef = PropertyRef("id")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class OpenAIApiKeyToProjectRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:OpenAIApiKey)<-[:RESOURCE]-(:OpenAIProject)
+class OpenAIApiKeyToProjectRel(CartographyRelSchema):
+    target_node_label: str = "OpenAIProject"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("project_id", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "RESOURCE"
+    properties: OpenAIApiKeyToProjectRelProperties = (
+        OpenAIApiKeyToProjectRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class OpenAIApiKeyToUserRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:OpenAIUser)-[:OWNS]->(:OpenAIApiKey)
+class OpenAIApiKeyToUserRel(CartographyRelSchema):
+    target_node_label: str = "OpenAIUser"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("owner_user_id")},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "OWNS"
+    properties: OpenAIApiKeyToUserRelProperties = OpenAIApiKeyToUserRelProperties()
+
+
+@dataclass(frozen=True)
+class OpenAIApiKeyToSARelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:OpenAIServiceAccount)-[:OWNS]->(:OpenAIApiKey)
+class OpenAIApiKeyToSARel(CartographyRelSchema):
+    target_node_label: str = "OpenAIServiceAccount"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("owner_sa_id")},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "OWNS"
+    properties: OpenAIApiKeyToSARelProperties = OpenAIApiKeyToSARelProperties()
+
+
+@dataclass(frozen=True)
+class OpenAIApiKeySchema(CartographyNodeSchema):
+    label: str = "OpenAIApiKey"
+    properties: OpenAIApiKeyNodeProperties = OpenAIApiKeyNodeProperties()
+    sub_resource_relationship: OpenAIApiKeyToProjectRel = OpenAIApiKeyToProjectRel()
+    other_relationships: OtherRelationships = OtherRelationships(
+        [OpenAIApiKeyToUserRel(), OpenAIApiKeyToSARel()],
+    )

cartography/models/openai/organization.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 OpenAIOrganizationNodeProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("id")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class OpenAIOrganizationSchema(CartographyNodeSchema):
+    label: str = "OpenAIOrganization"
+    properties: OpenAIOrganizationNodeProperties = OpenAIOrganizationNodeProperties()

cartography/models/openai/project.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 OpenAIProjectNodeProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("id")
+    object: PropertyRef = PropertyRef("object")
+    name: PropertyRef = PropertyRef("name")
+    created_at: PropertyRef = PropertyRef("created_at")
+    archived_at: PropertyRef = PropertyRef("archived_at")
+    status: PropertyRef = PropertyRef("status")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class OpenAIProjectToOrganizationRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:OpenAIOrganization)-[:RESOURCE]->(:OpenAIProject)
+class OpenAIProjectToOrganizationRel(CartographyRelSchema):
+    target_node_label: str = "OpenAIOrganization"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("ORG_ID", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "RESOURCE"
+    properties: OpenAIProjectToOrganizationRelProperties = (
+        OpenAIProjectToOrganizationRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class OpenAIProjectToUserRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:OpenAIUser)-[:MEMBER_OF]->(:OpenAIProject)
+class OpenAIProjectToUserRel(CartographyRelSchema):
+    target_node_label: str = "OpenAIUser"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("users", one_to_many=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "MEMBER_OF"
+    properties: OpenAIProjectToUserRelProperties = OpenAIProjectToUserRelProperties()
+
+
+@dataclass(frozen=True)
+class OpenAIProjectToUserAdminRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:OpenAIUser)-[:ADMIN_OF]->(:OpenAIProject)
+class OpenAIProjectToUserAdminRel(CartographyRelSchema):
+    target_node_label: str = "OpenAIUser"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("admins", one_to_many=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "ADMIN_OF"
+    properties: OpenAIProjectToUserAdminRelProperties = (
+        OpenAIProjectToUserAdminRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class OpenAIProjectSchema(CartographyNodeSchema):
+    label: str = "OpenAIProject"
+    properties: OpenAIProjectNodeProperties = OpenAIProjectNodeProperties()
+    sub_resource_relationship: OpenAIProjectToOrganizationRel = (
+        OpenAIProjectToOrganizationRel()
+    )
+    other_relationships: OtherRelationships = OtherRelationships(
+        [OpenAIProjectToUserRel(), OpenAIProjectToUserAdminRel()],
+    )