-
Notifications
You must be signed in to change notification settings - Fork 2
/
settings.py
89 lines (73 loc) · 3.13 KB
/
settings.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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
import json
import os
from pprint import pformat
import toml
import logging
LOGGER = logging.getLogger("Settings")
FOLDER = os.path.abspath(os.path.dirname(__file__))
class _Settings():
_settings: dict = {}
def __init__(self, settings_folder=FOLDER):
self.settings_folder = settings_folder
self.reload_settings()
def reload_settings(self) -> None:
settings_backup = self._settings.copy()
with open(self.settings_folder + "/settings.base.toml") as f:
base_settings = toml.load(f, dotdict)
self.update_settings(base_settings)
try:
with open(self.settings_folder + "/settings.toml", "r") as f:
settings_user = toml.load(f, dotdict)
diff = list(set(settings_user.keys()) -
set(base_settings.keys()))
if len(diff) > 0:
self.update_settings(settings_backup)
LOGGER.error(
"Invalid user settings! Restoring previous settings")
raise ValueError("User settings has unknown item(s) not defined in base TOML: %s." %
diff)
self._settings.update(settings_user)
except FileNotFoundError:
LOGGER.critical(
"Failed to find settings.toml!")
LOGGER.debug("Iterating over os.environ")
for key, value in os.environ.items():
key = key.lower()
if key.startswith("ob_"):
key = key[3:]
try:
self._settings[key] = json.loads(value)
except json.JSONDecodeError as e:
LOGGER.warning(
"The key %s has an invalid JSON value of '%s' (error: %s), "
"loading just as plain string", key, value, e)
self._settings[key] = value
else:
LOGGER.debug(
"Loaded settings key %s from environment", key)
LOGGER.debug("Result settings: %s", pformat(self._settings))
LOGGER.info("Settings reloaded")
def update_settings(self, settings: dict):
self._settings = settings
def __getattr__(self, item: str):
if item in self._settings:
return self._settings[item]
else:
raise KeyError("Unknown settings entry: %s" % item)
def get(self, item: str, fallback=None):
return self._settings.get(item, fallback)
def __setitem__(self, key: str, value):
return self._settings.__setitem__(key, value)
def save_settings_to_disk(self) -> None:
try:
with open(self.settings_folder + "/settings.toml", "w") as f:
toml.dump(self._settings, f)
except FileNotFoundError:
LOGGER.critical(
"Failed to find settings.toml! Continuing anyway cause who knows, maybe doc is getting built?")
class dotdict(dict):
"""dot.notation access to dictionary attributes"""
__getattr__ = dict.get
__setattr__ = dict.__setitem__
__delattr__ = dict.__delitem__
Settings: _Settings = _Settings()