Add generic ENV string cleaner (#32)

Closes #29 Closes #31 Co-authored-by: Edward Delaporte <[email protected]> Co-authored-by: David Riddle <[email protected]> Co-authored-by: Michelle Pitcel <[email protected]> Co-authored-by: Zach Carrington <[email protected]>
techservicesillinois · Jul 16, 2024 · bd3beee · bd3beee
1 parent 9b4c3b8
commit bd3beee
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/src/vcr_cleaner/cleaners/env_strings.py b/src/vcr_cleaner/cleaners/env_strings.py
@@ -57,9 +57,22 @@ def clean_env_helper(dirty):
 
     # Clean response body
     for clean_me in clean_strings:
-        body = body.replace(clean_me, 'CLEANED')
+        if clean_me.strip() != '':
+            body = body.replace(clean_me, 'CLEANED')
 
     if 'string' in dirty['body']:
         dirty['body']['string'] = body
     else:
         dirty['body'] = body
+
+
+def simple_clean_env_string(dirty):
+    '''Clean any strings set in the CLEAN_STRING environment variable.
+    export CLEAN_STRINGS='my_name,my_email'
+    '''
+    clean_strings = os.environ.get('CLEAN_STRINGS', "").split(',')
+    for clean_me in clean_strings:
+        if clean_me.strip() != '':
+            dirty = dirty.replace(clean_me, 'CLEANED')
+
+    return dirty