Python编程:使用sys、argparse、click、fire实现命令行参数解析

python实现脚本命令行的库有:

  1. 内置库sys
  2. 内置库argparse
  3. 第三方库click
  4. 第三方库fire

内置库sys

sys.argv 包含命令行参数列表,第一个参数是文件名

sys_demo.py

import sys


def add(a, b):
    return a + b

if __name__ == '__main__':
    ret = add(sys.argv[1], sys.argv[2])
    print(ret)

运行结果

(py3) $ python sys_demo.py 1 2
12

内置库argparse

python自带的参数解析,思路还算清晰,就是代码较多

argparse_demo.py

import argparse


def add(a, b):
    return a + b


if __name__ == '__main__':
    # 初始化解析器
    parser = argparse.ArgumentParser()

    # 定义参数
    parser.add_argument("-a", "--a", help="参数a")
    parser.add_argument("-b", "--b", help="参数b")

    # 解析
    args = parser.parse_args()
    ret = add(args.a, args.b)
    print(ret)

运行结果

(py3) $ python argparse_demo.py -a 1 -b 2
12

click库

click库使用装饰器,不过已经对源代码进行了修改

click_demo.py

import click


@click.command()
@click.option("-a", "--a", help="参数a")
@click.option("-b", "--b", help="参数b")
def add(a, b):
    click.echo(a + b)


if __name__ == '__main__':
    ret = add()
    print(ret)

运行结果

(py3) $ python click_demo.py -a 1 -b 2
12

fire库

说实话,fire库太优秀了,简洁,而且Fire可以

  • 不接收参数
  • 接收函数名
  • 接收字典
  • 接收对象
import fire


def add(a, b):
    return a + b


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

运行结果

(py3) $ python fire_demo.py add 1 2
3

比较发现,唯独fire将我输入的数字型字符串转为了数字,进行相加,输出3,其他三个库均默认为字符串,输出12

参考
一个“牛逼”的 Python 命令行解析库
The Python Fire Guide

你可能感兴趣的:(python)