fix(logs): suppress noisy logs during setup and upgrade

Author: aarthy-dk

testgen/commands/run_launch_db_config.py

@@ -86,6 +86,7 @@ def run_launch_db_config(delete_db: bool, drop_users_and_roles: bool = True) ->
         user_override=params_mapping["TESTGEN_ADMIN_USER"],
         password_override=params_mapping["TESTGEN_ADMIN_PASSWORD"],
         user_type="schema_admin",
+        suppress_logs=True,
     )
     import_metadata_records_from_yaml(params_mapping)
 

testgen/commands/run_upgrade_db_config.py

@@ -96,6 +96,7 @@ def _refresh_static_metadata(params_mapping):
         user_override=params_mapping["TESTGEN_ADMIN_USER"],
         password_override=params_mapping["TESTGEN_ADMIN_PASSWORD"],
         user_type="schema_admin",
+        suppress_logs=True,
     )
     import_metadata_records_from_yaml(params_mapping)
 
@@ -106,6 +107,7 @@ def _refresh_static_metadata(params_mapping):
         user_override=params_mapping["TESTGEN_ADMIN_USER"],
         password_override=params_mapping["TESTGEN_ADMIN_PASSWORD"],
         user_type="schema_admin",
+        suppress_logs=True,
     )
 
 

testgen/common/database/database_service.py

@@ -134,8 +134,9 @@ def execute_db_queries(
     user_override: str | None = None,
     password_override: str | None = None,
     user_type: UserType = "normal",
+    suppress_logs: bool = False,
 ) -> tuple[list[Any], list[int]]:
-    LOG.info(f"DB operation: execute_db_queries on {'Target' if use_target_db else 'App'} database (User type = {user_type})")
+    LOG.info(f"DB operation: execute_db_queries ({len(queries)}) on {'Target' if use_target_db else 'App'} database (User type = {user_type})")
 
     with _init_db_connection(use_target_db, user_override, password_override, user_type) as connection:
         return_values: list[Any] = []
@@ -144,7 +145,8 @@ def execute_db_queries(
             LOG.info("No queries to process")
         for index, (query, params) in enumerate(queries):
             LOG.debug(f"Query: {query}")
-            LOG.info(f"Processing {index + 1} of {len(queries)} queries")
+            if not suppress_logs:
+                LOG.info(f"Processing {index + 1} of {len(queries)} queries")
             transaction = connection.begin()
             result = connection.execute(text(query), params)
             row_counts.append(result.rowcount)

testgen/common/read_yaml_metadata_records.py

@@ -164,6 +164,7 @@ def _process_yaml_for_import(params_mapping: dict, data:dict, parent_table:str,
         user_override=params_mapping["TESTGEN_ADMIN_USER"],
         password_override=params_mapping["TESTGEN_ADMIN_PASSWORD"],
         user_type="schema_admin",
+        suppress_logs=True,
     )
     return