| Linux server27.hostingraja.org 2.6.32-954.3.5.lve1.4.93.el6.x86_64 #1 SMP Wed Oct 4 17:04:29 UTC 2023 x86_64 Path : /opt/cloudlinux/venv/lib/python3.11/site-packages/setoptconf/source/ |
| Current File : //opt/cloudlinux/venv/lib/python3.11/site-packages/setoptconf/source/yamlfile.py |
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