Convert string-to-bool in init

rommapp · Aug 29, 2024 · f44d127 · f44d127
1 parent 4c2c6e3
commit f44d127
Showing 1 changed file with 22 additions and 25 deletions.
diff --git a/backend/config/__init__.py b/backend/config/__init__.py
@@ -6,6 +6,11 @@
 
 load_dotenv()
 
+
+def str_to_bool(value: str) -> bool:
+    return value.lower() in ("true", "1")
+
+
 # GUNICORN
 DEV_PORT: Final = int(os.environ.get("VITE_BACKEND_DEV_PORT", "5000"))
 DEV_HOST: Final = "127.0.0.1"
@@ -53,49 +58,41 @@
 ROMM_AUTH_SECRET_KEY: Final = os.environ.get(
     "ROMM_AUTH_SECRET_KEY", secrets.token_hex(32)
 )
-DISABLE_CSRF_PROTECTION = os.environ.get("DISABLE_CSRF_PROTECTION", "false") in [
-    "true",
-    True,
-    1,
-]
-DISABLE_DOWNLOAD_ENDPOINT_AUTH = os.environ.get(
-    "DISABLE_DOWNLOAD_ENDPOINT_AUTH", "false"
-) in ["true", True, 1]
+DISABLE_CSRF_PROTECTION = str_to_bool(
+    os.environ.get("DISABLE_CSRF_PROTECTION", "false")
+)
+DISABLE_DOWNLOAD_ENDPOINT_AUTH = str_to_bool(
+    os.environ.get("DISABLE_DOWNLOAD_ENDPOINT_AUTH", "false")
+)
 
 # SCANS
 SCAN_TIMEOUT: Final = int(os.environ.get("SCAN_TIMEOUT", 60 * 60 * 4))  # 4 hours
 
 # TASKS
-ENABLE_RESCAN_ON_FILESYSTEM_CHANGE: Final = os.environ.get(
-    "ENABLE_RESCAN_ON_FILESYSTEM_CHANGE", "false"
-) in ["true", True, 1]
+ENABLE_RESCAN_ON_FILESYSTEM_CHANGE: Final = str_to_bool(
+    os.environ.get("ENABLE_RESCAN_ON_FILESYSTEM_CHANGE", "false")
+)
 RESCAN_ON_FILESYSTEM_CHANGE_DELAY: Final = int(
     os.environ.get("RESCAN_ON_FILESYSTEM_CHANGE_DELAY", 5)  # 5 minutes
 )
-ENABLE_SCHEDULED_RESCAN: Final = os.environ.get("ENABLE_SCHEDULED_RESCAN", "false") in [
-    "true",
-    True,
-    1,
-]
+ENABLE_SCHEDULED_RESCAN: Final = str_to_bool(
+    os.environ.get("ENABLE_SCHEDULED_RESCAN", "false")
+)
 SCHEDULED_RESCAN_CRON: Final = os.environ.get(
     "SCHEDULED_RESCAN_CRON",
     "0 3 * * *",  # At 3:00 AM every day
 )
-ENABLE_SCHEDULED_UPDATE_SWITCH_TITLEDB: Final = os.environ.get(
-    "ENABLE_SCHEDULED_UPDATE_SWITCH_TITLEDB", "false"
-) in ["true", True, 1]
+ENABLE_SCHEDULED_UPDATE_SWITCH_TITLEDB: Final = str_to_bool(
+    os.environ.get("ENABLE_SCHEDULED_UPDATE_SWITCH_TITLEDB", "false")
+)
 SCHEDULED_UPDATE_SWITCH_TITLEDB_CRON: Final = os.environ.get(
     "SCHEDULED_UPDATE_SWITCH_TITLEDB_CRON",
     "0 4 * * *",  # At 4:00 AM every day
 )
 
 # EMULATION
-DISABLE_EMULATOR_JS = os.environ.get("DISABLE_EMULATOR_JS", "false") in [
-    "true",
-    True,
-    1,
-]
-DISABLE_RUFFLE_RS = os.environ.get("DISABLE_RUFFLE_RS", "false") in ["true", True, 1]
+DISABLE_EMULATOR_JS = str_to_bool(os.environ.get("DISABLE_EMULATOR_JS", "false"))
+DISABLE_RUFFLE_RS = str_to_bool(os.environ.get("DISABLE_RUFFLE_RS", "false"))
 
 # TESTING
 IS_PYTEST_RUN: Final = bool(os.environ.get("PYTEST_VERSION", False))