题图:Photo by Mark Rabe on Unsplash
在 Python 中,命令行解析库有非常多的选择方案,系统自带的 模块有 argparse,有 Flask 作者写的 click,但是 argparse 使用起来啰嗦, 要经历解析器初始化、参数定义、解析一套流程。例如:
# -*- coding: utf-8 -*-
# arg_test.py
import argparse
def counter(file_type=None):
return
# 初始化解析器
parser = argparse.ArgumentParser()
# 定义参数
parser.add_argument("-f", "--file", help="统计指定文件类型")
# 解析
args = parser.parse_args()
print(counter(args.file))
运行:
$ python arg_test.py -f python
{'python': 100}
而 click 则是以装饰器的形式作用在目标函数中,要侵入函数本身,本质上还是对 argparse 的一种改进,并没有太多创新,算是一种改良作品
# -*- coding: utf-8 -*-
import argparse
import click
@click.command()
@click.option("-f", "--file", help="统计制定文件类型")
def counter(file=None):
click.echo()
if __name__ == '__main__':
print(counter())
运行:
$ python firetest.py -f python
{u'python': 100}
接下来要介绍的这个命令行解析工具非常牛逼,把命令行工具做到了极致,算是一种颠覆式创新,一行代码能把函数导出到命令行窗口,这个工具叫Python-fire,可能经常写命令行工具的人知道,该项目是 Google 开源的 Python 库(可能是 Google 工程师的 Side Project,因为项目主页申明说它不是 Google 官方项目),名字就像一团?。
安装
pip install fire
用法1: fire.Fire()
直接在程序中调用 fire.Fire(),不需要修改目标函数,fire 会把当前模块下的所有对象导出到命令行中
# firetest.py
import fire
def foo(name):
return 'foo !'.format(name=name)
def bar(name):
return "bar ".format(name=name)
if __name__ == '__main__':
fire.Fire()
运行:
# 调用方式:python [文件名] [函数名] [参数]
# 函数名后面直接跟参数值
$ python firetest.py foo hello
foo hello!
# 也可以显示地先指定参数名,再跟参数值
$ python firetest.py bar --name hello
bar hello!
用法2: fire.Fire()
Fire 可以指定某个函数导出到命令行
import fire
def foo(name):
return 'foo !'.format(name=name)
if __name__ == '__main__':
fire.Fire(foo)
运行:
# 调用方式:python [函数名] [参数]
$ python firetest.py hello
foo hello!
当 Fire 接收函数 foo 作为参数时,只加载 foo 函数到命令行中,此时,在命令行中运行时也无需再指定函数名字,只需要指定参数就好。
用法3:fire.Fire()
Fire 不仅可以接收函数,还可以接收字典对象作为参数,可在字典中配置那些函数需要导出到命令行中。例如,有加减乘3个函数,我们可以选择性的选择其中2个导出到命令行:
import fire
def add(x, y):
return x + y
def multiply(x, y):
return x * y
def subtract(x, y):
return x - y
if __name__ == '__main__':
fire.Fire({
'add': add,
'subtract': subtract,
})
运行:
$ python firetest.py add 1 4
5
$ python firetest.py subtract 1 4
-3
$ python firetest.py multiply 1 4
# multiply 会报错,因为没有导出
用法4:fire.Fire()
Fire 还可以接收类的实例对象
import fire
class Calculator(object):
def add(self, x, y):
return x + y
def multiply(self, x, y):
return x * y
if __name__ == '__main__':
calculator = Calculator()
fire.Fire(calculator)
使用方式还是和前面的一样
$ python firetest.py add 10 20
30
$ python firetest.py multiply 10 20
200
更多用法可以参考python-fire的文档:https://github.com/google/python-fire/blob/master/docs/guide.md
关注Python之禅,学点Python
2017年12月1日-2日,由51CTO主办的WOTD全球软件开发技术峰会将在深圳举办,有来自Airbnb、Google、BAT、滴滴、等一线互联网公司的技术大咖带来40个成功经验分享案例。我从官方渠道拿了10张专业门票(享受两日所有会场参会权、不含午餐)作为粉丝福利,有兴趣的同学赶紧点击【阅读原文】,填写申请表单,邀请人列填写【Python之禅】,确保审核通过。数量有限,先到先得!未尽事宜,联系主办方Rachel同学,电话/微信:18401576051。