Merge pull request #836 from hkad98/jkd/sql-datasets

feat: modify_mapped_data_source add support for SQL datasets
gooddata · Oct 8, 2024 · 774eb85 · 774eb85
2 parents ebe3139 + 2f33e34
commit 774eb85
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/gooddata-sdk/gooddata_sdk/catalog/workspace/declarative_model/workspace/logical_model/ldm.py b/gooddata-sdk/gooddata_sdk/catalog/workspace/declarative_model/workspace/logical_model/ldm.py
@@ -188,6 +188,10 @@ def modify_mapped_data_source(self, data_source_mapping: Optional[dict]) -> Cata
                     data_source_id = dataset.data_source_table_id.data_source_id
                     if data_source_id in data_source_mapping:
                         dataset.data_source_table_id.data_source_id = data_source_mapping[data_source_id]
+                if dataset.sql is not None:
+                    data_source_id = dataset.sql.data_source_id
+                    if data_source_id in data_source_mapping:
+                        dataset.sql.data_source_id = data_source_mapping[data_source_id]
         return self
 
     @staticmethod