jellyseer-exporter/src/jellyseer_exporter/config.py

97 lines
2.1 KiB
Python
Raw Normal View History

2023-10-22 16:59:09 +02:00
"""
Config module for Jellyseer prometheus collector.
"""
from collections.abc import Mapping
def config_from_yaml(yaml):
"""
Given a dictionary parsed from a yaml file return a config object.
"""
if not isinstance(yaml, Mapping):
return ConfigInvalid(
"Not a dictionary. Check the syntax of the YAML config file."
)
modules = {
key: config_module_from_yaml(value) for
key, value in
yaml.items()
}
invalid = [
f" - {key}: {module}" for
key, module in
modules.items() if
not module.valid
]
if invalid:
return ConfigInvalid("\n".join(
["Invalid module config entries in config file"] + invalid
))
if not modules:
return ConfigInvalid("Empty dictionary. No modules specified.")
return ConfigMapping(modules)
def config_module_from_yaml(yaml):
"""
Given a dictionary parsed from a yaml file return a module config object.
"""
if not isinstance(yaml, Mapping):
return ConfigInvalid(
"Not a dictionary. Check the syntax of the YAML config file."
)
if not yaml:
return ConfigInvalid(
"Empty dictionary. No pve API parameters specified."
)
return ConfigMapping(yaml)
class ConfigMapping(Mapping):
"""
Valid config object.
"""
valid = True
def __init__(self, mapping):
super().__init__()
self._mapping = mapping
def __str__(self):
num = len(self._mapping)
keys = ", ".join(self._mapping.keys())
return f"Valid config: with {num} keys: {keys}"
def __getitem__(self, key):
return self._mapping[key]
def __iter__(self):
return iter(self._mapping)
def __len__(self):
return len(self._mapping)
class ConfigInvalid:
"""
Invalid config object.
"""
# pylint: disable=too-few-public-methods
valid = False
def __init__(self, error="Unspecified reason."):
self._error = error
def __str__(self):
return f"Invalid config: {self._error}"