feat(assets): add DynamoDBAttribute asset class

pyatlan/model/assets/__init__.py

@@ -430,6 +430,7 @@
     "azure_event_hub_consumer_group": ["AzureEventHubConsumerGroup"],
     "dynamo_d_b_local_secondary_index": ["DynamoDBLocalSecondaryIndex"],
     "dynamo_d_b_global_secondary_index": ["DynamoDBGlobalSecondaryIndex"],
+    "dynamo_d_b_attribute": ["DynamoDBAttribute"],
 }
 
 lazy_loader = lazy.attach(__name__, submod_attrs=__PYATLAN_ASSETS__)

pyatlan/model/assets/__init__.pyi

@@ -422,6 +422,7 @@ __all__ = [
     "AzureEventHubConsumerGroup",
     "DynamoDBLocalSecondaryIndex",
     "DynamoDBGlobalSecondaryIndex",
+    "DynamoDBAttribute",
     "IndistinctAsset",
 ]
 
@@ -688,6 +689,7 @@ from .dremio_source import DremioSource
 from .dremio_space import DremioSpace
 from .dremio_virtual_dataset import DremioVirtualDataset
 from .dynamo_d_b import DynamoDB
+from .dynamo_d_b_attribute import DynamoDBAttribute
 from .dynamo_d_b_global_secondary_index import DynamoDBGlobalSecondaryIndex
 from .dynamo_d_b_local_secondary_index import DynamoDBLocalSecondaryIndex
 from .dynamo_dbtable import DynamoDBTable

pyatlan/model/assets/dynamo_d_b_attribute.py

