目前,我有一个python脚本,它接收一个文本文件,检查由标记##Somethinghere#括起来的段落,并询问用户他/她要复制多少次。例如,如果我有文本文件:Random Text File
##RandomLine1##
Random Line 1
##RandomLine1##
Random Line 2
##RandomLine3##
Random Line 2
##RandomLine3##
End of file
系统将提示用户:
^{pr2}$
一旦用户输入数字,特定的封闭线被复制到指定的次数,并且标签被删除。复制多次后的文本将输出到指定的输出文件。在
要启动脚本,命令如下所示:python script.py inputfile outputfile
我想做的不是提示用户输入,而是用户可以选择输入循环的数量作为可选的命令行参数。比如:python script.py inputfile outputfile --RandomLine1 2 --RandomLine3 2
对于python脚本,这是可能的吗?我附上以下脚本的当前版本:import re
import argparse
pattern = '##([^#]*)##'
def main():
parser = argparse.ArgumentParser()
parser.add_argument('infile', type=argparse.FileType('r'))
parser.add_argument('outfile', type=argparse.FileType('w'))
args = parser.parse_args()
matcher = re.compile(pattern)
tagChecker = False
strList = []
for line in args.infile:
if tagChecker is True:
lineTest = matcher.search(line)
if lineTest:
tagChecker = False
for _ in range(int(raw_input('Loop ' + lineTest.string[2:-3] + ' how many times?')) - 1):
for copyLine in strList:
args.outfile.write(copyLine)
new_line = matcher.sub("", line)
args.outfile.write(new_line)
strList = []
continue
else:
strList.append(line)
args.outfile.write(line)
if tagChecker is False:
lineTest = matcher.search(line)
if lineTest:
tagChecker = True
new_line = matcher.sub("", line)
args.outfile.write(new_line)
else:
args.outfile.write(line)
args.infile.close()
args.outfile.close()
if __name__ == '__main__':
main()