前言：上篇使用jinja2模板渲染数据是在读取yaml文件的时候，进行渲染。这种肯定不是关键字框架的渲染的方式。正常我们想要的而是在yaml文件中写入我们想要渲染的数据，然后读取之后渲染到测试yaml用例中。

1. 在yaml文件中引入变量

由于变量在yaml中先声明，后引用，所以不能直接渲染整个yaml。

解决的方法是：先读取yaml，转dict , 再获取config的数据并对其进行收集，最后再进行jinja2对测试用例的数据的渲染。

import yaml
from jinja2 import Template
from pathlib import Path

def read_yaml(file_path: Path):
    """
    读取 yaml 数据，转 python 类型
    :param file_path:
    :return: dict
    """
    with open(file_path, 'r', encoding='utf-8') as fp:
        read_yml_data = yaml.safe_load(fp) # 读到的是dict

    return {
        'config': read_yml_data.get('config') if read_yml_data.get('config') else {},
        'data': [[key, value] for key, value in read_yml_data.items() if key != 'config']
    }

if __name__ == '__main__':
    file_path = Path(__file__).parent.parent.joinpath('data', 'login.yml')
    res = read_yaml(file_path)
    print(res)
    print(res['config'])
    print(res['data'])
    print(dict(res['data']))
    print(type(res))  # -->dict

"""利用反射运行关键字"""
import jinja2
import json
from requests import Response
class RunByKey:
    def __init__(self, session, config: dict = None):
        self.session = session
        self.config = config # 获取yaml config 配置
        self.context = {}  # 变量容器
        self.yaml_variables = {}  # 用例变量

    @staticmethod
    def name(value):
        print(f'用例名称：{value}')

    def request(self, request_data: dict):
        print(f"执行request: {request_data}")
        res = self.session.send_request(**request_data)
        return res

    def run(self, data: dict):
        # 1.先获取到config中的变量variables
        config_variables = self.config.get('variables', {}) if self.config else {}
        print('获取到的config中的变量：', config_variables)
        # 2.更新到变量容器中
        self.context.update(config_variables)
        for key, value in data.items():
            if hasattr(self, key):
                # 执行的request前，需要替换的数据
                print(f'渲染之前的数据：{value}')
                t2 = jinja2.Template(json.dumps(value), # dict-->str
                                     variable_start_string='${',
                                     variable_end_string='}')
                value = json.loads(t2.render(**self.context))  # str-->dict
                print(f'渲染之后的数据：{value}')
                res = getattr(self, key)(value)  # 按关键字执行
                print(f"=========当前的key：{key}")
                if key == "request":
                    print(f"执行结果：{res.text}")
            else:
                print(f"关键字未定义： {key}")