如何用Jinja2生成包含类成员对象的Python列表?
解决方案
要生成不带引号的成员变量列表,核心是避免传递字符串列表给Jinja,而是直接在模板中构造Python表达式。以下是具体修改:
1. 修改Jinja模板
将模板中生成parameters1和parameters2的代码替换为直接循环参数构造列表元素:
class {{class_name}}({{base_class_name}}): def __init__(self, {{ parameters|join(', ') }}): {{comment}} super().__init__() {% for param in parameters%} self.{{param}} = {{param}} {%- endfor %} # 生成目标列表 self.parameters = [{% for param in parameters %}self.{{ param }}{% if not loop.last %}, {% endif %}{% endfor %}]
2. 简化Python脚本
不需要再生成字符串列表parameter_list,也无需注册product过滤器,简化后的脚本如下:
from pathlib import Path import jinja2 def create_class_from_template(): templates_path = Path(__file__).parent / 'templates' jinja_env = jinja2.Environment(loader=jinja2.FileSystemLoader(templates_path)) parameters = ['param1', 'param2', 'param3'] template_variables = { 'class_name': 'TestClass', 'base_class_name': 'BaseClass', 'comment': '"""My generated class"""', 'parameters': parameters } template = jinja_env.get_template('class_template.py.jinja2') my_class = template.render(template_variables) with open('my_new_class.py', 'w') as file: file.write(my_class) create_class_from_template()
3. 生成结果
修改后运行脚本,生成的Python文件会包含:
self.parameters = [self.param1, self.param2, self.param3]
原理说明
原来的方法之所以生成带引号的字符串列表,是因为你传递给Jinja的是字符串类型的元素,Jinja渲染时会自动为字符串添加引号。而直接在模板中构造self.{{param}},本质是生成Python表达式,而非字符串,因此输出的是成员变量本身,不带引号。
内容的提问来源于stack exchange,提问作者Dave
相关产品推荐
相关产品推荐

