快捷導(dǎo)航

Python 中使用 argparse 解析命令行參數(shù)

更新時(shí)間：2021年11月15日 15:47:06 作者：Moshe Zadka

這篇文章主要介紹了Python 中使用 argparse 解析命令行參數(shù)，argparse 模塊是一個(gè)強(qiáng)大的命令行參數(shù)解析器，還有很多功能沒能在這里介紹。下面文化在哪個(gè)詳細(xì)介紹該內(nèi)容,需要的朋友可以參考一下

1、Python 中的參數(shù)解析

使用 argparse 解析命令行參數(shù)時(shí)，第一步是配置一個(gè) ArgumentParser 對象。這通常在全局模塊內(nèi)完成，因?yàn)閱螁蝊配置_一個(gè)解析器沒有副作用。

import argparse
 
PARSER = argparse.ArgumentParser()

ArgumentParser 中最重要的方法是 .add_argument() ，它有幾個(gè)變體。默認(rèn)情況下，它會添加一個(gè)參數(shù)，并期望一個(gè)值。

PARSER.add_argument("--value")

查看實(shí)際效果，調(diào)用 .parse_args()：

PARSER.parse_args(["--value", "some-value"])
Namespace(value='some-value')

也可以使用 = 語法：

PARSER.parse_args(["--value=some-value"])
Namespace(value='some-value')

為了縮短在命令行輸入的命令，你還可以為選項(xiàng)指定一個(gè)短“別名”：

PARSER.add_argument("--thing", "-t")

可以傳入短選項(xiàng)：

PARSER.parse_args("-t some-thing".split())
Namespace(value=None, thing='some-thing')

或者長選項(xiàng)：

PARSER.parse_args("--thing some-thing".split())
Namespace(value=None, thing='some-thing')

2、類型

有很多類型的參數(shù)可供你使用。除了默認(rèn)類型，最流行的兩個(gè)是布爾類型和計(jì)數(shù)器。布爾類型有一個(gè)默認(rèn)為 True 的變體和一個(gè)默認(rèn)為 False 的變體。

PARSER.add_argument("--active", action="store_true")
PARSER.add_argument("--no-dry-run", action="store_false", dest="dry_run")
PARSER.add_argument("--verbose", "-v", action="count")

除非顯式傳入 --active，否則 active 就是 False。dry-run 默認(rèn)是 True，除非傳入 --no-dry-run。無值的短選項(xiàng)可以并列。

傳遞所有參數(shù)會導(dǎo)致非默認(rèn)狀態(tài)：

PARSER.parse_args("--active --no-dry-run -vvvv".split())
Namespace(value=None, thing=None, active=True, dry_run=False, verbose=4)

默認(rèn)值則比較單一：

PARSER.parse_args("".split())
Namespace(value=None, thing=None, active=False, dry_run=True, verbose=None)

3、子命令

經(jīng)典的 Unix 命令秉承了“一次只做一件事，并做到極致”，但現(xiàn)代的趨勢把“幾個(gè)密切相關(guān)的操作”放在一起。

git、podman 和 kubectl 充分說明了這種范式的流行。

argparse 庫也可以做到：

MULTI_PARSER = argparse.ArgumentParser()
subparsers = MULTI_PARSER.add_subparsers()
get = subparsers.add_parser("get")
get.add_argument("--name")
get.set_defaults(command="get")
search = subparsers.add_parser("search")
search.add_argument("--query")
search.set_defaults(command="search")

MULTI_PARSER.parse_args("get --name awesome-name".split())

Namespace(name='awesome-name', command='get')

MULTI_PARSER.parse_args("search --query name~awesome".split())

Namespace(query='name~awesome', command='search')`

4、程序架構(gòu)

使用 argparse 的一種方法是使用下面的結(jié)構(gòu)：

## my_package/__main__.py
import argparse
import sys
 
from my_package import toplevel
 
parsed_arguments = toplevel.PARSER.parse_args(sys.argv[1:])
toplevel.main(parsed_arguments)

## my_package/toplevel.py
 
PARSER = argparse.ArgumentParser()
## .add_argument, etc.
 
def main(parsed_args):
 
    ...
 
    # do stuff with parsed_args

在這種情況下，使用 python -m my_package 運(yùn)行?；蛘?，你可以在包安裝時(shí)使用 console_scprits 入口點(diǎn)。

總結(jié)：

argparse 模塊是一個(gè)強(qiáng)大的命令行參數(shù)解析器，還有很多功能沒能在這里介紹。它能實(shí)現(xiàn)你想象的一切。

到此這篇關(guān)于Python 中使用 argparse 解析命令行參數(shù)的文章就介紹到這了,更多相關(guān)Python 中使用 argparse 內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: