python中关于传递参数模块argprase的一些小坑

今天在写代码的时候遇到了一个关于parser的一些小坑，记录在此备用。

我们知道在python中可以用argprase来传递一些参数给代码执行，来看下面的例子，假设现在有一个test文件夹，下面有3个python文件，分别用a.py；b.py；c.py来表示，目录树如下。

 每一个的初始代码为一个简单的print函数。

1 #a.py
2 def out_a():
3     print("I am a.py")
4 
5 
6 if __name__ == '__main__':
7     out_a()
1 #b.py
2 def out_b():
3     print("I am b.py")
4 
5 
6 if __name__ == '__main__':
7     out_b()
1 #c.py
2 def out_c():
3     print("I am c.py")
4 
5 
6 if __name__ == '__main__':
7     out_c()

现在在a.py中引入模块argprase，并定义一些简单的参数，代码如下

 1 import argparse
 2 parser = argparse.ArgumentParser()
 3 parser.add_argument('--first_parameter', default='first')
 4 parser.add_argument('--second_parameter', default='second')
 5 parser.add_argument('--third_flag', action='store_true')
 6 args = parser.parse_args()
 7 
 8 
 9 def out_a():
10     print("I am a.py")
11 
12 
13 if __name__ == '__main__':
14     out_a()

这里面简单说一下第3个参数，这也是我今天想记录文章的原因，这个参数是argparse里面提供的开关布尔选项，actions记录的是一个动作，意味着在调用这个函数的时候，如果在命令行添加这个参数，则该参数为True，如果不添加这个参数，则该参数为False，归纳起来为如下的两个图。

这个是没有指定第3个参数的情况

 

这个是指定第3个参数的情况

对于这种开关布尔选项更为详细的介绍，可以参考知乎问题：Argparse中action的可选参数store_true,store_false到底是什么意思？

到目前为止没有出现问题，接下来，我希望b.py也使用参数，并且还希望使用a.py里面的函数，因此我对b.py进行如下修改。

 1 from a import out_a
 2 import argparse
 3 parser = argparse.ArgumentParser()
 4 parser.add_argument('--fourth_parameter', default='fourth')
 5 parser.add_argument('--fifth_parameter', default='fifth')
 6 parser.add_argument('--sixth_flag', action='store_true')
 7 args = parser.parse_args()
 8 
 9 
10 def out_b():
11     print("I am b.py")
12 
13 
14 if __name__ == '__main__':
15     out_b()
16     out_a()

然后同样的，我们分别用两种方式来测试b.py，效果如下。

这个是不使用参数的情况 

这个是使用参数的情况

可以看到报错了，当时我看到这里的时候想了很久，排除了拼写错误的情况以后，观察这里面的输出，发现看到的是a.py当中的3个参数，而不是b.py当中设置的参数，于是我将a.py和b.py的参数表打印出来，看到这样子的结果。

输出两个python文件的参数表

可以发现尽管我使用的是 from a import out_a ，但依然引入了a.py当中的参数表，并且后引入的b.py的参数表没有办法覆盖掉。下面在c.py中同样引入3个参数，然后引入b.py的方法，代码如下：

 1 from b import out_b
 2 import argparse
 3 parser = argparse.ArgumentParser()
 4 parser.add_argument('--seventh_parameter', default='seventh')
 5 parser.add_argument('--eighth_parameter', default='eighth')
 6 parser.add_argument('--ninth_flag', action='store_true')
 7 args = parser.parse_args()
 8 print(args)
 9 
10 def out_c():
11     print("I am c.py")
12 
13 
14 if __name__ == '__main__':
15     out_c()
16     out_b()

效果如下

不使用任何参数调用c.py

看到有3个参数列表输出就知道c.py的参数也是无效的了，验证一下。

使用参数调用c.py

 

解决方案：

其实只需要将所有的参数表放到同一个文件里面就可以了，比如utils.py，由于这里是同一个文件夹下的3个文件，在import调用的时候就只需要初始化一次所有参数就可以使用了，有点类似于C语言当中的全局变量，因为这个东西排查了一个下午，也是有点恼火了。

 

posted @ 2019-12-08 17:51  JCChan  阅读(1508)  评论(0)  编辑  收藏  举报