# coding: utf-8
"""
Amorphic Data Platform
Amorphic Data Platform - API Definition documentation
The version of the OpenAPI document: 0.3.0
Generated by OpenAPI Generator (https://openapi-generator.tech)
Do not edit the class manually.
""" # noqa: E501
from __future__ import annotations
import pprint
import re # noqa: F401
import json
from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr
from typing import Any, ClassVar, Dict, List, Optional
from openapi_client.models.column_name_and_type import ColumnNameAndType
from typing import Optional, Set
from typing_extensions import Self
[docs]
class CompleteRegistration(BaseModel):
"""
CompleteRegistration
""" # noqa: E501
dataset_schema: List[ColumnNameAndType] = Field(alias="DatasetSchema")
record_keys: Optional[List[StrictStr]] = Field(default=None, alias="RecordKeys")
latest_record_indicator: Optional[ColumnNameAndType] = Field(default=None, alias="LatestRecordIndicator")
partition_keys: Optional[List[Dict[str, Any]]] = Field(default=None, alias="PartitionKeys")
sort_type: Optional[StrictStr] = Field(default=None, alias="SortType")
dist_type: Optional[StrictStr] = Field(default=None, alias="DistType")
sort_keys: Optional[List[StrictStr]] = Field(default=None, alias="SortKeys")
dist_key: Optional[StrictStr] = Field(default=None, alias="DistKey")
primary_keys: Optional[List[StrictStr]] = Field(default=None, alias="PrimaryKeys")
unique_columns: Optional[List[StrictStr]] = Field(default=None, alias="UniqueColumns")
is_identity_needed: Optional[StrictStr] = Field(default=None, alias="IsIdentityNeeded")
identity_attribute_values: Optional[List[StrictStr]] = Field(default=None, alias="IdentityAttributeValues")
data_conversion_params: Optional[Dict[str, Any]] = Field(default=None, alias="DataConversionParams")
partition_key: Optional[ColumnNameAndType] = Field(default=None, alias="PartitionKey")
sort_key: Optional[ColumnNameAndType] = Field(default=None, alias="SortKey")
local_secondary_indexes: Optional[List[Dict[str, Any]]] = Field(default=None, alias="LocalSecondaryIndexes")
global_secondary_indexes: Optional[List[Dict[str, Any]]] = Field(default=None, alias="GlobalSecondaryIndexes")
storage_type: Optional[StrictStr] = Field(default=None, alias="StorageType")
primary_key: Optional[StrictStr] = Field(default=None, alias="PrimaryKey")
precombine_key: Optional[StrictStr] = Field(default=None, alias="PrecombineKey")
dq_constraints: Optional[List[Dict[str, Any]]] = Field(default=None, alias="DqConstraints")
are_auto_constraint_suggestions_enabled: Optional[StrictBool] = Field(default=None, alias="AreAutoConstraintSuggestionsEnabled")
__properties: ClassVar[List[str]] = ["DatasetSchema", "RecordKeys", "LatestRecordIndicator", "PartitionKeys", "SortType", "DistType", "SortKeys", "DistKey", "PrimaryKeys", "UniqueColumns", "IsIdentityNeeded", "IdentityAttributeValues", "DataConversionParams", "PartitionKey", "SortKey", "LocalSecondaryIndexes", "GlobalSecondaryIndexes", "StorageType", "PrimaryKey", "PrecombineKey", "DqConstraints", "AreAutoConstraintSuggestionsEnabled"]
model_config = ConfigDict(
populate_by_name=True,
validate_assignment=True,
protected_namespaces=(),
)
[docs]
def to_str(self) -> str:
"""Returns the string representation of the model using alias"""
return pprint.pformat(self.model_dump(by_alias=True))
[docs]
def to_json(self) -> str:
"""Returns the JSON representation of the model using alias"""
# TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
return json.dumps(self.to_dict())
[docs]
@classmethod
def from_json(cls, json_str: str) -> Optional[Self]:
"""Create an instance of CompleteRegistration from a JSON string"""
return cls.from_dict(json.loads(json_str))
[docs]
def to_dict(self) -> Dict[str, Any]:
"""Return the dictionary representation of the model using alias.
This has the following differences from calling pydantic's
`self.model_dump(by_alias=True)`:
* `None` is only added to the output dict for nullable fields that
were set at model initialization. Other fields with value `None`
are ignored.
"""
excluded_fields: Set[str] = set([
])
_dict = self.model_dump(
by_alias=True,
exclude=excluded_fields,
exclude_none=True,
)
# override the default output from pydantic by calling `to_dict()` of each item in dataset_schema (list)
_items = []
if self.dataset_schema:
for _item_dataset_schema in self.dataset_schema:
if _item_dataset_schema:
_items.append(_item_dataset_schema.to_dict())
_dict['DatasetSchema'] = _items
# override the default output from pydantic by calling `to_dict()` of latest_record_indicator
if self.latest_record_indicator:
_dict['LatestRecordIndicator'] = self.latest_record_indicator.to_dict()
# override the default output from pydantic by calling `to_dict()` of partition_key
if self.partition_key:
_dict['PartitionKey'] = self.partition_key.to_dict()
# override the default output from pydantic by calling `to_dict()` of sort_key
if self.sort_key:
_dict['SortKey'] = self.sort_key.to_dict()
return _dict
[docs]
@classmethod
def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
"""Create an instance of CompleteRegistration from a dict"""
if obj is None:
return None
if not isinstance(obj, dict):
return cls.model_validate(obj)
_obj = cls.model_validate({
"DatasetSchema": [ColumnNameAndType.from_dict(_item) for _item in obj["DatasetSchema"]] if obj.get("DatasetSchema") is not None else None,
"RecordKeys": obj.get("RecordKeys"),
"LatestRecordIndicator": ColumnNameAndType.from_dict(obj["LatestRecordIndicator"]) if obj.get("LatestRecordIndicator") is not None else None,
"PartitionKeys": obj.get("PartitionKeys"),
"SortType": obj.get("SortType"),
"DistType": obj.get("DistType"),
"SortKeys": obj.get("SortKeys"),
"DistKey": obj.get("DistKey"),
"PrimaryKeys": obj.get("PrimaryKeys"),
"UniqueColumns": obj.get("UniqueColumns"),
"IsIdentityNeeded": obj.get("IsIdentityNeeded"),
"IdentityAttributeValues": obj.get("IdentityAttributeValues"),
"DataConversionParams": obj.get("DataConversionParams"),
"PartitionKey": ColumnNameAndType.from_dict(obj["PartitionKey"]) if obj.get("PartitionKey") is not None else None,
"SortKey": ColumnNameAndType.from_dict(obj["SortKey"]) if obj.get("SortKey") is not None else None,
"LocalSecondaryIndexes": obj.get("LocalSecondaryIndexes"),
"GlobalSecondaryIndexes": obj.get("GlobalSecondaryIndexes"),
"StorageType": obj.get("StorageType"),
"PrimaryKey": obj.get("PrimaryKey"),
"PrecombineKey": obj.get("PrecombineKey"),
"DqConstraints": obj.get("DqConstraints"),
"AreAutoConstraintSuggestionsEnabled": obj.get("AreAutoConstraintSuggestionsEnabled")
})
return _obj