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