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
|
"""Spamcheck service configuration."""
import argparse
import os
from vyper import v
def load() -> None:
"""Load the configuration options.
Precedence for config:
CLI arguments
Environment variables
Config file
Defaults
"""
if v.get("config_loaded"):
return
# Set CLI config options
parser = argparse.ArgumentParser(description="Application settings")
parser.add_argument(
"--env", type=str, choices=["test", "dev", "prod"], help="Application env"
)
parser.add_argument("--grpc-addr", type=str, help="Application bind address")
parser.add_argument(
"--log_level",
type=str,
choices=["debug", "info", "warning", "error", "fatal"],
help="Application log level",
)
parser.add_argument(
"--ml-classifiers", type=str, help="Directory location for ML classifiers"
)
parser.add_argument("--tls-certificate", type=str, help="TLS certificate path")
parser.add_argument("--tls-private-key", type=str, help="TLS private key path")
parser.add_argument("-c", "--config", type=str, help="Path to config file")
# ignore unknown args. These will be present when running unit tests
args, _ = parser.parse_known_args()
v.bind_args(args.__dict__)
# Set config file options
v.set_config_type("yaml")
v.set_config_name("config")
v.add_config_path("./config")
# Set environment variable options
v.set_env_prefix("spamcheck")
v.automatic_env()
# Set defaults
v.set_default("env", "development")
v.set_default("grpc_addr", "0.0.0.0:8001")
v.set_default("log_level", "info")
v.set_default("tls_certificate", "./ssl/cert.pem")
v.set_default("tls_private_key", "./ssl/key.pem")
v.set_default("ml_classifiers", "./classifiers")
v.set_default("filter.allow_list", {})
v.set_default("filter.deny_list", {})
v.set_default("filter.allowed_domains", [])
# Load config file
config_file = None
if args.config:
config_file = args.config
elif v.is_set("config"):
config_file = v.get_string("config")
if config_file:
with open(config_file, encoding="utf-8") as handle:
config_data = handle.read()
v.read_config(config_data)
else:
try:
v.read_in_config()
except FileNotFoundError:
pass
if os.path.exists(v.get("tls_private_key")) and os.path.exists(
v.get("tls_certificate")
):
v.set("tls_enabled", True)
v.set("config_loaded", True)
|
