使用遞歸替換參數${變量}


#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
@Project: py-gql-fast-test-api
@File: handle.py
@Author: zy7y
@Time: 2021/3/31 16:48
@Blog: https://www.cnblogs.com/zy7y
@Github: https://gitee.com/zy7y
@Desc: 處理方法
"""


class Handle:
    var_pool = {}

    @staticmethod
    def set_var(key: object, value: object):
        """存變量"""
        Handle.var_pool[key] = value

    @staticmethod
    def get_var(key: object, default: object = None):
        """取變量"""
        return Handle.var_pool.get(key, object)

    @staticmethod
    def handle_var_list(value: list):
        """
        處理參數替換~ 類型是list的邏輯
        """
        for index, liv in enumerate(value):
            if isinstance(liv, str) and liv[:2] == '${' and liv[-1] == '}':
                # 獲得key部分內容
                liv = liv[2:-1]
                # 取到key對應的value,複製給當前下標
                value[index] = Handle.get_var(liv)
            elif isinstance(liv, list):
                # 如果 子元素還是 list 就再迭代list
                Handle.handle_var_list(liv)
            elif isinstance(liv, dict):
                # 如果子元素是 dict 就再迭代字典
                Handle.handle_var(liv)

    @staticmethod
    def handle_var(var: dict) -> dict:
        """處理自定義參數類型"""
        for key, value in var.items():
            # 如果是字典就 迭代字典
            if isinstance(value, dict):
                Handle.handle_var(value)
            # 如果是list 就迭代list, 遞歸 list處理方法
            elif isinstance(value, list):
                Handle.handle_var_list(value)
            # 如果是str 就替換
            elif isinstance(value, str) and value[:2] == '${' and value[-1] == '}':
                # 獲得key部分內容
                value = value[2:-1]
                # 取到key對應的value
                var[key] = Handle.get_var(value)
        return var


if __name__ == '__main__':

    Handle.var_pool = {"name": "att", "age": 18, "height": 17.8}
    # 需要處理的字典
    data = {
        #  字典 提取
        "name": "${name}",
        # 字典嵌套字典提取
        "info": {"${name}": ["${age}", "${height}"]},
            "lists": ["${name}", ["${age}", "${height}"], {"name": "${name}"}]

    }
    print(Handle.handle_var(data))

可能有Bug

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章