fix(ingestion/powerbi): fix issue with broken report lineage

metadata-ingestion/src/datahub/emitter/mcp_patch_builder.py

@@ -71,6 +71,16 @@ def _add_patch(
         # TODO: Validate that aspectName is a valid aspect for this entityType
         self.patches[aspect_name].append(_Patch(op, path, value))
 
+    def add_patch(
+        self, aspect_name: str, op: str, path: Union[str, Sequence[str]], value: Any
+    ) -> None:
+        return self._add_patch(
+            aspect_name=aspect_name,
+            op=op,
+            path=path,
+            value=value,
+        )
+
     def build(self) -> Iterable[MetadataChangeProposalClass]:
         return [
             MetadataChangeProposalClass(

metadata-ingestion/src/datahub/ingestion/api/incremental_lineage_helper.py

@@ -1,4 +1,4 @@
-from typing import Iterable, Optional
+from typing import ClassVar, Iterable, List, Optional, Union
 
 from pydantic.fields import Field
 
@@ -11,6 +11,7 @@
     DashboardInfoClass,
     FineGrainedLineageClass,
     MetadataChangeEventClass,
+    MetadataChangeProposalClass,
     SystemMetadataClass,
     UpstreamLineageClass,
 )
@@ -19,6 +20,72 @@
 from datahub.specific.dataset import DatasetPatchBuilder
 
 
+class PatchEntityAspect:
+    SKIPPABLE_ATTRIBUTES: ClassVar[List[str]] = [
+        "ASPECT_INFO",
+        "ASPECT_NAME",
+        "ASPECT_TYPE",
+        "RECORD_SCHEMA",
+    ]
+    aspect: Union[ChartInfoClass, DashboardInfoClass]
+    patch_builder: DashboardPatchBuilder
+    attributes: List[str]
+
+    def __init__(
+        self,
+        # The PatchEntityAspect can patch any Aspect, however to silent the lint Union is added for DashboardInfoClass
+        # We can use it with any Aspect
+        aspect: Union[DashboardInfoClass],
+        patch_builder: DashboardPatchBuilder,
+    ):
+        self.aspect = aspect
+        self.patch_builder = patch_builder
+        self.attributes = dir(self.aspect)
+
+    def is_attribute_includable(self, attribute_name: str) -> bool:
+        """
+        a child class can override this to add additional attributes to skip while generating patch aspect
+        """
+        if (
+            attribute_name.startswith("__")
+            or attribute_name.startswith("_")
+            or attribute_name in PatchEntityAspect.SKIPPABLE_ATTRIBUTES
+        ):
+            return False
+
+        return True
+
+    def attribute_path(self, attribute_name: str) -> str:
+        """
+        a child class can override this if path is not equal to attribute_name
+        """
+        return f"/{attribute_name}"
+
+    def patch(self) -> Optional[MetadataChangeProposalClass]:
+        # filter property
+        properties = {
+            attr: getattr(self.aspect, attr)
+            for attr in self.attributes
+            if self.is_attribute_includable(attr)
+            and not callable(getattr(self.aspect, attr))
+        }
+
+        for property_ in properties:
+            if properties[property_]:
+                self.patch_builder.add_patch(
+                    aspect_name=self.aspect.ASPECT_NAME,
+                    op="add",
+                    path=self.attribute_path(property_),
+                    value=properties[property_],
+                )
+
+        mcps: List[MetadataChangeProposalClass] = list(self.patch_builder.build())
+        if mcps:
+            return mcps[0]
+
+        return None
+
+
 def convert_upstream_lineage_to_patch(
     urn: str,
     aspect: UpstreamLineageClass,
@@ -62,26 +129,18 @@ def convert_dashboard_info_to_patch(
 ) -> Optional[MetadataWorkUnit]:
     patch_builder = DashboardPatchBuilder(urn, system_metadata)
 
-    if aspect.customProperties:
-        for key in aspect.customProperties:
-            patch_builder.add_custom_property(
-                key, str(aspect.customProperties.get(key))
-            )
-
-    if aspect.datasetEdges:
-        for datasetEdge in aspect.datasetEdges:
-            patch_builder.add_dataset_edge(datasetEdge)
+    patch_entity_aspect: PatchEntityAspect = PatchEntityAspect(
+        aspect=aspect,
+        patch_builder=patch_builder,
+    )
 
-    if aspect.chartEdges:
-        for chartEdge in aspect.chartEdges:
-            patch_builder.add_chart_edge(chartEdge)
+    mcp: Optional[MetadataChangeProposalClass] = patch_entity_aspect.patch()
 
-    values = patch_builder.build()
-    if values:
-        mcp = next(iter(values))
+    if mcp:
         return MetadataWorkUnit(
             id=MetadataWorkUnit.generate_workunit_id(mcp), mcp_raw=mcp
         )
+
     return None
 
 

metadata-ingestion/src/datahub/ingestion/source/powerbi/powerbi.py

@@ -1197,8 +1197,7 @@ def report_to_datahub_work_units(
     ) -> Iterable[MetadataWorkUnit]:
         mcps: List[MetadataChangeProposalWrapper] = []
 
-        logger.debug(f"Converting dashboard={report.name} to datahub dashboard")
-
+        logger.debug(f"Converting report={report.name} to datahub dashboard")
         # Convert user to CorpUser
         user_mcps = self.to_datahub_users(report.users)
         # Convert pages to charts. A report has single dataset and same dataset used in pages to create visualization
@@ -1215,9 +1214,7 @@ def report_to_datahub_work_units(
         mcps.extend(chart_mcps)
         mcps.extend(report_mcps)
 
-        # Convert MCP to work_units
-        work_units = map(self._to_work_unit, mcps)
-        return work_units
+        return map(self._to_work_unit, mcps)
 
 
 @platform_name("PowerBI")

metadata-ingestion/tests/integration/powerbi/golden_test_admin_access_not_allowed.json

@@ -300,18 +300,49 @@
         "json": [
             {
                 "op": "add",
-                "path": "/customProperties/chartCount",
-                "value": "2"
+                "path": "/charts",
+                "value": [
+                    "urn:li:chart:(powerbi,charts.B8E293DC-0C83-4AA0-9BB9-0A8738DF24A0)",
+                    "urn:li:chart:(powerbi,charts.23212598-23b5-4980-87cc-5fc0ecd84385)"
+                ]
             },
             {
                 "op": "add",
-                "path": "/customProperties/workspaceName",
-                "value": "demo-workspace"
+                "path": "/customProperties",
+                "value": {
+                    "chartCount": "2",
+                    "workspaceName": "demo-workspace",
+                    "workspaceId": "64ED5CAD-7C10-4684-8180-826122881108"
+                }
+            },
+            {
+                "op": "add",
+                "path": "/dashboardUrl",
+                "value": "https://localhost/dashboards/web/1"
+            },
+            {
+                "op": "add",
+                "path": "/description",
+                "value": "Description of test dashboard"
+            },
+            {
+                "op": "add",
+                "path": "/lastModified",
+                "value": {
+                    "created": {
+                        "time": 0,
+                        "actor": "urn:li:corpuser:unknown"
+                    },
+                    "lastModified": {
+                        "time": 0,
+                        "actor": "urn:li:corpuser:unknown"
+                    }
+                }
             },
             {
                 "op": "add",
-                "path": "/customProperties/workspaceId",
-                "value": "64ED5CAD-7C10-4684-8180-826122881108"
+                "path": "/title",
+                "value": "test_dashboard"
             }
         ]
     },

metadata-ingestion/tests/integration/powerbi/golden_test_cll.json

@@ -1264,18 +1264,49 @@
         "json": [
             {
                 "op": "add",
-                "path": "/customProperties/chartCount",
-                "value": "2"
+                "path": "/charts",
+                "value": [
+                    "urn:li:chart:(powerbi,charts.B8E293DC-0C83-4AA0-9BB9-0A8738DF24A0)",
+                    "urn:li:chart:(powerbi,charts.23212598-23b5-4980-87cc-5fc0ecd84385)"
+                ]
             },
             {
                 "op": "add",
-                "path": "/customProperties/workspaceName",
-                "value": "demo-workspace"
+                "path": "/customProperties",
+                "value": {
+                    "chartCount": "2",
+                    "workspaceName": "demo-workspace",
+                    "workspaceId": "64ED5CAD-7C10-4684-8180-826122881108"
+                }
+            },
+            {
+                "op": "add",
+                "path": "/dashboardUrl",
+                "value": "https://localhost/dashboards/web/1"
+            },
+            {
+                "op": "add",
+                "path": "/description",
+                "value": "Description of test dashboard"
+            },
+            {
+                "op": "add",
+                "path": "/lastModified",
+                "value": {
+                    "created": {
+                        "time": 0,
+                        "actor": "urn:li:corpuser:unknown"
+                    },
+                    "lastModified": {
+                        "time": 0,
+                        "actor": "urn:li:corpuser:unknown"
+                    }
+                }
             },
             {
                 "op": "add",
-                "path": "/customProperties/workspaceId",
-                "value": "64ED5CAD-7C10-4684-8180-826122881108"
+                "path": "/title",
+                "value": "test_dashboard"
             }
         ]
     },