fix(ingest/powerbi): Fix column lineage

Author: mminichino

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

@@ -334,14 +334,16 @@ def create_lineage(
             qualified_table_name=qualified_table_name,
         )
 
+        column_lineage = self.create_table_column_lineage(urn)
+
         return Lineage(
             upstreams=[
                 DataPlatformTable(
                     data_platform_pair=self.get_platform_pair(),
                     urn=urn,
                 )
             ],
-            column_lineage=[],
+            column_lineage=column_lineage
         )
 
 
@@ -399,14 +401,16 @@ def create_lineage(
             qualified_table_name=qualified_table_name,
         )
 
+        column_lineage = self.create_table_column_lineage(urn)
+
         return Lineage(
             upstreams=[
                 DataPlatformTable(
                     data_platform_pair=self.get_platform_pair(),
                     urn=urn,
                 )
             ],
-            column_lineage=[],
+            column_lineage=column_lineage
         )
 
 
@@ -484,14 +488,16 @@ def create_lineage(
                 qualified_table_name=qualified_table_name,
             )
 
+            column_lineage = self.create_table_column_lineage(urn)
+
             return Lineage(
                 upstreams=[
                     DataPlatformTable(
                         data_platform_pair=self.get_platform_pair(),
                         urn=urn,
                     )
                 ],
-                column_lineage=[],
+                column_lineage=column_lineage
             )
 
         return Lineage.empty()
@@ -544,14 +550,17 @@ def two_level_access_pattern(
             server=server,
             qualified_table_name=qualified_table_name,
         )
+
+        column_lineage = self.create_table_column_lineage(urn)
+
         return Lineage(
             upstreams=[
                 DataPlatformTable(
                     data_platform_pair=self.get_platform_pair(),
                     urn=urn,
                 )
             ],
-            column_lineage=[],
+            column_lineage=column_lineage
         )
 
 
@@ -715,7 +724,7 @@ def create_lineage(
                     urn=urn,
                 )
             ],
-            column_lineage=column_lineage,
+            column_lineage=column_lineage
         )
 
 
@@ -763,6 +772,7 @@ def create_urn_using_old_parser(self, query: str, server: str) -> Lineage:
 
         tables: List[str] = native_sql_parser.get_tables(query)
 
+        column_lineage = []
         for qualified_table_name in tables:
             if len(qualified_table_name.split(".")) != 3:
                 logger.debug(
@@ -785,11 +795,13 @@ def create_urn_using_old_parser(self, query: str, server: str) -> Lineage:
                 )
             )
 
+            column_lineage = self.create_table_column_lineage(urn)
+
         logger.debug(f"Generated dataplatform_tables {dataplatform_tables}")
 
         return Lineage(
             upstreams=dataplatform_tables,
-            column_lineage=[],
+            column_lineage=column_lineage
         )
 
     def get_db_name(self, data_access_tokens: List[str]) -> Optional[str]: