Mini Shell

import codecs
import json

from .filebased import FileBasedSource


__all__ = ("JsonFileSource",)


class JsonFileSource(FileBasedSource):
    def __init__(self, *args, **kwargs):
        self.encoding = kwargs.pop("encoding", "utf-8")
        super(JsonFileSource, 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 = json.loads(content)
        if not content:
            return None

        if not isinstance(content, dict):
            raise TypeError("JSON files must contain only objects")

        for setting in settings:
            if setting.name in content:
                setting.value = content[setting.name]

        return settings