最近工作需要,需要根据输入的参数自动生成批量的python脚本。这些python脚本除了一些参数不同之外,其他内容都一样。
网上看到不少同志使用template模块来生成python脚本,于是自己也做了一些尝试。
1)首先需要先生成一个模板文件。命名为template.tmpl,如下。模板文件中使用${}括住的变量为需要传参进去修改的参数:
import sys
${A} is owned by youki
${B} is owned by kaiba
${C} is owned by maliku
2) 再定义一个python脚本,传入A|B|C的具体参数来生成实际的python脚本:
from string import Template
import sys
def build_tmpl(tmpl_filename, inst_filename, a, b, c):
tmpl_file = open(tmpl_filename, 'r')
tmpl = Template(tmpl_file.read())
code_str = []
code_str.append(tmpl.substitute(
A = a,
B = b,
C = c
))
inst_file = open(inst_filename, 'w')
inst_file.writelines(code_str)
inst_file.close()
if __name__ == '__main__':
a_str = 'Osiris'
b_str = 'Obelisk'
c_str = 'La'
template_file = './template.tmpl'
output_file = 'test.py'
build_tmpl(template_file, output_file, a_str, b_str, c_str)
具体逻辑如下:
- 读取template文件;
- 将template文件中的A,B,C分别替换为a,b,c
- 将替换后的文本写入实际的脚本文件中
3)最后生成的脚本文件如下:
import sys
Osiris is owned by youki
Obelisk is owned by kaiba
La is owned by maliku