%PDF-
%PDF-
Mini Shell
Mini Shell
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
Zerion Mini Shell 1.0