import argparse
parser = argparse.ArgumentParser()
def arg_parser():
    parser = argparse.ArgumentParser(
        usage="usage: %(prog)s [參數(shù)名] [參數(shù)值]"
    )
    parser.add_argument(
        "--welcome", help="是否展示歡迎詞", action='store_true'
    )
    parser.add_argument(
        "--login", help="是否登錄", action='store_true', default=True
    )
    parser.add_argument(
        "--login_token", help="登錄參數(shù)", default=None, type=str
    )
    model_name = parser.add_argument_group("Model", "使用什么模型處理數(shù)據(jù)？")
    model_name.add_argument(
        "--knn", help="使用KNN方法",
        type=str, default=None
    )
    model_name.add_argument(
        "--tree", help="使用決策樹方法",
        type=str, default=None
    )
    return parser
def argument_parser() -> argparse.ArgumentParser:
    """
    Construct an return an `argparse.ArgumentParser` instance
    """
    return arg_parser()
if __name__ == '__main__':
    parser = argument_parser()
    options, argv_rest = parser.parse_known_args()
    if options.knn is not None:
        argv_rest = ["-knn", options.knn] + argv_rest
    if options.login_token is not None:
        print('login_token參數(shù)為：', options.login_token)
    options = options
    arguments = argv_rest
    print('options為：', options, '\n', "arguments為：", arguments)

第7行：定義parser，參數(shù)中的usage是介紹使用參數(shù)的格式，用于用戶在運(yùn)行 -h 時(shí)展示提示信息。如下圖所示：

第10、14行：給parser添加屬性，重點(diǎn)介紹 action 這個(gè)參數(shù)

相同點(diǎn)：

  第10行和第14行均是action = ‘store_true’，此參數(shù)用于布爾類型的參數(shù)  

區(qū)別：

  第10行中，當(dāng)命令行中觸發(fā)welcome參數(shù)時(shí)，返回為True，沒有觸發(fā)的時(shí)候返回False。   第14行中，當(dāng)命令行中觸發(fā)login參數(shù)時(shí)，返回為True，沒有觸發(fā)的時(shí)候取默認(rèn)值返回仍為True。也就是說，傳不傳login參數(shù)，返回的都是True，這就是default參數(shù)的作用。

第16行：

新增login_token參數(shù)，傳入登錄參數(shù)token，限制token類型為str，默認(rèn)為None

提醒一句，這個(gè)type不要和action結(jié)合使用，action返回的只有布爾類型，注意參數(shù)名字大小寫是不同的。

第20行:        add_argument_group(title=None, description=None)方法：

官方解釋：

      add_argument_group() 方法返回一個(gè)具有 add_argument() 方法的參數(shù)分組對(duì)象，這與常規(guī)的ArgumentParser一樣。當(dāng)一個(gè)參數(shù)被加入分組時(shí)，解析器會(huì)將它視為一個(gè)正常的參數(shù)，但是會(huì)在不同的幫助消息分組中顯示該參數(shù)。 add_argument_group()方法接受 title 和 description 參數(shù)，它們可被用來定制顯示內(nèi)容

是不是有點(diǎn)迷糊， 其實(shí)就是自定義一個(gè)參數(shù)分組，如下圖所示：

第41行：       parser.parse_known_args() 和 parser.parse_args() 的區(qū)別            

options用來接收命令行中已定義了參數(shù)的信息             

argv_rest 接收在命令中未被定義的信息

作用：有時(shí)一個(gè)腳本可能只解析部分命令行參數(shù)，而將其余的參數(shù)繼續(xù)傳遞給另一個(gè)腳本或程序，它的作用方式很類似 parse_args() 但區(qū)別在于當(dāng)存在額外參數(shù)時(shí)它不會(huì)產(chǎn)生錯(cuò)誤，額外的參數(shù)被argv_rest給接收了。

使用：好比在PyQt中，像定義style這種，就可將參數(shù)賦給argv_rest，然后讓系統(tǒng)「QApplication」去處理此參數(shù)

第42行：

假如當(dāng)命令行中的參數(shù)中有knn這個(gè)參數(shù)，而這個(gè)參數(shù)我想交給系統(tǒng)去處理，所以我就把它添加到argv_rest之中，將這種參數(shù)，統(tǒng)一交給系統(tǒng)處理

第45行：

傳入的參數(shù)中存在login_token，對(duì)login_token進(jìn)行單獨(dú)處理。

到此這篇關(guān)于Python自定義命令行參數(shù)選項(xiàng)和解析器的文章就介紹到這了,更多相關(guān)Python自定義命令行內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！