一日一技：快速实现Python 命令行参数

Posted 2020-12-21

一日一技：快速实现Python 命令行参数

摄影：产品经理
厨师：kingname
我们有时候需要使用 Python 开发一些命令行工具，并在运行命令的时候指定一些参数，例如：

python3 xxx.py --name kingname --salary 999999

对于简单的，只有几个参数的工具，我们可以使用 sys.argv来获取。

如果我们还需要在参数里面输出帮助信息，并且同时兼顾两种格式：

python3 xx.py --name=kingname --salary=999

python3 xxx.py --name kingname --salary=999

那么使用 sys模块就不好用了。此时我们可以使用 argparse。但是如果你阅读它的官方文档：https://docs.python.org/zh-cn/3/howto/argparse.html，你就会发现 argparse写起来还是有点麻烦。

那么有没有简单的方法实现命令行参数呢？答案就是 fire这个库。

首先我们来安装一下它：

python3 -m pip install fire

然后我们来实现一个简单的支持命令行的程序：

import fire

def intro(name, salary):
    print(f‘我的名字是：{name}, 我的月薪是{salary}‘)

if __name__ == ‘__main__‘:
    fire.Fire(intro)

这段代码支持三种调用方法，如下图所示：

使用参数 python3 test_fire.py--help可以看到自动生成的帮助信息：

如果漏掉了某些信息，那么就会报错：

那么如果有些参数可要可不要怎么办呢？也非常简单，给函数参数带上默认值即可，例如：

import fire

def intro(name=‘产品经理‘, salary=88888):
    print(f‘我的名字是：{name}, 我的月薪是{salary}‘)

if __name__ == ‘__main__‘:
    fire.Fire(intro)

运行效果如下图所示：

使用 fire，让你的命令行参数更容易。