【Lab】Python改bat文件
给出一个特定的树形结构，每一层的数字依次递增后，按照从上到下，同时从左到右这样的顺序生成。这么说还是不太明白，比如下面这个简单的树形结构。
按照顺序应该写成这样[3_0, 3_4, 4_1, 4_2]
对应的bat文件内容是这样的。
%%%%%%%%%%%%%%%%[3_0]%%%%%%%%%%%%%%%%
totalLMH2 F:\xdm\test\mainBuilding3\case2\new\[3].jpg F:\xdm\test\mainBuilding3\case2\new\[0].jpg F:\xdm\test\mainBuilding3\case2\new\single\QP32\
removeRepeatedCoors F:\xdm\test\mainBuilding3\case2\new\single\QP32\LMH\
dTAndHP2_1 F:\xdm\test\mainBuilding3\case2\new\single\QP32\LMH\
EnergyOpti_11 F:\xdm\test\mainBuilding3\case2\new\single\QP32\LMH\
perspectiveTAndLumaT F:\xdm\test\mainBuilding3\case2\new\ F:\xdm\test\mainBuilding3\case2\new\single\QP32\ [3] [0]
jpgToYUV F:\xdm\test\mainBuilding3\case2\new\single\QP32\PWW\ F:\xdm\test\mainBuilding3\case2\new\single\QP32\bat\photometricTransfImageYUV\
combine5frames F:\xdm\test\mainBuilding3\case2\new\yuv\[3].yuv F:\xdm\test\mainBuilding3\case2\new\yuv\[0].yuv F:\xdm\test\mainBuilding3\case2\new\single\QP32\LMH\fiveFramesForHEVCCoding.yuv
HEVC_5Frames -c QP32_01.cfg -c testPaper.cfg -i F:\xdm\test\mainBuilding3\case2\new\single\QP32\LMH\fiveFramesForHEVCCoding.yuv -o F:\xdm\test\mainBuilding3\case2\new\single\QP32\PWW\recYUV\rec[3_0].yuv > F:\xdm\test\mainBuilding3\case2\new\single\QP32\PWW\result\result[3_0].txt
%%%%%%%%%%%%%%%%[3]%%%%%%%%%%%%%%%%
HEVC_5Frames -c QP32_01.cfg -c testPaper.cfg -i F:\xdm\test\mainBuilding3\case2\new\yuv\[3].yuv -o F:\xdm\test\mainBuilding3\case2\new\single\QP32\PWW\recYUV\rec[3].yuv > F:\xdm\test\mainBuilding3\case2\new\single\QP32\PWW\result\result[3].txt
%%%%%%%%%%%%%%%%3_4%%%%%%%%%%%%%%%%
totalLMH2 F:\xdm\test\mainBuilding3\case2\new\3.jpg F:\xdm\test\mainBuilding3\case2\new\4.jpg F:\xdm\test\mainBuilding3\case2\new\single\QP32\
removeRepeatedCoors F:\xdm\test\mainBuilding3\case2\new\single\QP32\LMH\
dTAndHP2_1 F:\xdm\test\mainBuilding3\case2\new\single\QP32\LMH\
EnergyOpti_11 F:\xdm\test\mainBuilding3\case2\new\single\QP32\LMH\
perspectiveTAndLumaT F:\xdm\test\mainBuilding3\case2\new\ F:\xdm\test\mainBuilding3\case2\new\single\QP32\ 3 4
jpgToYUV F:\xdm\test\mainBuilding3\case2\new\single\QP32\PWW\ F:\xdm\test\mainBuilding3\case2\new\single\QP32\bat\photometricTransfImageYUV\
combine5frames F:\xdm\test\mainBuilding3\case2\new\yuv\3.yuv F:\xdm\test\mainBuilding3\case2\new\yuv\4.yuv F:\xdm\test\mainBuilding3\case2\new\single\QP32\LMH\fiveFramesForHEVCCoding.yuv
HEVC_5Frames -c QP32_02.cfg -c testPaper.cfg -i F:\xdm\test\mainBuilding3\case2\new\single\QP32\LMH\fiveFramesForHEVCCoding.yuv -o F:\xdm\test\mainBuilding3\case2\new\single\QP32\PWW\recYUV\rec3_4.yuv > F:\xdm\test\mainBuilding3\case2\new\single\QP32\PWW\result\result3_4.txt
%%%%%%%%%%%%%%%%4_1%%%%%%%%%%%%%%%%
totalLMH2 F:\xdm\test\mainBuilding3\case2\new\4.jpg F:\xdm\test\mainBuilding3\case2\new\1.jpg F:\xdm\test\mainBuilding3\case2\new\single\QP32\
removeRepeatedCoors F:\xdm\test\mainBuilding3\case2\new\single\QP32\LMH\
dTAndHP2_1 F:\xdm\test\mainBuilding3\case2\new\single\QP32\LMH\
EnergyOpti_11 F:\xdm\test\mainBuilding3\case2\new\single\QP32\LMH\
perspectiveTAndLumaT F:\xdm\test\mainBuilding3\case2\new\ F:\xdm\test\mainBuilding3\case2\new\single\QP32\ 4 1
jpgToYUV F:\xdm\test\mainBuilding3\case2\new\single\QP32\PWW\ F:\xdm\test\mainBuilding3\case2\new\single\QP32\bat\photometricTransfImageYUV\
combine5frames F:\xdm\test\mainBuilding3\case2\new\yuv\4.yuv F:\xdm\test\mainBuilding3\case2\new\yuv\1.yuv F:\xdm\test\mainBuilding3\case2\new\single\QP32\LMH\fiveFramesForHEVCCoding.yuv
HEVC_5Frames -c QP32_02.cfg -c testPaper.cfg -i F:\xdm\test\mainBuilding3\case2\new\single\QP32\LMH\fiveFramesForHEVCCoding.yuv -o F:\xdm\test\mainBuilding3\case2\new\single\QP32\PWW\recYUV\rec4_1.yuv > F:\xdm\test\mainBuilding3\case2\new\single\QP32\PWW\result\result4_1.txt
%%%%%%%%%%%%%%%%4_2%%%%%%%%%%%%%%%%
totalLMH2 F:\xdm\test\mainBuilding3\case2\new\4.jpg F:\xdm\test\mainBuilding3\case2\new\2.jpg F:\xdm\test\mainBuilding3\case2\new\single\QP32\
removeRepeatedCoors F:\xdm\test\mainBuilding3\case2\new\single\QP32\LMH\
dTAndHP2_1 F:\xdm\test\mainBuilding3\case2\new\single\QP32\LMH\
EnergyOpti_11 F:\xdm\test\mainBuilding3\case2\new\single\QP32\LMH\
perspectiveTAndLumaT F:\xdm\test\mainBuilding3\case2\new\ F:\xdm\test\mainBuilding3\case2\new\single\QP32\ 4 2
jpgToYUV F:\xdm\test\mainBuilding3\case2\new\single\QP32\PWW\ F:\xdm\test\mainBuilding3\case2\new\single\QP32\bat\photometricTransfImageYUV\
combine5frames F:\xdm\test\mainBuilding3\case2\new\yuv\4.yuv F:\xdm\test\mainBuilding3\case2\new\yuv\2.yuv F:\xdm\test\mainBuilding3\case2\new\single\QP32\LMH\fiveFramesForHEVCCoding.yuv
HEVC_5Frames -c QP32_03.cfg -c testPaper.cfg -i F:\xdm\test\mainBuilding3\case2\new\single\QP32\LMH\fiveFramesForHEVCCoding.yuv -o F:\xdm\test\mainBuilding3\case2\new\single\QP32\PWW\recYUV\rec4_2.yuv > F:\xdm\test\mainBuilding3\case2\new\single\QP32\PWW\result\result4_2.txt其中被 []括起来的是主要需要修改的地方，第二部分很简单，是根节点（这里是3）。内容结构与其余有点不一样，需要单独处理。
通过观察发现，实际上需要改的地方几乎使用第一行%%%%%%%%%%%%%%%%[\d+_\d+]%%%%%%%%%%%%%%%%里面括起来的编号。处理好其中一部分，其他的for循环就能搞定。写两个函数，一个单独处理根节点部分，另一个处理剩下的部分。
会用到了一个lab_node.bat的模板文件，内容和上面编好的bat几乎一样，读取这个bat文件，用新的数据替换相应部分就好了。共有30部分，若提供的树形结构复杂，超过了30部分，需要自己补充，注意保持QPXX_XX.cfg保持1234的顺序。
上面的修改好后，就剩下全局修改了。因为路径不会是\mainBuilding3\case2\new\，得看具体需求，另外case是多图像才有的。最后的最后，将文件里面的QP32替换成QP27和QP22生成两个副本。在程序中我使用了w+模式（可读写）打开文件，每次读后需要写或者写后需要读，都需要使用seek(0)将指针定位到文本开头。
这样下来，运行程序就会同时生成三个bat文件了。
import re
# 若树的路径数大于30，请到lab_node.bat增加一些部分。
# 注意每部分的最后一行，QPXX_XX.cfg保持1234的顺序
# 1.从上到下、从左到右，根据排序后的树形图依次填入
l = ('7_9', '7_13', '9_14', '13_3', '13_4', '13_5', '13_6',
'13_10', '13_11', '13_12', '13_15', '13_16', '13_17')
# 2.填写方法名，层数, reference，多图像的时候填入case
# 单图像路径大概是这样xdm\test\patheon\reference1\
# 多图像路径大概是xdm\test\patheon\case2\reference1\
method, layer, refer, case = 'patheon', '3', 'reference2', ''
# 单图像
if not case:
file_name = method + '_' + layer + refer + '_'
# 多图像
else:
file_name = file_name = method + '_' + layer + case + '_' + refer + '_'
def check():
if len(l) > 30:
raise Exception('路径数超过30！请到模板文件lab_node.bat下添加缺少的部分')
def split_to_frames():
"""
输入模板文件，分割成若*分并返回
:return: 分割好的部分的列表
"""
# 输入文件
with open('lab_node.bat', encoding='utf-8') as f:
# 读取全部内容
text_plain = f.read()
# 分成结构相似的多个部分, DOTALL模式'.'匹配任意，包括换行符
each_frame_p = re.compile(r'%+\d+.*?%+.*?HEVC_5Frames.*?txt', flags=re.DOTALL)
frames = re.findall(each_frame_p, text_plain)
return frames
def modify_frame(out_bat, path, frame):
"""
处理非根节点部分
:param out_bat: 输出文件
:param path: 树的路径
:param frame: 每一部分
"""
# 处理每个frame的第一行%%%
line0_p = re.compile(r'%+\d+_\d+%+')
modify0 = re.sub(line0_p, '%' * 16 + path + '%' * 16, frame)
# 处理totalLMH2那行,只有该行带有 数字.jpg, 且有两个
jpg1_p = re.compile(r'totalLMH2.*?\d+.jpg')
jpg2_p = re.compile(r'jpg.*?\d+.jpg')
# 第一个数字
modify1 = re.sub(jpg1_p, r'totalLMH2 F:\\xdm\\test\\mainBuilding3\\new\\'
+ path.split('_')[0] + '.jpg', modify0)
# 第二个数字
modify2 = re.sub(jpg2_p, r'jpg F:\\xdm\\test\\mainBuilding3\\new\\' + path.split('_')[1] + '.jpg', modify1)
# .*?为懒惰模式，匹配成功第一个就停止匹配
# 替换perspectiveTAndLumaT这行末尾的两个数字
line6_p = re.compile(r'perspectiveTAndLumaT.*?QP\d+\\\s*\d+\s*\d+')
modify3 = re.sub(line6_p, r'perspectiveTAndLumaT '
r'F:\\xdm\\test\\mainBuilding3\\new\\ '
r'F:\\xdm\\test\\mainBuilding3\\new\\single\\QP32\\ ' + path.split('_')[0]
+ r' ' + path.split('_')[1],
modify2)
# 替换'.yuv'，共3个带数字
yuv1_p = re.compile(r'frames.*?YUV\\\d+.yuv')
yuv2_p = re.compile(r'yuv\\\d+.yuv\s*F.*?QP')
yuv3_p = re.compile(r'-o F.*?rec\d+_\d+')
modify4 = re.sub(yuv1_p,
r'frames F:\\xdm\\test\\mainBuilding3\\new\\yuv\\' + path.split('_')[0] + '.yuv',
modify3)
modify5 = re.sub(yuv2_p, r'yuv\\' + path.split('_')[0]
+ r'.yuv F:\\xdm\\test\\mainBuilding3\\new\\yuv\\' + path.split('_')[1] +
r'.yuv F:\\xdm\\test\\mainBuilding3\\new\\single\\QP', modify4)
modify6 = re.sub(yuv3_p, r'-o F:\\xdm\\test\\mainBuilding3\\new\\single\\QP32\\PWW\\recYUV\\rec' + path, modify5)
# 替换'.txt', 共1个带数字
txt_p = re.findall(r'\d+_\d+.txt', modify6)[0]
modify7 = re.sub(txt_p, path + '.txt', modify6)
out_bat.write(modify7 + '\n\n')
def modify_root_frame(out_bat, node, frame1):
"""
处理根节点部分
:param out_bat: 输出文件
:param node: 根节点
:param frame1: 第一部分
"""
# 处理每个frame的第一行%%%
line0_p = re.compile(r'%+\d+%+')
modify0 = re.sub(line0_p, '%' * 16 + node + '%' * 16, frame1)
# 替换yuv,共两个,是一样的,选第一个就行
yuv_p = re.findall(r'\d+.yuv', modify0)[0]
modify1 = re.sub(yuv_p, node + '.yuv', modify0)
# 替换txt,共一个
txt_p = re.findall(r'\d+.txt', modify1)[0]
modify2 = re.sub(txt_p, node + '.txt', modify1)
out_bat.write(modify2 + '\n\n')
def replace_all(out_bat):
"""
替换掉模板文件的路径为当前指定
:param out_bat: 输出文件
"""
# 指针在文本末，直接read为空，需要定位到文本开始
out_bat.seek(0)
text = out_bat.read()
# 单图像
if not case:
text = re.sub(r'mainBuilding3\\new', method + r'\\' + layer + refer, text)
# 多图像
else:
text = re.sub(r'mainBuilding3\\new', method + r'\\' + case + r'\\' + layer + refer, text)
# 替换后需要重新写入，由于上面进行过读操作，再将指针指向文本开头，覆盖
out_bat.seek(0)
out_bat.write(text)
def create_other_qp(out_bat):
"""
将QP32替换成QP27和QP22
:param out_bat: QP32的输出文件
"""
other_qps = ('27', '22')
# 刚才覆写了，所以定位到文本开头才能读到
out_bat.seek(0)
text = out_bat.read()
for qp in other_qps:
with open(file_name + qp + '.bat', 'w', encoding='utf-8') as f:
new_qp_text = re.sub('QP32', 'QP' + qp, text)
f.write(new_qp_text)
def run():
# 检查路径数是否超过30
check()
# 第一部分
first_path = l[0]
# 根节点
root = first_path.split('_')[0]
# 获得所有部分
frames = split_to_frames()
with open(file_name + '32.bat', 'w+', encoding='utf-8') as f:
# 1. 处理第一部分
modify_frame(f, first_path, frames[0])
# 2. 再处理根节点部分（第二部分）
modify_root_frame(f, root, frames[1])
# 3. 处理余下部分
# l的index,第一部分处理过了，所以从1开始.
i = 1
for frame in frames[2:len(l) + 1]:
modify_frame(f, l[i], frame)
i += 1
# 在全文进行替换
replace_all(f)
create_other_qp(f)
if __name__ == '__main__':
run()虽然写这个程序的时间用手动改也能完成了。不过下次再给我这种任务，就不大费时间了，根据具体树形结构，修改下列表就行了。
by @sunhaiyu
2017.6.20