用户输入要从文本文件中删除的args值

时间:2017-11-23 10:42:25

标签: python python-3.x

当用户运行script.py -r sampletext时,我希望此操作从文件中删除该文本。

#!/usr/bin/python3
import shutil
import argparse

parser = argparse.ArgumentParser()
parser.add_argument('-r', action='store_true')
args = parser.parse_args()

if args.r:
        f = open("config.dat","r")
        lines = f.readlines()
        f.close()

我在这里尝试做的是在上面的代码读完文件之后,下面的代码将重新写入文件并省略args命令:

f = open("config.dat","w")
for line in lines:
    print(line)
    if line!= (args.r + "\n"): #Replace this string with your arg var
        f.write(line)
f.close()

1 个答案:

答案 0 :(得分:1)

指定action="store_true"时,表示args.r将存储为布尔值(True / False)。 如果你想在你必须使用-r选项之后检索参数:

parser.add_argument('-r', dest='to_remove')

然后您可以使用args.to_remove访问它。

请详细了解add_argument方法。