@@ -0,0 +1,303 @@
+# SPDX-License-Identifier: Apache-2.0
+# Copyright 2025 Atlan Pte. Ltd.
+
+
+from __future__ import annotations
+
+from typing import ClassVar, List, Optional
+
+from pydantic.v1 import Field, validator
+
+from pyatlan.model.enums import AtlanConnectorType, DynamoDBStatus
+from pyatlan.model.fields.atlan_fields import (
+    KeywordField,
+    NumericField,
+    RelationField,
+    TextField,
+)
+from pyatlan.utils import init_guid, validate_required_fields
+
+from .core.column import Column
+
+
+class DynamoDBAttribute(Column):
+    """Description"""
+
+    @classmethod
+    @init_guid
+    def creator(
+        cls,
+        *,
+        name: str,
+        parent_qualified_name: str,
+        order: int,
+        parent_name: Optional[str] = None,
+        connection_qualified_name: Optional[str] = None,
+    ) -> DynamoDBAttribute:
+        """
+        Builds the minimal object necessary to create a DynamoDBAttribute.
+
+        :param name: name of the DynamoDBAttribute
+        :param parent_qualified_name: unique name of the DynamoDBTable
+        in which this attribute exists
+        :param order: the order the attribute appears within its parent
+        :param parent_name: simple name of the DynamoDBTable
+        in which the attribute is contained
+        :param connection_qualified_name: unique name of the connection
+        in which the attribute should be created
+        :returns: the minimal request necessary to create the DynamoDBAttribute
+        """
+        return DynamoDBAttribute(
+            attributes=DynamoDBAttribute.Attributes.create(
+                name=name,
+                parent_qualified_name=parent_qualified_name,
+                order=order,
+                parent_name=parent_name,
+                connection_qualified_name=connection_qualified_name,
+            )
+        )
+
+    type_name: str = Field(default="DynamoDBAttribute", allow_mutation=False)
+
+    @validator("type_name")
+    def validate_type_name(cls, v):
+        if v != "DynamoDBAttribute":
+            raise ValueError("must be DynamoDBAttribute")
+        return v
+
+    def __setattr__(self, name, value):
+        if name in DynamoDBAttribute._convenience_properties:
+            return object.__setattr__(self, name, value)
+        super().__setattr__(name, value)
+
+    DYNAMO_DB_STATUS: ClassVar[KeywordField] = KeywordField(
+        "dynamoDBStatus", "dynamoDBStatus"
+    )
+    """
+    Status of the DynamoDB Asset
+    """
+    DYNAMO_DB_PARTITION_KEY: ClassVar[KeywordField] = KeywordField(
+        "dynamoDBPartitionKey", "dynamoDBPartitionKey"
+    )
+    """
+    Specifies the partition key of the DynamoDB Table/Index
+    """
+    DYNAMO_DB_SORT_KEY: ClassVar[KeywordField] = KeywordField(
+        "dynamoDBSortKey", "dynamoDBSortKey"
+    )
+    """
+    Specifies the sort key of the DynamoDB Table/Index
+    """
+    DYNAMO_DB_READ_CAPACITY_UNITS: ClassVar[NumericField] = NumericField(
+        "dynamoDBReadCapacityUnits", "dynamoDBReadCapacityUnits"
+    )
+    """
+    The maximum number of strongly consistent reads consumed per second before DynamoDB returns a ThrottlingException
+    """
+    DYNAMO_DB_WRITE_CAPACITY_UNITS: ClassVar[NumericField] = NumericField(
+        "dynamoDBWriteCapacityUnits", "dynamoDBWriteCapacityUnits"
+    )
+    """
+    The maximum number of writes consumed per second before DynamoDB returns a ThrottlingException
+    """
+    NO_SQL_SCHEMA_DEFINITION: ClassVar[TextField] = TextField(
+        "noSQLSchemaDefinition", "noSQLSchemaDefinition"
+    )
+    """
+    Represents attributes for describing the key schema for the table and indexes.
+    """
+
+    DYNAMO_DB_TABLE: ClassVar[RelationField] = RelationField("dynamoDBTable")
+    """
+    TBC
+    """
+
+    _convenience_properties: ClassVar[List[str]] = [
+        "dynamo_d_b_status",
+        "dynamo_d_b_partition_key",
+        "dynamo_d_b_sort_key",
+        "dynamo_d_b_read_capacity_units",
+        "dynamo_d_b_write_capacity_units",
+        "no_s_q_l_schema_definition",
+        "dynamo_dbtable",
+    ]
+
+    @property
+    def dynamo_d_b_status(self) -> Optional[DynamoDBStatus]:
+        return None if self.attributes is None else self.attributes.dynamo_d_b_status
+
+    @dynamo_d_b_status.setter
+    def dynamo_d_b_status(self, dynamo_d_b_status: Optional[DynamoDBStatus]):
+        if self.attributes is None:
+            self.attributes = self.Attributes()
+        self.attributes.dynamo_d_b_status = dynamo_d_b_status
+
+    @property
+    def dynamo_d_b_partition_key(self) -> Optional[str]:
+        return (
+            None
+            if self.attributes is None
+            else self.attributes.dynamo_d_b_partition_key
+        )
+
+    @dynamo_d_b_partition_key.setter
+    def dynamo_d_b_partition_key(self, dynamo_d_b_partition_key: Optional[str]):
+        if self.attributes is None:
+            self.attributes = self.Attributes()
+        self.attributes.dynamo_d_b_partition_key = dynamo_d_b_partition_key
+
+    @property
+    def dynamo_d_b_sort_key(self) -> Optional[str]:
+        return None if self.attributes is None else self.attributes.dynamo_d_b_sort_key
+
+    @dynamo_d_b_sort_key.setter
+    def dynamo_d_b_sort_key(self, dynamo_d_b_sort_key: Optional[str]):
+        if self.attributes is None:
+            self.attributes = self.Attributes()
+        self.attributes.dynamo_d_b_sort_key = dynamo_d_b_sort_key
+
+    @property
+    def dynamo_d_b_read_capacity_units(self) -> Optional[int]:
+        return (
+            None
+            if self.attributes is None
+            else self.attributes.dynamo_d_b_read_capacity_units
+        )
+
+    @dynamo_d_b_read_capacity_units.setter
+    def dynamo_d_b_read_capacity_units(
+        self, dynamo_d_b_read_capacity_units: Optional[int]
+    ):
+        if self.attributes is None:
+            self.attributes = self.Attributes()
+        self.attributes.dynamo_d_b_read_capacity_units = dynamo_d_b_read_capacity_units
+
+    @property
+    def dynamo_d_b_write_capacity_units(self) -> Optional[int]:
+        return (
+            None
+            if self.attributes is None
+            else self.attributes.dynamo_d_b_write_capacity_units
+        )
+
+    @dynamo_d_b_write_capacity_units.setter
+    def dynamo_d_b_write_capacity_units(
+        self, dynamo_d_b_write_capacity_units: Optional[int]
+    ):
+        if self.attributes is None:
+            self.attributes = self.Attributes()
+        self.attributes.dynamo_d_b_write_capacity_units = (
+            dynamo_d_b_write_capacity_units
+        )
+
+    @property
+    def no_s_q_l_schema_definition(self) -> Optional[str]:
+        return (
+            None
+            if self.attributes is None
+            else self.attributes.no_s_q_l_schema_definition
+        )
+
+    @no_s_q_l_schema_definition.setter
+    def no_s_q_l_schema_definition(self, no_s_q_l_schema_definition: Optional[str]):
+        if self.attributes is None:
+            self.attributes = self.Attributes()
+        self.attributes.no_s_q_l_schema_definition = no_s_q_l_schema_definition
+
+    @property
+    def dynamo_dbtable(self) -> Optional[DynamoDBTable]:
+        return None if self.attributes is None else self.attributes.dynamo_dbtable
+
+    @dynamo_dbtable.setter
+    def dynamo_dbtable(self, dynamo_dbtable: Optional[DynamoDBTable]):
+        if self.attributes is None:
+            self.attributes = self.Attributes()
+        self.attributes.dynamo_dbtable = dynamo_dbtable
+
+    class Attributes(Column.Attributes):
+        dynamo_d_b_status: Optional[DynamoDBStatus] = Field(
+            default=None, description=""
+        )
+        dynamo_d_b_partition_key: Optional[str] = Field(default=None, description="")
+        dynamo_d_b_sort_key: Optional[str] = Field(default=None, description="")
+        dynamo_d_b_read_capacity_units: Optional[int] = Field(
+            default=None, description=""
+        )
+        dynamo_d_b_write_capacity_units: Optional[int] = Field(
+            default=None, description=""
+        )
+        no_s_q_l_schema_definition: Optional[str] = Field(default=None, description="")
+        dynamo_dbtable: Optional[DynamoDBTable] = Field(
+            default=None, description=""
+        )  # relationship
+
+        @classmethod
+        @init_guid
+        def create(
+            cls,
+            *,
+            name: str,
+            parent_qualified_name: str,
+            order: int,
+            parent_name: Optional[str] = None,
+            connection_qualified_name: Optional[str] = None,
+        ) -> DynamoDBAttribute.Attributes:
+            """
+            Builds the minimal object necessary to create a DynamoDBAttribute.
+
+            :param name: name of the DynamoDBAttribute
+            :param parent_qualified_name: unique name of the DynamoDBTable
+            in which this attribute exists
+            :param order: the order the attribute appears within its parent
+            :param parent_name: simple name of the DynamoDBTable
+            in which the attribute is contained
+            :param connection_qualified_name: unique name of the connection
+            in which the attribute should be created
+            :returns: the minimal request necessary to create the DynamoDBAttribute
+            """
+            validate_required_fields(
+                ["name", "parent_qualified_name", "order"],
+                [name, parent_qualified_name, order],
+            )
+            if connection_qualified_name:
+                connector_name = AtlanConnectorType.get_connector_name(
+                    connection_qualified_name
+                )
+            else:
+                connection_qn, connector_name = AtlanConnectorType.get_connector_name(
+                    parent_qualified_name, "parent_qualified_name", 4
+                )
+            if order < 0:
+                raise ValueError("Order must be be a positive integer")
+
+            fields = parent_qualified_name.split("/")
+            qualified_name = f"{parent_qualified_name}/{name}"
+            connection_qualified_name = connection_qualified_name or connection_qn
+            parent_name = parent_name or fields[3]
+
+            return DynamoDBAttribute.Attributes(
+                name=name,
+                order=order,
+                qualified_name=qualified_name,
+                connector_name=connector_name,
+                connection_qualified_name=connection_qualified_name,
+                table_name=parent_name,
+                table_qualified_name=parent_qualified_name,
+                dynamo_dbtable=DynamoDBTable.ref_by_qualified_name(
+                    parent_qualified_name
+                ),
+            )
+
+    attributes: DynamoDBAttribute.Attributes = Field(
+        default_factory=lambda: DynamoDBAttribute.Attributes(),
+        description=(
+            "Map of attributes in the instance and their values. "
+            "The specific keys of this map will vary by type, "
+            "so are described in the sub-types of this schema."
+        ),
+    )
+
+
+from .dynamo_dbtable import DynamoDBTable  # noqa: E402, F401
+
+DynamoDBAttribute.Attributes.update_forward_refs()

