Python神器，两行代码，快速生成命令行

2022-02-12 06:13:15

问题导入：

Fire

Python是一门优雅的语言，简洁的语法，强大的功能。毋庸置疑，是世界上最……

问题导入：

有时候我们在写代码时，会产生这样的一个需求：

我们定义了一个 Python 的方法，用来接收一些参数，但是调用的时候想将这些参数用命令行暴露出来。

例如：我们先定义一个 scrape 方法，第一个参数接收 url，即爬取的网址，第二个参数接收 timeout，即指定超时时间。

import requests

def scrape(url, timeout=10):
    response = requests.get(url, timeout=timeout)
    print(response.text)

调用的时候我们可能会这样来调用：

scrape('https:///www.baidu.com', 10)

假如说我们想修改参数置换 url，那就得修改代码。

所以有时候我们想把这些参数用命令行暴露出来，这时候我们就可能用上了 argparse 等等相关的库，然后挨个声明各个参数是干嘛的，非常繁琐。其代码如下：

parser = argparse.ArgumentParser(description='Scrape Function')
parser.add_argument('url', type=str,
                    help='an integer for the accumulator')
parser.add_argument('timeout',  type=int,
                    help='sum the integers (default: find the max)')

if __name__ == '__main__':
    args = parser.parse_args()
    scrape(args.url, args.timeout)

如此操作之后，我们才能顺利地使用命令行来调用这个脚本：

python3 main.py https://www.baidu.com 10

是不是感觉非常的麻烦？argparse写起来费时又费力，想想就费劲。

下面就要给大家介绍一个库，用它我们只需要两行代码就可以做到如上操作。

Fire

这个库的名字叫做Fire，它可以快速为某个 Python 方法或者类添加命令行的参数支持。

先讲一下安装方法，使用 pip3 安装即可：

pip3 install fire

这样我们就安装好了。

使用途径

下面我们来看看几个例子。

一、方法支持

第一个代码示例如下：

import fire

def hello(name="World"):
  return "Hello %s!" % name

if __name__ == '__main__':
  fire.Fire(hello)

这里我们定义了一个 hello 方法，然后接收一个 name 参数，默认值是 World，接着输出了 Hello 加 name 这个字符串。

接着我们导入了 fire 这个库，调用它的 Fire 方法并传入 hello 这个方法声明，会发生什么事情呢？

我们把这段代码保存为 test.py，接着用 Python3 来运行一下：

python3 test.py

运行结果如下：

Hello World!

看起来并没有什么不同。

但是如果我们运行如下命令，就可以看到一些神奇的事情了：

python3 test.py --help

运行结果如下：

NAME
    test.py

SYNOPSIS
    test.py <flags>

FLAGS
    --name=NAME
        Default: 'World'

可以看到，这里它将 name 这个参数转化成了命令行的一个可选参数，我们可以通过 --name

来替换 name 参数。

我们继续来试一下：

python3 test.py --name 123

这里我们传入了一个 name 参数是 123，这时候我们就发现运行结果就变成了如下内容：

Hello 123!

是不是很方便？我们没有借助 argparse 就轻松完成了命令行参数的支持和替换。

那如果我们将 name 这个参数的默认值取消呢？具体代码改写如下：

import fire

def hello(name):
  return "Hello %s!" % name

if __name__ == '__main__':
  fire.Fire(hello)

然后重新运行：

python3 test.py --help

就可以看到结果变为如下内容：

NAME
    test.py

SYNOPSIS
    test.py NAME

POSITIONAL ARGUMENTS
    NAME

NOTES
    You can also use flags syntax for POSITIONAL ARGUMENTS

这时候我们发现 name 这个参数就变成了必传参数，我们必须在命令行里指定这个参数内容，调用就会变成如下命令：

python3 test.py 123

运行结果还是一样的。

二、类支持

当然 fire 这个库不仅仅支持给方法添加命令行的支持，还支持给一个类添加命令行的支持。

下面我们再看一个例子：

import fire

class Calculator(object):    
    def double(self, number):
        return 2 * number

if __name__ == '__main__':
    fire.Fire(Calculator)

我们把这个代码保存为 test2.py，然后运行：

python3 test2.py

运行结果如下：

NAME
    test2.py

SYNOPSIS
    test2.py COMMAND

COMMANDS
    COMMAND is one of the following:

     double

可以看到，在这里它将 Calculator 这个类中的方法识别出来了，COMMAND 之一就是 double，我们试着调用下：

python3 test2.py double

运行结果如下：

ERROR: The function received no value for the required argument: number
Usage: test2.py double NUMBER

For detailed information on this command, run:
  test2.py double --help

在这里就说了，必须要指定另外一个参数，叫做 NUMBER，同时这个参数还是必填参数，我们试着加一下：

python3 test2.py double 4

运行结果如下：

这时候就可以达到正确结果了。

所以说，综合来看，fire 可以为一个类命令行，每个命令都对应一个方法的名称，同时在后面添加额外的可选或必选参数，加到命令行参数的后面。

三、重新改写

最后，让我们回过头来，给我们一开始定义的 scrape 方法添加命令行的参数支持：

import requests
import fire

def scrape(url, timeout=10):
    response = requests.get(url, timeout=timeout)
    print(response.text)
    
    
if __name__ == '__main__':
    fire.Fire(scrape)

这样就可以了！省去了冗长的 argparse 的代码，是不是非常方便？

调用就是如下形式：

NAME
    main.py

SYNOPSIS
    main.py URL <flags>

POSITIONAL ARGUMENTS
    URL

FLAGS
    --timeout=TIMEOUT
        Default: 10

这里说了，URL 是必传参数，timeout 是可选参数。

最后调用下：

python3 main.py https://www.baidu.com

这样我们就可以轻松将 url 通过命令行传递过去了。

当然 timeout 还是可选值，我们可以通过 --timeout 来指定 timeout 参数：

python3 main.py https://www.baidu.com --timeout 5

这样两个参数就都能顺利赋值了，最后效果就是爬取百度，5 秒超时。

怎么样？是不是很方便呢？若有不足之处还请各位大佬指正！！！

码农公寓

问题导入：

Fire

使用途径

一、方法支持

二、类支持

三、重新改写

相关文章