mirror of
https://github.com/yavook/kiwi-scp.git
synced 2024-11-22 12:53:00 +00:00
83 lines
2.2 KiB
Python
83 lines
2.2 KiB
Python
import logging
|
|
import re
|
|
import os
|
|
import yaml
|
|
|
|
from .core import KIWI_ROOT, KIWI_CONF_NAME
|
|
|
|
###########
|
|
# CONSTANTS
|
|
|
|
DEFAULT_KIWI_CONF_NAME = KIWI_ROOT + "/default.kiwi.yml"
|
|
|
|
|
|
class Config:
|
|
__yml_content = {}
|
|
|
|
def __key_resolve(self, key):
|
|
# "a:b:c" => path = ['a', 'b'], key = 'c'
|
|
path = key.split(':')
|
|
path, key = path[:-1], path[-1]
|
|
|
|
# resolve path
|
|
container = self.__yml_content
|
|
for step in path:
|
|
container = container[step]
|
|
|
|
return container, key
|
|
|
|
def __getitem__(self, key):
|
|
container, key = self.__key_resolve(key)
|
|
return container[key]
|
|
|
|
def __setitem__(self, key, value):
|
|
container, key = self.__key_resolve(key)
|
|
container[key] = value
|
|
|
|
def __str__(self):
|
|
# dump yml content
|
|
yml_string = yaml.dump(self.__yml_content, default_flow_style=False, sort_keys=False)
|
|
|
|
# insert newline before every main key
|
|
yml_string = re.sub(r'^(\S)', r'\n\1', yml_string, flags=re.MULTILINE)
|
|
|
|
# extract header comment from default config
|
|
with open(DEFAULT_KIWI_CONF_NAME, 'r') as stream:
|
|
yml_header = stream.read().strip()
|
|
yml_header = re.sub(r'^[^#].*', r'', yml_header, flags=re.MULTILINE).strip()
|
|
yml_string = "{}\n{}".format(yml_header, yml_string)
|
|
|
|
return yml_string
|
|
|
|
def __update_from_file(self, filename):
|
|
with open(filename, 'r') as stream:
|
|
try:
|
|
self.__yml_content.update(yaml.safe_load(stream))
|
|
except yaml.YAMLError as exc:
|
|
logging.error(exc)
|
|
|
|
def __save_to_file(self, filename):
|
|
with open(filename, 'w') as stream:
|
|
stream.write(str(self))
|
|
|
|
@classmethod
|
|
def default(cls):
|
|
result = cls()
|
|
result.__update_from_file(DEFAULT_KIWI_CONF_NAME)
|
|
|
|
with open(KIWI_ROOT + "/version-tag", 'r') as stream:
|
|
result.__yml_content["version"] = stream.read().strip()
|
|
|
|
return result
|
|
|
|
@classmethod
|
|
def load(cls):
|
|
result = cls.default()
|
|
|
|
if os.path.isfile(KIWI_CONF_NAME):
|
|
result.__update_from_file(KIWI_CONF_NAME)
|
|
|
|
return result
|
|
|
|
def save(self):
|
|
self.__save_to_file(KIWI_CONF_NAME)
|