Merge pull request #862 from atlanhq/feat/add-to-atlas-dict-method

feat(model): add to_atlas_dict + mode_report on ModeChart

Author: Aryamanz29

pyatlan/generator/templates/referenceable_methods.jinja2

@@ -58,6 +58,15 @@
             self.business_attributes = business_attrs
         return super().json(**kwargs)
 
+    def to_atlas_dict(self) -> "Dict[str, Any]":
+        """Return an Atlas-compatible dict for use with application-sdk JsonFileWriter.
+
+        Produces the standard Atlas entity format with camelCase keys and no None values::
+
+            {"typeName": "...", "attributes": {"qualifiedName": "...", ...}}
+        """
+        return loads(self.json(by_alias=True, exclude_none=True))
+
     def validate_required(self):
         if not self.create_time or self.created_by:
             self.attributes.validate_required()

pyatlan/model/assets/core/referenceable.py

@@ -92,6 +92,15 @@ async def json_async(self, *args, **kwargs) -> str:
             self.business_attributes = business_attrs
         return super().json(**kwargs)
 
+    def to_atlas_dict(self) -> "Dict[str, Any]":
+        """Return an Atlas-compatible dict for use with application-sdk JsonFileWriter.
+
+        Produces the standard Atlas entity format with camelCase keys and no None values::
+
+            {"typeName": "...", "attributes": {"qualifiedName": "...", ...}}
+        """
+        return loads(self.json(by_alias=True, exclude_none=True))
+
     def validate_required(self):
         if not self.create_time or self.created_by:
             self.attributes.validate_required()

pyatlan/model/assets/mode_chart.py

@@ -41,9 +41,15 @@ def __setattr__(self, name, value):
     TBC
     """
 
+    MODE_REPORT: ClassVar[RelationField] = RelationField("modeReport")
+    """
+    Report to which this chart belongs.
+    """
+
     _convenience_properties: ClassVar[List[str]] = [
         "mode_chart_type",
         "mode_query",
+        "mode_report",
     ]
 
     @property
@@ -66,11 +72,24 @@ def mode_query(self, mode_query: Optional[ModeQuery]):
             self.attributes = self.Attributes()
         self.attributes.mode_query = mode_query
 
+    @property
+    def mode_report(self) -> Optional[ModeReport]:
+        return None if self.attributes is None else self.attributes.mode_report
+
+    @mode_report.setter
+    def mode_report(self, mode_report: Optional[ModeReport]):
+        if self.attributes is None:
+            self.attributes = self.Attributes()
+        self.attributes.mode_report = mode_report
+
     class Attributes(Mode.Attributes):
         mode_chart_type: Optional[str] = Field(default=None, description="")
         mode_query: Optional[ModeQuery] = Field(
             default=None, description=""
         )  # relationship
+        mode_report: Optional[ModeReport] = Field(
+            default=None, description=""
+        )  # relationship
 
     attributes: ModeChart.Attributes = Field(
         default_factory=lambda: ModeChart.Attributes(),
@@ -83,5 +102,6 @@ class Attributes(Mode.Attributes):
 
 
 from .mode_query import ModeQuery  # noqa: E402, F401
+from .mode_report import ModeReport  # noqa: E402, F401
 
 ModeChart.Attributes.update_forward_refs()