0

我想通过 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
4

1 回答 1

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 在评论中所说，您定义代码的方式不是定义主函数的方式。事实上，尝试这样的事情（另见这个 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()
于 2012-08-09T11:39:41.567 回答