-
Notifications
You must be signed in to change notification settings - Fork 1
/
config.py
62 lines (42 loc) · 1.27 KB
/
config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
from dotenv import load_dotenv
from pytz import timezone
import os
DEFAULTS = {
"MONGO_URL": "mongodb://localhost:27017",
"MONGO_DB": "annotator",
"IMAGE_FOLDER": "_images",
"LOG_FOLDER": "_logs",
"IS_DEBUG": False,
}
DATETIME_FORMAT = "%Y-%m-%d %H:%M:%S"
DATE_FORMAT = "%Y-%m-%d"
def load_config():
# read out existing os environment
load_dotenv()
config = {
"MONGO_URL": os.getenv("MONGO_URL"),
"MONGO_DB": os.getenv("MONGO_DB"),
"IMAGE_FOLDER": os.getenv("IMAGE_FOLDER"),
"LOG_FOLDER": os.getenv("LOG_FOLDER"),
"IS_DEBUG": os.getenv("IS_DEBUG") == "1",
}
# apply defaults for missing config params
for key in DEFAULTS:
if key not in config or config[key] is None:
config[key] = DEFAULTS[key]
# check if log folder exists
if not os.path.isdir(config["LOG_FOLDER"]):
os.mkdir(config["LOG_FOLDER"])
return config
def get_log_folder():
config = load_config()
return config["LOG_FOLDER"]
def get_image_folder():
config = load_config()
return config["IMAGE_FOLDER"]
def get_mongodb_config():
config = load_config()
return config["MONGO_URL"], config["MONGO_DB"]
def is_debug():
config = load_config()
return config["IS_DEBUG"]