Merge pull request #607 from atlanhq/APP-6398

APP-6398: Allow extra fields in `AtlanYamlModel` (`Extra.allow`) to support undefined contract fields

Author: Aryamanz29

pyatlan/model/contract.py

@@ -120,7 +120,7 @@ class DCTag(AtlanYamlModel):
         name: Optional[str] = Field(
             default=None, description="Human-readable name of the Atlan tag."
         )
-        propagate: Optional[str] = Field(
+        propagate: Optional[Union[bool, str]] = Field(
             default=None, description="Whether to propagate the tag or not."
         )
         propagate_through_lineage: Optional[bool] = Field(
@@ -156,6 +156,10 @@ class DCColumn(AtlanYamlModel):
             default=None,
             description="When true, this column is the primary key for the table.",
         )
+        required: Optional[bool] = Field(
+            default=None,
+            description="When true, this column is required for the table.",
+        )
         data_type: Optional[str] = Field(
             default=None,
             description="Physical data type of values in this column (e.g. varchar(20)).",
@@ -188,7 +192,7 @@ class DCColumn(AtlanYamlModel):
             default_factory=list,
             description="Enumeration of values that should be considered missing.",
         )
-        not_null: Optional[bool] = Field(
+        not_null: Optional[Any] = Field(
             default=None, description="When true, this column cannot be empty."
         )
         valid_length: Optional[int] = Field(
@@ -205,6 +209,6 @@ class DCColumn(AtlanYamlModel):
             default=None,
             description="Minimum length for a string to be considered valid.",
         )
-        unique: Optional[bool] = Field(
+        unique: Optional[Any] = Field(
             default=None, description="When true, this column must have unique values."
         )

pyatlan/model/core.py

@@ -146,7 +146,9 @@ class AtlanYamlModel(BaseModel):
     """
 
     class Config:
-        extra = Extra.ignore
+        # Allow extra fields for contracts
+        # if they are not defined in the model
+        extra = Extra.allow
         validate_assignment = True
         allow_population_by_field_name = True
 

tests/integration/test_sql_assets.py

@@ -787,6 +787,11 @@ def test_contact_init_spec(self, client: AtlanClient):
             contract_spec
             and contract_spec.dataset  # type: ignore[union-attr]
             and contract_spec.dataset == TestTable.table.name  # type: ignore[union-attr]
+            # Ensure non-modeled fields are retained correctly
+            # (enabled by AtlanYamlModel → Config → Extra.allow)
+            and contract_spec.columns
+            and len(contract_spec.columns) >= 1
+            and contract_spec.columns[0].tags == contract_spec.columns[0].terms == []
         )
         self._assert_table_contract(contract_spec.to_yaml(), TestColumn, is_raw=False)  # type: ignore[union-attr]