mirror of https://github.com/morpheus65535/bazarr
64 lines
2.0 KiB
Python
64 lines
2.0 KiB
Python
import os
|
|
import typing
|
|
from logging import NullHandler, getLogger
|
|
|
|
from pkg_resources import resource_stream
|
|
import yaml
|
|
|
|
from knowit.serializer import get_yaml_loader
|
|
|
|
logger = getLogger(__name__)
|
|
logger.addHandler(NullHandler())
|
|
|
|
|
|
class _Value(typing.NamedTuple):
|
|
code: str
|
|
default: str
|
|
human: str
|
|
technical: str
|
|
|
|
|
|
_valid_aliases = _Value._fields
|
|
|
|
|
|
class Config:
|
|
"""Application config class."""
|
|
|
|
@classmethod
|
|
def build(cls, path: typing.Optional[typing.Union[str, os.PathLike]] = None) -> 'Config':
|
|
"""Build config instance."""
|
|
loader = get_yaml_loader()
|
|
with resource_stream('knowit', 'defaults.yml') as stream:
|
|
cfgs = [yaml.load(stream, Loader=loader)]
|
|
|
|
if path:
|
|
with open(path, 'rb') as stream:
|
|
cfgs.append(yaml.load(stream, Loader=loader))
|
|
|
|
profiles_data = {}
|
|
for cfg in cfgs:
|
|
if 'profiles' in cfg:
|
|
profiles_data.update(cfg['profiles'])
|
|
|
|
knowledge_data = {}
|
|
for cfg in cfgs:
|
|
if 'knowledge' in cfg:
|
|
knowledge_data.update(cfg['knowledge'])
|
|
|
|
data: typing.Dict[str, typing.MutableMapping] = {'general': {}}
|
|
for class_name, data_map in knowledge_data.items():
|
|
data.setdefault(class_name, {})
|
|
for code, detection_values in data_map.items():
|
|
alias_map = (profiles_data.get(class_name) or {}).get(code) or {}
|
|
alias_map.setdefault('code', code)
|
|
alias_map.setdefault('default', alias_map['code'])
|
|
alias_map.setdefault('human', alias_map['default'])
|
|
alias_map.setdefault('technical', alias_map['human'])
|
|
value = _Value(**{k: v for k, v in alias_map.items() if k in _valid_aliases})
|
|
for detection_value in detection_values:
|
|
data[class_name][str(detection_value)] = value
|
|
|
|
config = Config()
|
|
config.__dict__ = data
|
|
return config
|