chore: add explicit log for dry-run (#29)

goto · Jul 8, 2024 · d514766 · d514766
1 parent 2e20092
commit d514766
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/task/bq2bq/executor/bumblebee/transformation.py b/task/bq2bq/executor/bumblebee/transformation.py
@@ -194,10 +194,10 @@ def execute(self):
 
         if not self.dry_run:
             result = self.loader.load(query)
-
-        logger.info(result)
-        logger.info("finished")
-
+            logger.info(result)
+            logger.info("finished")
+        else:
+            logger.info("dry-run finished")
 
 class TableTransformation:
     """
@@ -295,9 +295,10 @@ def execute(self):
         result = None
         if not self.dry_run:
             result = self.loader.load(self.query)
-
-        logger.info(result)
-        logger.info("finished")
+            logger.info(result)
+            logger.info("finished")
+        else:
+            logger.info("dry-run finished")
 
     async def async_execute(self):
         self.execute()
@@ -355,6 +356,8 @@ def transform(self):
         if not self.dry_run:
             result = self.loader.load(query)
             logger.info("finished {}".format(result.total_rows))
+        else:
+            logger.info("dry-run finished")
 
 
 class MultiPartitionTransformation: