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
|