Merge pull request #76 from syou6162/enhance_nested_column_treatment

Allow to keep root level column name that contains nested fields
z3z1ma · Aug 3, 2023 · 308494a · 308494a
2 parents 14f96f2 + a011d79
commit 308494a
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/src/dbt_osmosis/core/osmosis.py b/src/dbt_osmosis/core/osmosis.py
@@ -350,11 +350,10 @@ def get_columns(self, parts: Tuple[str, str, str]) -> List[str]:
                     )
                     return columns
                 try:
-                    columns = [
-                        self.column_casing(exp.name)
-                        for c in self.adapter.get_columns_in_relation(table)
-                        for exp in getattr(c, "flatten", lambda: [c])()
-                    ]
+                    for c in self.adapter.get_columns_in_relation(table):
+                        columns.append(self.column_casing(c.name))
+                        if hasattr(c, "flatten"):
+                            columns.extend([self.column_casing(exp.name) for exp in c.flatten()])
                 except Exception as error:
                     logger().info(
                         (