From 18c85bc6d24fc1015ff3a0288fd92aa7dbe71e02 Mon Sep 17 00:00:00 2001
From: Michael Manfre
Date: Sun, 27 Nov 2022 12:57:13 -0500
Subject: Specify settings .env file with TAKAHE_ENV_FILE

---
 takahe/settings.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

(limited to 'takahe/settings.py')

diff --git a/takahe/settings.py b/takahe/settings.py
index 142c690..cec4f2d 100644
--- a/takahe/settings.py
+++ b/takahe/settings.py
@@ -1,3 +1,4 @@
+import os
 import secrets
 import sys
 import urllib.parse
@@ -24,6 +25,10 @@ def as_bool(v: Optional[Union[str, List[str]]]):
 
 Environments = Literal["development", "production", "test"]
 
+TAKAHE_ENV_FILE = os.environ.get(
+    "TAKAHE_ENV_FILE", "test.env" if "pytest" in sys.modules else ".env"
+)
+
 
 class Settings(BaseSettings):
     """
@@ -90,7 +95,7 @@ class Settings(BaseSettings):
 
     class Config:
         env_prefix = "TAKAHE_"
-        env_file = str(BASE_DIR / ".env")
+        env_file = str(BASE_DIR / TAKAHE_ENV_FILE)
         env_file_encoding = "utf-8"
         # Case sensitivity doesn't work on Windows, so might as well be
         # consistent from the get-go.
@@ -106,7 +111,7 @@ class Settings(BaseSettings):
         }
 
 
-SETUP = Settings(_env_file="test.env" if "pytest" in sys.modules else None)
+SETUP = Settings()
 
 SECRET_KEY = SETUP.SECRET_KEY
 DEBUG = SETUP.DEBUG
-- 
cgit v1.2.3