脚本之家，脚本语言编程技术及教程分享平台！
分类导航

Python|VBS|Ruby|Lua|perl|VBA|Golang|PowerShell|Erlang|autoit|Dos|bat|shell|

服务器之家 - 脚本之家 - Python - 实现 Python 脚本生成命令行

实现 Python 脚本生成命令行

2023-02-14 11:05崔庆才丨静觅 Python

这篇文章主要介绍了实现 Python 脚本生成命令行，文章通过定义一个 scrape 方法展开详细内容介绍，具有一定的参考价值，需要的小伙伴可以参考一下

有时候我们会有这样的一个需求：

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

比如说这里有个爬取方法：

?
1
2
3
4
import requests
def scrape(url, timeout=10):
    response = requests.get(url, timeout=timeout)
    print(response.text)

这里定义了一个 scrape 方法，第一个参数接收 url，即爬取的网址，第二个参数接收 timeout，即指定超时时间。

调用的时候我们可能这么调用：

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

如果我们想改参数换 url，那就得改代码对吧。

所以有时候我们就想把这些参数用命令行暴露出来，这时候我们可能就用上了 argparse 等等的库，挨个声明各个参数是干嘛的，非常繁琐，

代码如下：

?
1
2
3
4
5
6
7
8
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)

这样我们才能顺利地使用命令行来调用这个脚本：

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

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

Fire

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

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

先看看安装方法，使用 pip3 安装即可：

?
1
pip3 install fire

这样我们就安装好了。

使用方法

下面我们来看几个例子。

方法支持

第一个代码示例如下：

?
1
2
3
4
5
import fire
def hello(name="World"):
  return "Hello %s!" % name
if __name__ == '__main__':
  fire.Fire(hello)

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

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

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

?
1
python3 demo1.py

运行结果如下：

Hello World!

看起来并没有什么不同。

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

?
1
python3 demo1.py --help

运行结果如下：

NAME
    demo1.py
SYNOPSIS
    demo1.py <flags>
FLAGS
    --name=NAME
        Default: 'World'

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

我们来试下：

?
1
python3 demo1.py --name 123

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

Hello 123!

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

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

?
1
2
3
4
5
import fire
def hello(name):
  return "Hello %s!" % name
if __name__ == '__main__':
  fire.Fire(hello)

这时候重新运行：

?
1
python3 demo1.py --help

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

NAME
    demo1.py
SYNOPSIS
    demo1.py NAME
POSITIONAL ARGUMENTS
    NAME
NOTES
    You can also use flags syntax for POSITIONAL ARGUMENTS

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

?
1
python3 demo1.py 123

运行结果还是一样的。

类支持

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

下面我们再看一个例子：

?
1
2
3
4
5
6
import fire
class Calculator(object):   
    def double(self, number):
        return 2 * number
if __name__ == '__main__':
    fire.Fire(Calculator)

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

?
1
python3 demo2.py

运行结果如下：

NAME
    demo2.py

SYNOPSIS
    demo2.py COMMAND

COMMANDS
    COMMAND is one of the following:

     double

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

?
1
python3 demo2.py double

运行结果如下：

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

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

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

?
1
python3 demo2.py double 4

运行结果如下：

8

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

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

重新改写

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

?
1
2
3
4
5
6
7
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 的代码，是不是非常方便？

调用就是如下形式：

?
1
2
3
4
5
6
7
8
9
10
NAME
    main.py
SYNOPSIS
    main.py URL <flags>
 
POSITIONAL ARGUMENTS
    URL
FLAGS
    --timeout=TIMEOUT
        Default: 10

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

最后调用下：

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

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

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

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

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

到此这篇关于实现 Python 脚本生成命令行的文章就介绍到这了,更多相关 Python 命令行内容请搜索服务器之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持服务器之家！

原文链接：https://juejin.cn/post/7049709896983379981

  • Python
  • 命令行
  • 脚本

    • 延伸 · 阅读

    精彩推荐
    • PythonPandas如何对Categorical类型字段数据统计实战案例

      Pandas如何对Categorical类型字段数据统计实战案例

      这篇文章主要介绍了Pandas如何对Categorical类型字段数据统计实战案例，文章围绕主题展开详细的内容介绍，具有一定的参考价值，需要的小伙伴可以参考一...

      菜鸟实战6312022-08-23
    • PythonPython使用PyAV提取视频关键帧的实践

      Python使用PyAV提取视频关键帧的实践

      本文主要介绍了Python使用PyAV提取视频关键帧的实践，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下...

      文艺酱Jayce5762022-08-04
    • PythonPython字符串常用方法以及其应用场景详解

      Python字符串常用方法以及其应用场景详解

      本人学习python时的总结,并不需要记住,知道有这个方法,拿起来用就行,下面这篇文章主要给大家介绍了关于Python字符串常用方法以及其应用场景的相关资料...

      Python编程学习圈4172022-08-07
    • PythonPython APScheduler执行使用方法详解

      Python APScheduler执行使用方法详解

      在本篇文章里小编给大家整理的是一篇关于Python APScheduler执行使用方法的相关内容，有兴趣的朋友们可以学习下。...

      十一10012021-08-12
    • Pythonpython多次绘制条形图的方法

      python多次绘制条形图的方法

      这篇文章主要为大家详细介绍了python多次绘制条形图的方法，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下...

      涂兆诚7902022-12-04
    • Pythonpython实现字符串中字符分类及个数统计

      python实现字符串中字符分类及个数统计

      这篇文章主要介绍了python实现字符串中字符分类及个数统计，具有一定的参考价值，感兴趣的小伙伴们可以参考一下...

      LYYCasablanca00018642021-04-05
    • PythonPycharm配置opencv与numpy的实现

      Pycharm配置opencv与numpy的实现

      本文总结了两种方法来导入opencv与numpy包，第一种是直接在Pycharm中导入两个包，第二种是在官网下载相关文件进行配置，感兴趣的小伙伴们可以参考一下...

      代什么码5342021-12-13
    • Pythonpython按行读取文件,去掉每行的换行符\n的实例

      python按行读取文件,去掉每行的换行符\n的实例

      下面小编就为大家分享一篇python按行读取文件,去掉每行的换行符\n的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧...

      rockly897562021-02-02
    最近更新
    编辑推荐
    阅读排行
    热门标签
    2048 　 6006 　 6174 　 SLOTS 　 MELIAE 　 FEEDPARSER 　 暂存 　 持久性管理 　 pickle模块 　 THREADING 　 reduce 　 nignx 　 装饰器 　 函数式编程 　 兔子毒药 　 contextlib 　 监控文件 　 流量监控 　 Web框架 　 PIL库 　 高斯模糊 　 wxPython 　 ​​​​​​​Python 　 自定义模块 　 Python开发 　 元类 　 metaclass 　 魔术方法 　 DNSPod 　 动态解析域名 　
    © 2019-2023 脚本之家 | 服务器之家(www.tuohang.net)旗下站点 版权所有关于我们联系我们版权申明网站地图