420fd84f6b
- load config from env vars (eg. TABBY_NETWORK_HOST) - remove print statements - improve command line args automation
89 lines
3.1 KiB
Python
89 lines
3.1 KiB
Python
import yaml
|
|
import pathlib
|
|
from loguru import logger
|
|
from typing import Optional, Union, get_origin, get_args
|
|
from os import getenv
|
|
|
|
from common.utils import unwrap, merge_dicts
|
|
from common.config_models import tabby_config_model
|
|
import common.config_models
|
|
|
|
|
|
class TabbyConfig(tabby_config_model):
|
|
def load_config(self, arguments: Optional[dict] = None):
|
|
"""load the global application config"""
|
|
|
|
# config is applied in order of items in the list
|
|
configs = [
|
|
self._from_file(pathlib.Path("config.yml")),
|
|
self._from_environment(),
|
|
self._from_args(unwrap(arguments, {})),
|
|
]
|
|
|
|
merged_config = merge_dicts(*configs)
|
|
|
|
for field in tabby_config_model.model_fields.keys():
|
|
value = unwrap(merged_config.get(field), {})
|
|
model = getattr(common.config_models, f"{field}_config_model")
|
|
|
|
setattr(self, field, model.parse_obj(value))
|
|
|
|
def _from_file(self, config_path: pathlib.Path):
|
|
"""loads config from a given file path"""
|
|
|
|
# try loading from file
|
|
try:
|
|
with open(str(config_path.resolve()), "r", encoding="utf8") as config_file:
|
|
return unwrap(yaml.safe_load(config_file), {})
|
|
except FileNotFoundError:
|
|
logger.info(f"The '{config_path.name}' file cannot be found")
|
|
except Exception as exc:
|
|
logger.error(
|
|
f"The YAML config from '{config_path.name}' couldn't load because of "
|
|
f"the following error:\n\n{exc}"
|
|
)
|
|
|
|
# if no config file was loaded
|
|
return {}
|
|
|
|
def _from_args(self, args: dict):
|
|
"""loads config from the provided arguments"""
|
|
config = {}
|
|
|
|
config_override = unwrap(args.get("options", {}).get("config"))
|
|
if config_override:
|
|
logger.info("Config file override detected in args.")
|
|
config = self.from_file(pathlib.Path(config_override))
|
|
return config # Return early if loading from file
|
|
|
|
for key in tabby_config_model.model_fields.keys():
|
|
override = args.get(key)
|
|
if override:
|
|
if key == "logging":
|
|
# Strip the "log_" prefix from logging keys if present
|
|
override = {k.replace("log_", ""): v for k, v in override.items()}
|
|
config[key] = override
|
|
|
|
return config
|
|
|
|
def _from_environment(self):
|
|
"""loads configuration from environment variables"""
|
|
|
|
config = {}
|
|
|
|
for field_name in tabby_config_model.model_fields.keys():
|
|
section_config = {}
|
|
for sub_field_name in getattr(
|
|
tabby_config_model(), field_name
|
|
).model_fields.keys():
|
|
setting = getenv(f"TABBY_{field_name}_{sub_field_name}".upper(), None)
|
|
if setting is not None:
|
|
section_config[sub_field_name] = setting
|
|
|
|
config[field_name] = section_config
|
|
|
|
return config
|
|
|
|
|
|
# Create an empty instance of the shared var to make sure nothing breaks
|
|
config: TabbyConfig = TabbyConfig()
|