Force CHROME_BIN_PATH env var to be validated at start of billing pipeline.

This makes use of `required_env_vars()`.

Author: KelvinLinBU

process_report/invoices/pi_specific_invoice.py

@@ -1,5 +1,4 @@
 import os
-import sys
 from dataclasses import dataclass
 import subprocess
 import tempfile
@@ -13,7 +12,6 @@
 
 
 TEMPLATE_DIR_PATH = "process_report/templates"
-CHROME_BIN_PATH = os.environ.get("CHROME_BIN_PATH", "/usr/bin/chromium")
 
 
 logger = logging.getLogger(__name__)
@@ -121,17 +119,12 @@ def _create_html_invoice(temp_fd):
             temp_fd.flush()
 
         def _create_pdf_invoice(temp_fd_name):
-            if not os.path.exists(CHROME_BIN_PATH):
-                sys.exit(
-                    f"Chrome binary does not exist at {CHROME_BIN_PATH}. Make sure the env var CHROME_BIN_PATH is set correctly and that Google Chrome is installed"
-                )
-
             invoice_pdf_path = (
                 f"{self.name}/{pi_instituition}_{pi}_{self.invoice_month}.pdf"
             )
             subprocess.run(
                 [
-                    CHROME_BIN_PATH,
+                    os.environ.get("CHROME_BIN_PATH", "/usr/bin/chromium"),
                     "--headless",
                     "--no-sandbox",
                     f"--print-to-pdf={invoice_pdf_path}",

process_report/process_report.py

@@ -241,7 +241,7 @@ def main():
     )
     args = parser.parse_args()
 
-    required_env_vars = []
+    required_env_vars = ["CHROME_BIN_PATH"]
     if not args.coldfront_data_file:
         required_env_vars.extend(["KEYCLOAK_CLIENT_ID", "KEYCLOAK_CLIENT_SECRET"])
     validate_required_env_vars(required_env_vars)

process_report/tests/unit/invoices/test_pi_specific_invoice.py

@@ -1,9 +1,9 @@
+import os
 import tempfile
 from unittest import TestCase, mock
 import pandas
 
 from process_report.tests import util as test_utils
-from process_report.invoices.pi_specific_invoice import CHROME_BIN_PATH
 
 
 class TestPISpecificInvoice(TestCase):
@@ -138,7 +138,7 @@ def test_export_pi(self, mock_subprocess_run, mock_path_exists, mock_filter_cols
             for i, pi_pdf_path in enumerate([pi_pdf_1, pi_pdf_2]):
                 chrome_arglist, _ = mock_subprocess_run.call_args_list[i]
                 answer_arglist = [
-                    CHROME_BIN_PATH,
+                    os.environ.get("CHROME_BIN_PATH", "/usr/bin/chromium"),
                     "--headless",
                     "--no-sandbox",
                     f"--print-to-pdf={pi_pdf_path}",