handles no env variable

mitre · Feb 13, 2024 · ab1f75a · ab1f75a
1 parent 80395d1
commit ab1f75a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/app/ascii_banner.py b/app/ascii_banner.py
@@ -41,7 +41,7 @@
 "
 
 
-if int(os.environ['NO_COLOR']) == 1:
+if int(os.environ.get('NO_COLOR', 0)) == 1:
     ASCII_BANNER = _BANNER
 else:
     ASCII_BANNER = f"{DARK_BLUE}{_BANNER_SECTION_1}{DARK_PURPLE}{_BANNER_SECTION_2}{DARK_RED}{BANNER_SECTION_3}{END}"