added dynamic regex with export

NOAA-EMC · Jun 11, 2024 · 86e31db · 86e31db
1 parent b8dd227
commit 86e31db
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/wxflow/configuration.py b/src/wxflow/configuration.py
@@ -1,5 +1,6 @@
 import glob
 import os
+import re
 import random
 import shutil
 import subprocess
@@ -98,8 +99,9 @@ def _get_script_env(cls, scripts: List) -> Dict[str, Any]:
         default_env = cls._get_shell_env([])
         and_script_env = cls._get_shell_env(scripts)
         keys_in_scripts = set()
+        regex_pattern = 'export\\s+\\b(' + '|'.join(map(re.escape, default_env.keys())) + ')(?==)'
         for script in scripts:
-            result = subprocess.run(['grep', '-o', '-P', '\\b(' + '|'.join(default_env.keys()) + ')(?==)', script], stdout=subprocess.PIPE)
+            result = subprocess.run(['grep', '-o', '-P', regex_pattern, script], stdout=subprocess.PIPE)
             keys_in_scripts.update(result.stdout.decode().split())
         vars_just_in_script = set(and_script_env) - set(default_env) | keys_in_scripts
         union_env = dict(default_env)