| 1234567891011121314151617181920212223242526272829303132 |
- import codecs
- import yaml
- from .filebased import FileBasedSource
- __all__ = ("YamlFileSource",)
- class YamlFileSource(FileBasedSource):
- def __init__(self, *args, **kwargs):
- self.encoding = kwargs.pop("encoding", "utf-8")
- super(YamlFileSource, self).__init__(*args, **kwargs)
- def get_settings_from_file(self, file_path, settings, manager=None):
- content = codecs.open(file_path, "r", self.encoding).read().strip()
- if not content:
- return None
- content = yaml.safe_load(content)
- if not content:
- return None
- if not isinstance(content, dict):
- raise TypeError("YAML files must contain only mappings")
- for setting in settings:
- if setting.name in content:
- setting.value = content[setting.name]
- return settings
|
