Specify settings .env file with TAKAHE_ENV_FILE

This commit is contained in:
Michael Manfre 2022-11-27 12:57:13 -05:00 committed by GitHub
parent 9d634c993f
commit 18c85bc6d2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 2 deletions

View File

@ -1,3 +1,4 @@
import os
import secrets import secrets
import sys import sys
import urllib.parse import urllib.parse
@ -24,6 +25,10 @@ def as_bool(v: Optional[Union[str, List[str]]]):
Environments = Literal["development", "production", "test"] 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): class Settings(BaseSettings):
""" """
@ -90,7 +95,7 @@ class Settings(BaseSettings):
class Config: class Config:
env_prefix = "TAKAHE_" env_prefix = "TAKAHE_"
env_file = str(BASE_DIR / ".env") env_file = str(BASE_DIR / TAKAHE_ENV_FILE)
env_file_encoding = "utf-8" env_file_encoding = "utf-8"
# Case sensitivity doesn't work on Windows, so might as well be # Case sensitivity doesn't work on Windows, so might as well be
# consistent from the get-go. # 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 SECRET_KEY = SETUP.SECRET_KEY
DEBUG = SETUP.DEBUG DEBUG = SETUP.DEBUG