pyatlan/model/assets/dynamo_dbtable.py

@@ -347,6 +347,10 @@ def __setattr__(self, name, value):
     """
     TBC
     """
+    DYNAMO_DB_COLUMNS: ClassVar[RelationField] = RelationField("dynamoDBColumns")
+    """
+    TBC
+    """
 
     _convenience_properties: ClassVar[List[str]] = [
         "dynamo_dbtable_g_s_i_count",
@@ -404,6 +408,7 @@ def __setattr__(self, name, value):
         "no_s_q_l_schema_definition",
         "dynamo_d_b_local_secondary_indexes",
         "dynamo_d_b_global_secondary_indexes",
+        "dynamo_d_b_columns",
     ]
 
     @property
@@ -1068,6 +1073,21 @@ def dynamo_d_b_global_secondary_indexes(
             dynamo_d_b_global_secondary_indexes
         )
 
+    @property
+    def dynamo_d_b_columns(
+        self,
+    ) -> Optional[List[DynamoDBAttribute]]:
+        return None if self.attributes is None else self.attributes.dynamo_d_b_columns
+
+    @dynamo_d_b_columns.setter
+    def dynamo_d_b_columns(
+        self,
+        dynamo_d_b_columns: Optional[List[DynamoDBAttribute]],
+    ):
+        if self.attributes is None:
+            self.attributes = self.Attributes()
+        self.attributes.dynamo_d_b_columns = dynamo_d_b_columns
+
     class Attributes(Table.Attributes):
         dynamo_dbtable_g_s_i_count: Optional[int] = Field(default=None, description="")
         dynamo_dbtable_l_s_i_count: Optional[int] = Field(default=None, description="")
@@ -1144,6 +1164,9 @@ class Attributes(Table.Attributes):
         dynamo_d_b_global_secondary_indexes: Optional[
             List[DynamoDBGlobalSecondaryIndex]
         ] = Field(default=None, description="")  # relationship
+        dynamo_d_b_columns: Optional[List[DynamoDBAttribute]] = Field(
+            default=None, description=""
+        )  # relationship
 
     attributes: DynamoDBTable.Attributes = Field(
         default_factory=lambda: DynamoDBTable.Attributes(),
@@ -1155,6 +1178,7 @@ class Attributes(Table.Attributes):
     )
 
 
+from .dynamo_d_b_attribute import DynamoDBAttribute  # noqa: E402, F401
 from .dynamo_d_b_global_secondary_index import (
     DynamoDBGlobalSecondaryIndex,  # noqa: E402, F401
 )