feat: modify_mapped_data_source add support for SQL datasets

JIRA: PSDK-206 risk: low
gooddata · Oct 8, 2024 · 2f33e34 · 2f33e34
1 parent ebe3139
commit 2f33e34
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