argparse学习笔记

cnblogs 2024-08-02 17:39:00 阅读 88

argparse是 Python 的一个内置模块,用于编写用户友好的命令行接口。使用 argparse,你可以很容易地为 Python 脚本添加参数解析功能,使得脚本可以接受命令行选项和参数。学起来也比较简单,接下来我会详细讲解。

在argparse里面,有位置参数和可选参数。位置参数是必不可少的,在命令行里面不可以缺少这个选项,并且这个选项前不需要加-或者--一类的符号;可选参数则可有可无,使用的时候需要在选项之前加上-或者--。

<code>import argparse

parser = argparse.ArgumentParser()

parser.add_argument("square", type=int,

help="display a square of a given number")code>

parser.add_argument("-v", "--verbosity", type=int,

help="increase output verbosity")code>

args = parser.parse_args()

answer = args.square**2

if args.verbosity == 2:

print(f"the square of {args.square} equals {answer}")

elif args.verbosity == 1:

print(f"{args.square}^2 == {answer}")

else:

print(answer)

在这个例子中,首先需要创建一个ArgumentParser的对象,然后可以通过add_argument添加需要的参数。"square"前面没有带-或者--,这是一个位置参数,而且-v或者--verbose则是可选参数。此外,还可以在add_argument当中设定type参数来指定参数的类型。在设定完参数之后,调用parse_args来完成参数的解析,之后就可以愉快地去解析具体参数的值了。

python demo.py 20 -v 1/python demo.py 20 --verbose 1/python demo.py -v 1 20/python demo.py --verbose 1 20

>>20^2 == 400

// 注意,位置参数的位置顺序是可以调换的。

python demo.py 20 -v 2/python demo.py 20 --verbose 2/python demo.py -v 2 20/python demo.py --verbose 2 20

>>the square of 20 equals 400

也可以这样玩。

parser.add_argument("-v", "--verbosity", action='count',code>

help="increase output verbosity")code>

python demo.py 20 -v/python demo.py 20 -vv

不少Linux的命令行都是可以通过这种方式来输出更加详细的信息,这样就不用手动指定参数了。不过这段代码依然有缺陷,来看看:

import argparse

parser = argparse.ArgumentParser()

parser.add_argument("square", type=int,

help="display a square of a given number")code>

parser.add_argument("-v", "--verbosity", action='count',code>

help="increase output verbosity")code>

args = parser.parse_args()

answer = args.square**2

if args.verbosity >= 2:

print(f"the square of {args.square} equals {answer}")

elif args.verbosity >= 1:

print(f"{args.square}^2 == {answer}")

else:

print(answer)

当我们指定参数-vvv或者-vvvv或者更多的v的时候,反而只能输出一个最简单的计算结果,所以代码的分支条件里面"=="改为">=",这样就大功告成啦。为了更加完美,还可以引入静默模式。

import argparse

parser = argparse.ArgumentParser()

parser.add_argument("square", type=int,

help="display a square of a given number")code>

parser.add_argument("-v", "--verbosity", action='store_true',code>

help="increase output verbosity")code>

parser.add_argument('-q', '--quiet', action='store_true')code>

args = parser.parse_args()

answer = args.square**2

if args.quiet:

print(answer)

elif args.verbosity:

print(f"the square of {args.square} equals {answer}")

else:

print(f"{args.square}^2 == {answer}")

这样就有模有样了,命令行的基本功能也齐全了。还可以组合参数:

import argparse

parser = argparse.ArgumentParser()

parser.add_argument("-v", "--verbosity", action='store_true',code>

help="increase output verbosity")code>

parser.add_argument('-q', '--quiet', action='store_true')code>

args = parser.parse_args()

if args.verbosity:

print('v is enabled')

if args.quiet:

print('q is enabled')

python demo.py -v/python demo.py -q/python demo.py -vq/python demo.py -qv

看到这里的朋友可以点点赞或者分享给身边的朋友,以后会继续带来更多有关C++或者Python语言的有趣知识。



声明

本文内容仅代表作者观点,或转载于其他网站,本站不以此文作为商业用途
如有涉及侵权,请联系本站进行删除
转载本站原创文章,请注明来源及作者。