feat: allow to overwrite data dir (#1703)

New env variable "DATA_DIR" to overwrite data dir.

Co-authored-by: e <e@e.e>
This commit is contained in:
1848 2022-11-11 01:18:40 +01:00 committed by GitHub
parent e7de0c90a1
commit f0ed2ad115
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -18,16 +18,17 @@ ENV = BASE_DIR.joinpath(".env")
dotenv.load_dotenv(ENV) dotenv.load_dotenv(ENV)
PRODUCTION = os.getenv("PRODUCTION", "True").lower() in ["true", "1"] PRODUCTION = os.getenv("PRODUCTION", "True").lower() in ["true", "1"]
TESTING = os.getenv("TESTING", "False").lower() in ["true", "1"] TESTING = os.getenv("TESTING", "False").lower() in ["true", "1"]
DATA_DIR = os.getenv("DATA_DIR")
def determine_data_dir() -> Path: def determine_data_dir() -> Path:
global PRODUCTION, TESTING, BASE_DIR global PRODUCTION, TESTING, BASE_DIR, DATA_DIR
if TESTING: if TESTING:
return BASE_DIR.joinpath("tests/.temp") return BASE_DIR.joinpath(DATA_DIR if DATA_DIR else "tests/.temp")
if PRODUCTION: if PRODUCTION:
return Path("/app/data") return Path(DATA_DIR if DATA_DIR else "/app/data")
return BASE_DIR.joinpath("dev", "data") return BASE_DIR.joinpath("dev", "data")