wordmarker.contexts.yaml_context 源代码

import copy
from wordmarker.contexts.context import Context
from wordmarker.loaders.yaml_resource_loader import YamlResourceLoader
from wordmarker.utils import YamlUtils


class YamlContext(Context, YamlResourceLoader):
    """
    ::

        yaml文件的上下文
    """

    def __init__(self, path):
        super().__init__()
        self.__path = path
        self.__data = None
        self._init()

    def _init(self):
        """
        ::

            初始化yaml上下文

        :return: 从yaml文件中加载的数据，类型为dict
        """
        resource = self.get_resource(self.path)
        self.__data = self.load(resource)

    def get_yaml(self):
        """
        .. note::

            获取从yaml文件中读取的数据，类型为dict

        :return: - path为文件，返回一个字典，内容为yaml文件的内容

                 - path为目录，返回一个嵌套的字典

                    - key为yaml文件的绝对路径

                    - value为yaml文件的内容，是一个字典
        """
        return self.__data

    def get_value(self, prop):
        """
        .. note::

            从yaml字典中，根据属性获取对应的值

            加载多个yaml文件，排在后面的文件里的值，会覆盖前面的文件里的值

        :param prop: 属性，用 ``.`` 分隔，例如，``pdbc.engine.url``
        :return: - yaml字典中对应的值
        """
        prop_list = prop.split('.')
        if self.get_resource(self.path).is_file():
            yaml_dict = self.get_yaml()
            value = YamlUtils().get_value(yaml_dict, prop_list, prop, self.get_resource(self.path).get_file())
            return value
        else:
            yaml = self.get_yaml()
            value = None
            for key, val in yaml.items():
                temp_list = copy.deepcopy(prop_list)
                v = YamlUtils().get_value(val, temp_list, prop, key)
                if v is not None:
                    value = v
            return value

    @property
    def path(self):
        """
        .. note::

            存放yaml文件的路径，可以是文件，也可以是目录

        :return: - yaml文件的路径
        """
        return self.__path

    @path.setter
    def path(self, path):
        self.__path = path