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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
# Copyright (C) 2017-present Linaro Limited
#
# Author: Neil Williams <neil.williams@linaro.org>
# Remi Duraffort <remi.duraffort@linaro.org>
# Milosz Wasilewski <milosz.wasilewski@linaro.org>
#
# SPDX-License-Identifier: GPL-2.0-or-later
# pylint: disable=wildcard-import
# pylint: disable=unused-wildcard-import
import base64
import contextlib
import json
import os
from pathlib import Path
import environ
import yaml
from lava_server.settings.common import *
from lava_server.settings.config_file import ConfigFile
############################
# Load configuration files #
############################
# We only rely on django-environ for the neat database configuration helper,
# which handles several special and corner cases, like sqlite memory
# configurations, for one.
#
# The reason for this is its support for proxy variables makes anything else
# highly fragile. For instance, if the SECRET_KEY happens to start with a $
# character, it will try to use the rest of the key as a variable name, and
# expose it on an error message.
env = environ.Env()
if os.environ.get("DATABASE_URL"):
DATABASES = {"default": env.db()}
INSTANCE_NAME = os.environ.get("INSTANCE_NAME", INSTANCE_NAME)
else:
with contextlib.suppress(FileNotFoundError):
config = ConfigFile.load("/etc/lava-server/instance.conf")
DATABASES = {
"default": {
"ENGINE": "django.db.backends.postgresql",
"NAME": getattr(config, "LAVA_DB_NAME", ""),
"USER": getattr(config, "LAVA_DB_USER", ""),
"PASSWORD": getattr(config, "LAVA_DB_PASSWORD", ""),
"HOST": getattr(config, "LAVA_DB_SERVER", "127.0.0.1"),
"PORT": getattr(config, "LAVA_DB_PORT", "5432"),
}
}
if conn_max_age_str := getattr(config, "CONN_MAX_AGE", None):
DATABASES["default"]["CONN_MAX_AGE"] = int(conn_max_age_str)
INSTANCE_NAME = config.LAVA_INSTANCE
if os.environ.get("SECRET_KEY"):
SECRET_KEY = os.environ["SECRET_KEY"]
else:
with contextlib.suppress(FileNotFoundError):
SECRET_KEY = ConfigFile.load("/etc/lava-server/secret_key.conf").SECRET_KEY
if os.environ.get("ALLOWED_HOSTS"):
ALLOWED_HOSTS += os.environ["ALLOWED_HOSTS"].split(",")
# Load settings
FILES = [
Path("/etc/lava-server/settings.conf"),
Path("/etc/lava-server/settings.yaml"),
*sorted(Path("/etc/lava-server/settings.d").glob("*.yaml")),
]
for filename in FILES:
try:
with contextlib.suppress(FileNotFoundError):
for k, v in yaml.safe_load(filename.read_text(encoding="utf-8")).items():
globals()[k] = v
except yaml.YAMLError as exc:
print(f"[INIT] Unable to load {filename.name}: invalid yaml")
print(exc)
raise Exception(f"Unable to load {filename.name}") from exc
for k in os.environ:
if k.startswith("LAVA_SETTINGS_"):
value = env(k)
globals()[k[len("LAVA_SETTINGS_") :]] = value
elif k.startswith("LAVA_YAML_SETTINGS_"):
try:
value = yaml.safe_load(env(k))
except yaml.YAMLError as exc:
print(f"[INIT] Unable to parse {k}: invalid yaml")
print(exc)
raise Exception(f"Unable to parse {k}") from exc
globals()[k[len("LAVA_YAML_SETTINGS_") :]] = value
if "LAVA_JSON_SETTINGS" in os.environ:
try:
for k, v in json.loads(
base64.b64decode(os.environ["LAVA_JSON_SETTINGS"])
).items():
globals()[k] = v
except Exception as exc:
print(f"[INIT] Unable to load LAVA_JSON_SETTINGS: invalid string")
print(exc)
raise Exception(f"Unable to load LAVA_JSON_SETTINGS") from exc
if "DATABASES" in locals() and DATABASES.get("default"):
if db_conn_max_age_str := globals().get("DB_CONN_MAX_AGE"):
with contextlib.suppress(ValueError):
DATABASES["default"]["CONN_MAX_AGE"] = int(db_conn_max_age_str)
if db_conn_health_check_str := globals().get("DB_CONN_HEALTH_CHECKS"):
if db_conn_health_check_str in ["True", "true", "1", "yes", "on"]:
DATABASES["default"]["CONN_HEALTH_CHECKS"] = True
elif db_conn_health_check_str in ["False", "false", "0", "no", "off"]:
DATABASES["default"]["CONN_HEALTH_CHECKS"] = False
# Update settings with custom values
for k, v in update(globals()).items():
globals()[k] = v
|
