admin/log-the-target-db-env (#41)

* Log the target db env * Remove .strip call
SIGLA-GU · Sep 17, 2021 · c277084 · c277084
1 parent 1e0c1bf
commit c277084
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 5 deletions.
diff --git a/siglatools/bin/load_spreadsheets.py b/siglatools/bin/load_spreadsheets.py
@@ -326,7 +326,7 @@ def main():
         ]
         if not spreadsheet_ids:
             raise Exception("No spreadsheet ids found.")
-        if args.db_env.strip() not in [Environment.staging, Environment.production]:
+        if args.db_env not in [Environment.staging, Environment.production]:
             raise Exception(
                 "Incorrect database enviroment specification. Use 'staging' or 'production'."
             )
@@ -335,6 +335,9 @@ def main():
             if args.db_env == Environment.staging
             else args.prod_db_connection_url
         )
+        log.info(
+            f"""Loading spreadsheets {", ".join(spreadsheet_ids)} to the {args.db_env} database."""
+        )
         load_spreadsheets(
             spreadsheet_ids,
             db_connection_url,

diff --git a/siglatools/bin/run_qa_test.py b/siglatools/bin/run_qa_test.py
@@ -1054,9 +1054,9 @@ def main():
     try:
         args = Args()
         dbg = args.debug
-        if args.db_env.strip() not in [Environment.staging, Environment.production]:
+        if args.db_env not in [Environment.staging, Environment.production]:
             raise Exception(
-                "Incorrect database enviroment specification. Use 'staging' or 'production'."
+                "Incorrect database environment specification. Use 'staging' or 'production'."
             )
         run_qa_test(
             args.master_spreadsheet_id,

diff --git a/siglatools/bin/run_sigla_pipeline.py b/siglatools/bin/run_sigla_pipeline.py
@@ -208,15 +208,19 @@ def main():
         dbg = args.debug
         if args.master_spreadsheet_id is None:
             raise Exception("No main spreadsheet id found.")
-        if args.db_env.strip() not in [Environment.staging, Environment.production]:
+        if args.db_env not in [Environment.staging, Environment.production]:
             raise Exception(
                 "Incorrect database enviroment specification. Use 'staging' or 'production'."
             )
+        log.info(
+            f"""Loading all spreadsheets in the master spreadsheet {args.master_spreadsheet_id}""",
+            f" to the {args.db_env} database.",
+        )
         run_sigla_pipeline(
             args.master_spreadsheet_id,
             args.google_api_credentials_path,
             args.staging_db_connection_url
-            if args.db_env.strip() == Environment.staging
+            if args.db_env == Environment.staging
             else args.prod_db_connection_url,
         )
     except Exception as e: