fixup! fixup! fixup! feat: allow to customize the blocked labels

Signed-off-by: Yoan Blanc <[email protected]>
zakkg3 · Nov 12, 2024 · 0a64d68 · 0a64d68
1 parent 060a3a2
commit 0a64d68
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/os_utils.py b/src/os_utils.py
@@ -20,12 +20,11 @@ def get_replace_existing() -> bool:
 
 @cache
 def get_blocked_labels() -> list[str]:
-    blocked_labels = os.getenv('BLOCKED_LABELS')
+    if blocked_labels := os.getenv('BLOCKED_LABELS'):
+        return [label.strip() for label in blocked_labels.split(',')]
 
-    if not blocked_labels:
-        return BLOCKED_LABELS
+    return BLOCKED_LABELS
 
-    return [label.strip() for label in blocked_labels.split(',')]
 
 @cache
 def in_cluster() -> bool: