fix: remove hardcoding of table

goto · Jul 3, 2024 · e2c2fb2 · e2c2fb2
1 parent cf9ca0d
commit e2c2fb2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/task/bq2bq/executor/bumblebee/transformation.py b/task/bq2bq/executor/bumblebee/transformation.py
@@ -231,7 +231,9 @@ def transform(self):
                                        self.dry_run,
                                        self.execution_time)
 
-        self.bigquery_service.execute_query("truncate table `gtfndata-integration.playground.replace_table_target`;")
+        truncate_query = "truncate table `{}`;".format(self.task_config.destination_table)
+
+        self.bigquery_service.execute_query(truncate_query)
         task.execute()