在Python中通过命令行参数编辑文本文件

时间:2012-08-09 10:14:08

标签: python command-line-arguments

我想通过Python中的命令行参数传递整数来编辑文本文件。但是我的代码不起作用,有人可以指出我错在哪里。

    import sys, argparse
    def main(argv=None):
    if argv is None:
        argv=sys.argv[1:]
        p = argparse.ArgumentParser(description="Editing omnetpp.ini")
        p.add_argument('arg1', action='store', default= 1, type=int, help="number of clients")
        args = p.parse_args(argv)
        n = args.arg1
        f = open('C:\\Users\Abcd\Desktop\Omnet\omnetpp.ini', 'a')
        for i in range(n):
            f.write('*.voipClient['+str(i)+'].udpApp['+str(i)+'].destAddresses = "voipGateway"\n')
            f.write('*.voipGateway.udpApp['+str(i)+'].destAddresses   = "voipClient['+str(i)+']"\n')
        f.close()

如果通过命令行参数传递整数5,那么它应该在文本文件中添加以下行,这不会发生 输出

*.voipClient[0].udpApp[0].destAddresses = "voipGateway"
*.voipGateway.udpApp[0].destAddresses   = "voipClient[0]"
*.voipClient[1].udpApp[1].destAddresses = "voipGateway"
*.voipGateway.udpApp[1].destAddresses   = "voipClient[1]"
*.voipClient[2].udpApp[2].destAddresses = "voipGateway"
*.voipGateway.udpApp[2].destAddresses   = "voipClient[2]"
*.voipClient[3].udpApp[3].destAddresses = "voipGateway"
*.voipGateway.udpApp[3].destAddresses   = "voipClient[3]"
*.voipClient[4].udpApp[4].destAddresses = "voipGateway"
*.voipGateway.udpApp[4].destAddresses   = "voipClient[4]"

我正在遵循以下步骤:

  1. 代码保存在test.py
  2. 从命令行C:\ Users \ Abcd \ Desktop> python test.py 5

1 个答案:

答案 0 :(得分:1)

不要在循环中关闭文件,一旦它关闭就不能再写入它了(实际上,如果你尝试写入一个已关闭的文件对象,就应该抛出一个错误)。 相反,在循环后关闭它。 另外,要将每个句子放在一个新行上,请使用换行符号\n结束该字符串(按“ENTER”键)。

f = open('C:\\Users\Abcd\Desktop\Omnet\omnetpp.ini', 'a')
for i in range(n):
    f.write('*.voipClient['+str(i)+'].udpApp['+str(i)+'].destAddresses = "voipGateway"\n')
    f.write('*.voipGateway.udpApp['+str(i)+'].destAddresses   = "voipClient['+str(i)+']"\n')
f.close()

修改

顺便说一下,正如Rostyslav Dzinko在评论中所说,你定义代码的方式是你如何定义主函数。事实上,尝试这样的事情(另见this SO question):

if __name__ == '__main__':
    p = argparse.ArgumentParser(description="Editing omnetpp.ini")
    p.add_argument('arg1', action='store', default= 1, type=int, help="number of clients")
    args = p.parse_args()