31 lines
1.0 KiB
Python
31 lines
1.0 KiB
Python
import os
|
|
import sys
|
|
import tomllib
|
|
from pathlib import Path
|
|
from collections import UserDict
|
|
|
|
from .exceptions import ConfigLoadError
|
|
|
|
|
|
NODEAGENT_CONFIG_FILE = os.getenv('NODEAGENT_CONFIG_FILE')
|
|
NODEAGENT_DEFAULT_CONFIG_FILE = '/etc/node-agent/config.toml'
|
|
|
|
|
|
class ConfigLoader(UserDict):
|
|
def __init__(self, file: Path | None = None):
|
|
if file is None:
|
|
file = NODEAGENT_CONFIG_FILE or NODEAGENT_DEFAULT_CONFIG_FILE
|
|
self.file = Path(file)
|
|
self.data = self._load()
|
|
# todo: load deafult configuration
|
|
|
|
def _load(self):
|
|
try:
|
|
with open(self.file, 'rb') as config:
|
|
return tomllib.load(config)
|
|
# todo: config schema validation
|
|
except (OSError, ValueError) as readerr:
|
|
raise ConfigLoadError(f'Cannot read config file: {self.file}: {readerr}') from readerr
|
|
except tomllib.TOMLDecodeError as tomlerr:
|
|
raise ConfigLoadError(f'Bad TOML syntax in config file: {self.file}: {tomlerr}') from tomlerr
|