我們知道在Linux下有優(yōu)雅的shell終端命令行界面，shell腳本都可以優(yōu)雅用命令行的方式來運行。而且shell也再帶優(yōu)化命令行參數(shù)解析的bash內部命令getopts和大多數(shù)發(fā)行版附帶的外部命令getops。Perl語言也有Getopt：：XX系列模塊來實現(xiàn)類似功能;Golang也有flag標準庫以及更加強大的第三庫cobra。

公司主營業(yè)務：網(wǎng)站設計、成都網(wǎng)站制作、移動網(wǎng)站開發(fā)等業(yè)務。幫助企業(yè)客戶真正實現(xiàn)互聯(lián)網(wǎng)宣傳，提高企業(yè)的競爭能力。創(chuàng)新互聯(lián)公司是一支青春激揚、勤奮敬業(yè)、活力青春激揚、勤奮敬業(yè)、活力澎湃、和諧高效的團隊。公司秉承以“開放、自由、嚴謹、自律”為核心的企業(yè)文化，感謝他們對我們的高要求，感謝他們從不同領域給我們帶來的挑戰(zhàn)，讓我們激情的團隊有機會用頭腦與智慧不斷的給客戶帶來驚喜。創(chuàng)新互聯(lián)公司推出旬陽免費做網(wǎng)站回饋大家。

說了這么多你肯定要問，那么Python呢?Python不像shell，perl一樣有便捷強大的oneline腳本，雖然Python有強大的web界面下的腳本執(zhí)行環(huán)境Jupyter Note，但是對終端命令行下的執(zhí)行也是我們要追求的目標。本文蟲蟲就給大家介紹如何在Python下實現(xiàn)優(yōu)雅的終端命令行界面CLI，涉及的模塊有Click和argparse。

Click

Python開始的時候也模仿bansh和其他語言有getopt，optparse等模塊來做CLI，后來推出argparser模塊后getopt，optparse等都已經(jīng)被廢棄。在介紹argparser模塊之前我們先來介紹click。

首先最簡單的方式是使用click。我們先來看下python 3中的print()函數(shù)，注意該print()強制必須以帶括號方式， 這是和python2比較大的區(qū)別。其定義如下：

 
 
 

  
  
  
print(*objects, sep=' ', end='\n', file=sys.stdout, flush=False)
 
 
 

可以看到該函數(shù)可以支持各種規(guī)范格式的輸出，而不單單一個打印輸出了。

比如，通過可以自定義打印的結束字符，默認是"\n"。

 
 
 

  
  
  
print('hello，chongchong! ', end='\r')
 
 
 

可以通過sep自定義分割符，而不是默認的空格。

 
 
 

  
  
  
print('hello', ' chongchong!', sep=',')
 
 
 

通過file來重定向輸出到文件，而不是stdout的默認標準輸出。

 
 
 

  
  
  
print(' hello，chongchong!', file=open('chongchong.out', 'w'))
 
 
 

我們可以利用click在命令行中實現(xiàn)對這些參數(shù)的輸入和解析，實現(xiàn)一個簡單的打印命令行界面。廢話少說直接上代碼：

 
 
 

  
  
  
#!/usr/bin/env python3
  
  
  
import click
  
  
  
import sys
  
  
  
@click.command()
  
  
  
@click.argument('args', nargs=-1)
  
  
  
@click.option('-s', '--sep', default=' ', help='分隔符，默認為空')
  
  
  
@click.option('-e', '--end', default='\n', help='結尾字符，默認為 "\\n"')
  
  
  
@click.option('-o', '--out', default=None, help='輸出文件，默認為stdout)')
  
  
  
def print_prog(args, sep, end, out):
  
  
  
'''打印輸出結果。
  
  
  
'''
  
  
  
print(*args, sep=sep, end=end, file=open(out, 'w') if out else sys.stdout)
  
  
  
if __name__ == '__main__':
  
  
  
print_prog()
 
 
 

腳本很簡單，就是獲取輸入的參數(shù)，并將其傳遞給print()函數(shù)執(zhí)行。執(zhí)行如下：

打印文本

 
 
 

  
  
  
python print.py hello,chongchong!
 
 
 

顯示幫助

 
 
 

  
  
  
python print.py –help
 
 
 

設置風格符號

 
 
 

  
  
  
python print.py -s %% hello chongchong!
 
 
 

重定向到文件

 
 
 

  
  
  
python print.py -o chongchong.out -s %% hello chongchong!
 
 
 

通過幾行代碼就實現(xiàn)了一個很棒的交互界面!

當然你可以依次為基礎增加更多的內容，讓它更實用這是一個演示，向您展示創(chuàng)建自己的CLI是多么容易。你不再需要編寫普通的python腳本文件了。更多功能和高級特性可以參考click官方文檔：

更多的基于click的項目和模塊也可以通過其github倉clone和fork使用，你也可以給該項目添加自己的貢獻。

具體地址為：github：/click-contrib/。

argparse

