标签：函数 render python yaml jinja2 引用 path import

前言

经常看到很多同学问到，如何在 yaml 文件中引用一个 python 的函数？

问题分析

大家对yaml文件还处于比较陌生的阶段，yaml 和 json 文件本质上是一样的，都是静态的文件，当然不能直接引用 python 的函数。
那这时候就有人问到了，那为什么 httprunner 框架可以在yaml文件中引用函数呢？

• 这是因为 httprunner 框架封装过对 yaml 文件的读取了，它是先读取文件内容，正则提取到 ${} 括号里面的函数内容，再把函数的值替换过去

那么我们能不能实现这种效果呢？

• 当然是可以的，可以参考httprunner的实现，也可以用到 python 的模板 jinja2 来实现。

使用模板可以编写出可读性更好，更容易理解和维护的代码，并且使用范围非常广泛，因此怎么使用模板主要取决于我们的想象力和创造力。
python的模板库jinja2 功能是非常强大的。

jinja2 模板库

先需要pip安装

pip install jinja2

render 函数实现

在yaml文件中，通过 {{ 函数名称() }} 来引用函数

写个 render 函数读取 yaml 文件内容

import os
import jinja2
import yaml
import random


def render(tpl_path, **kwargs):
    path, filename = os.path.split(tpl_path)
    return jinja2.Environment(loader=jinja2.FileSystemLoader(path or './')
    ).get_template(filename).render(**kwargs)

读取到的yaml文件本质上都是字符串来读取的，通过jinja2 模板来读取，会先把函数的值替换进去。最后再转成python的dict结构

import os
import jinja2
import yaml
import random


def render(tpl_path, **kwargs):
    path, filename = os.path.split(tpl_path)
    return jinja2.Environment(loader=jinja2.FileSystemLoader(path or './')
    ).get_template(filename).render(**kwargs)


# yaml 文件调用以下函数
def rand_str():
    return str(random.randint(1000000, 2000000))


if __name__ == '__main__':
    r = render("aa.yml", **{"tel": rand_str})
    print(r)
    print(yaml.safe_load(r))

运行结果

name: yoyo
age: 22
tel: 1616350
{'name': 'yoyo', 'age': 22, 'tel': 1616350}

标签：函数,render,python,yaml,jinja2,引用,path,import
来源： https://www.cnblogs.com/yoyoketang/p/15906054.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。