[feat] Added support for IndistinctRelationship

Author: Aryamanz29

pyatlan/generator/class_generator.py

@@ -1010,9 +1010,8 @@ def filter_attributes_of_custom_entity_type():
             generator.render_module(asset_info, EnumDefInfo.enum_def_info)
 
     for file in (ASSETS_RELATIONS_DIR).glob("*.py"):
-        # Remove all files in the core assets
-        # directory except `relationship_attributes.py`
-        if file.name == "relationship_attributes.py":
+        # Ignore non-generator files
+        if file.name in ("relationship_attributes.py", "indistinct_relationship.py"):
             continue
         file.unlink()
 

pyatlan/generator/templates/relations_init.jinja2

@@ -4,6 +4,7 @@ import lazy_loader as lazy
 
 __PYATLAN_ASSET_RELATIONS__ = {
     "relationship_attributes": ["RelationshipAttributes"],
+    "indistinct_relationship": ["IndistinctRelationship"],
     {% for relation in relation_def_infos -%}
     "{{ relation.relationship_def.name | to_snake_case }}": ["{{ relation.relationship_def.name | to_cls_name}}"],
     {% endfor %}

pyatlan/generator/templates/relations_mypy_init.jinja2

@@ -2,12 +2,14 @@
 
 __all__ = [
     "RelationshipAttributes",
+    "IndistinctRelationship",
     {% for relation in relation_def_infos -%}
     "{{ relation.relationship_def.name | to_cls_name}}",
     {% endfor %}
 ]
 
 from .relationship_attributes import RelationshipAttributes
+from .indistinct_relationship import IndistinctRelationship
 {% for relation in relation_def_infos -%}
 from .{{ relation.relationship_def.name | to_snake_case }} import {{ relation.relationship_def.name | to_cls_name }}
 {% endfor %}

pyatlan/model/assets/relations/__init__.py

@@ -4,6 +4,7 @@
 
 __PYATLAN_ASSET_RELATIONS__ = {
     "relationship_attributes": ["RelationshipAttributes"],
+    "indistinct_relationship": ["IndistinctRelationship"],
     "custom_parent_entity_custom_child_entities": [
         "CustomParentEntityCustomChildEntities"
     ],

pyatlan/model/assets/relations/__init__.pyi

@@ -2,6 +2,7 @@
 
 __all__ = [
     "RelationshipAttributes",
+    "IndistinctRelationship",
     "CustomParentEntityCustomChildEntities",
     "AtlasGlossaryAntonym",
     "AtlasGlossarySynonym",
@@ -33,5 +34,6 @@ from .custom_parent_entity_custom_child_entities import (
 from .custom_related_from_entities_custom_related_to_entities import (
     CustomRelatedFromEntitiesCustomRelatedToEntities,
 )
+from .indistinct_relationship import IndistinctRelationship
 from .relationship_attributes import RelationshipAttributes
 from .user_def_relationship import UserDefRelationship

pyatlan/model/assets/relations/indistinct_relationship.py

@@ -0,0 +1,18 @@
+from __future__ import annotations
+
+from typing import Any, Dict, Optional
+
+from pydantic.v1 import Field
+
+from .relationship_attributes import RelationshipAttributes
+
+
+class IndistinctRelationship(RelationshipAttributes):
+    type_name: str = Field(
+        default="IndistinctRelationship",
+        description="Name of the relationship type that defines the relationship.",
+    )
+    attributes: Optional[Dict[str, Any]] = Field(
+        default=None,
+        description="Map of attributes in the instance and their values",
+    )

pyatlan/model/assets/relations/relationship_attributes.py

@@ -33,6 +33,13 @@ def _parse_relationship_attributes(cls, data):
         relationship_attribute_cls = getattr(
             importlib.import_module("pyatlan.model.assets.relations"),
             to_python_class_name(type_name),
-            RelationshipAttributes,
+            None,
         )
+
+        # If relationship is not modeled in the SDK
+        # use IndistinctRelationship model for deserialization
+        if not relationship_attribute_cls:
+            from .indistinct_relationship import IndistinctRelationship
+
+            relationship_attribute_cls = IndistinctRelationship
         return relationship_attribute_cls(**data)

tests/unit/data/custom_relationships/indistinct_relationship_deserialization.json

@@ -0,0 +1,78 @@
+{
+    "typeName": "AtlasGlossaryTerm",
+    "attributes": {
+        "validValues": [
+            {
+                "guid": "color-red-guid",
+                "typeName": "AtlasGlossaryTerm",
+                "attributes": {
+                    "relationshipAttributes": {
+                        "typeName": "UnknownRelationship1",
+                        "attributes": {
+                            "description": "test description 11",
+                            "expression": "test expression 11",
+                            "status": "ACTIVE"
+                        }
+                    },
+                    "name": "red"
+                },
+                "uniqueAttributes": {
+                    "qualifiedName": "red@color-values"
+                }
+            },
+            {
+                "guid": "color-blue-guid",
+                "typeName": "AtlasGlossaryTerm",
+                "attributes": {
+                    "relationshipAttributes": {
+                        "typeName": "AtlasGlossaryValidValue",
+                        "attributes": {
+                            "description": "test description 12",
+                            "expression": "test expression 12",
+                            "status": "ACTIVE"
+                        }
+                    },
+                    "name": "blue"
+                },
+                "uniqueAttributes": {
+                    "qualifiedName": "blue@color-values"
+                }
+            }
+        ],
+        "validValuesFor": [
+            {
+                "guid": "parent-concept-guid",
+                "typeName": "AtlasGlossaryTerm",
+                "attributes": {
+                    "relationshipAttributes": {
+                        "typeName": "UnknownRelationship2",
+                        "attributes": {
+                            "description": "test description 21",
+                            "alias": "test alias 21",
+                            "status": "ACTIVE"
+                        }
+                    },
+                    "name": "visual-attributes"
+                },
+                "uniqueAttributes": {
+                    "qualifiedName": "visual-attributes@business-glossary"
+                }
+            }
+        ],
+        "qualifiedName": "color@business-glossary",
+        "name": "color"
+    },
+    "guid": "color-term-guid",
+    "status": "ACTIVE",
+    "displayText": "color",
+    "classificationNames": [],
+    "classifications": [],
+    "meaningNames": [],
+    "meanings": [],
+    "isIncomplete": false,
+    "labels": [],
+    "createdBy": "data-modeler",
+    "updatedBy": "data-modeler",
+    "createTime": 1750078015371,
+    "updateTime": 1750156224299
+}