我們首先從官方的數(shù)字計算程序小例子來開始：

 
 
 

  
  
  
import argparse
  
  
  
parser = argparse.ArgumentParser(description='Process some integers.')
  
  
  
parser.add_argument('integers', metavar='N', type=int, nargs='+',
  
  
  
help='an integer for the accumulator')
  
  
  
parser.add_argument('--sum', dest='accumulate', action='store_const',
  
  
  
const=sum, default=max,
  
  
  
help='sum the integers (default: find the max)')
  
  
  
args = parser.parse_args()
  
  
  
print args.accumulate(args.integers)
 
 
 

代碼解釋：

首先，通過argparse.ArgumentParser()創(chuàng)建一個解析對象;

接著，給對象添加兩個參數(shù)，用add_argument()方法。比如本例中是一個type為一個或者一列整數(shù);還有一個目標為accumulate屬性用來選擇兩個函數(shù)sum()和max()求和和求最大值，默認為求最大值。

然后，通過parse_args()函數(shù)解析對象的參數(shù)，并把各個參數(shù)解析為合適的類型(int，accumulate)，觸發(fā)對應的行為(sum，max)。

最后通過print打印結果。

ArgumentParser.add_argument()

使用方法如下：

ArgumentParser構造對象

ArgumentParser對象的初始化原型為(構造方法)：

 
 
 

  
  
  
ArgumentParser(prog=None, usage=None, description=None, epilog=None, parents=[], formatter_class=argparse.HelpFormatter, prefix_chars='-', fromfile_prefix_chars=None, argument_default=None, conflict_handler='error', add_help=True)
 
 
 

其各個參數(shù)為：

prog - 程序的名稱(默認為sys.argv[0])

usage - 描述程序使用幫助信息(默認值：從添加的參數(shù)來生成)

description - 在參數(shù)幫助之前顯示的文本(默認值：無)

epilog - 在參數(shù)幫助后顯示的文本(默認值：無)

parents - ArgumentParser也應包含其參數(shù)的對象列表

formatter_class - 用于自定義幫助輸出的類

prefix_chars - 前綴可選參數(shù)的字符集(默認值：' - ')

fromfile_prefix_chars -文件前綴字符參數(shù)(默認值：None)

argument_default -為參數(shù)的全局默認值(默認值：None)

conflict_handler - 解決沖突選項的策略(通常不需要)

add_help -添加-h/--help選項解析器(默認值：True)

一般情況下，我們無需管其他情況，只需設置個程序描述即可。

add_argument()方法

add_argument()方法用來創(chuàng)建解析參數(shù)，定義對參數(shù)具體的解析，其原型如下：

add_argument(name or flags...[, action][, nargs][, const][, default][, type][, choices][, required][, help][, metavar][, dest])

參數(shù)有：

name或者flags - 名稱或選項字符串列表，必填寫項。例如foo或-f, --foo。

action - 在命令行中遇到此參數(shù)時要采取的操作類型。

nargs – 要使用的命令行參數(shù)的數(shù)量，用通配符號表示，比如"*"，"+"和"?"，分別表示不限數(shù)量，至少一個和一個參數(shù)。

const - 一些動作和 nargs 選擇所需的常量。

default - 如果參數(shù)在命令行中不存在，則默認使用的值。

type - 命令行參數(shù)應該轉換的類型，默認為字符串string

choices – 可選項， 一個允許參數(shù)值的容器。

required - 是否可以省略命令行選項(僅對可選項)。

help - 幫助提示信息，用來對參數(shù)進行必要的描述。

metavar – 幫助消息替代參數(shù)的顯示名稱。

dest - 要添加到parse_args()函數(shù)返回對象的屬性的名稱。

parse_args()方法

parse_args方法用來把命令行中的字符，解析到參數(shù)解析命名空間(add_argumnet()創(chuàng)建)定義的各個參數(shù)。其原型如下：

 
 
 

  
  
  
parse_args(args=None, namespace=None)
 
 
 

args為參數(shù)列表，參數(shù)字符串通過調用sys.argv獲取，Namespace為屬性創(chuàng)建一個新的空對象。

實例學習sqlmap

最后我們節(jié)選一個著名sql注入滲透工具sqlmap的cmdline.py部分代碼,來展示一個大型軟件中，如何實際通過argparser來構建起命令行界面的?；旧弦哺厦媪鞒谭椒ㄒ粯?，不過額外使用一些函數(shù)和功能。

全部代碼詳見sqlmap github官方倉庫：

(github /sqlmapproject/sqlmap/blob/master/lib/parse/cmdline.py)

總結

本文我們實例介紹了python下構建優(yōu)雅終端命令行界面兩種方法Click和argparser。善用他們可以為我們的代碼工作減少很多繁瑣的命令行參數(shù)的定義和解析過程，提高編碼的效率。

本文名稱：利用Click和argparse給你Python程序構建一個優(yōu)雅的命令行界面
文章來源：http://uogjgqi.cn/article/dhjhohd.html