ICode9

精准搜索请尝试: 精确搜索
首页 > 编程语言> 文章详细

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

2022-02-21 20:03:30  阅读:286  来源: 互联网

标签:__ name Python py timeout 神器 fire 参数 命令行


目录

问题导入:

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

运行结果如下:

8

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

所以说,综合来看,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 秒超时。

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

标签:__,name,Python,py,timeout,神器,fire,参数,命令行
来源: https://blog.csdn.net/CoCo_biancheng/article/details/123035384

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有