Python詳解argparse參數模塊之命令行參數

import argparse
parser = argparse.ArgumentParser(description='最簡單的測試')
parser.add_argument('--test', type=str, default='China')
args = parser.parse_args()
print(args.test)

#p.py
import argparse
#創建ArgumentParser對象，用來解析
parser = argparse.ArgumentParser(description='處理一些整數')
#向對象添加參數
parser.add_argument('integers', metavar='N', type=int, nargs='+',help='累加的整數')
parser.add_argument('--sum', dest='accumulate', action='store_const',const=sum, default=max,help='求和(默認求最大值)')
#解析參數
args = parser.parse_args()
print(args.accumulate(args.integers))

print(parser.parse_args(['--sum','1','33','5','6']))
'''
Namespace(accumulate=<built-in function sum>, integers=[1, 33, 5, 6])
'''

#m.py
import os
from hashlib import md5
import argparse
 
def parse_args():
    parser =argparse.ArgumentParser()
    parser.add_argument('--file_dir', type=str)
    return parser.parse_args()
def generate_md5(file_path):
    m =md5()
    file = open(file_path, 'rb')
    f = file.read()
    m.update(f)
    file.close()
    return m.hexdigest()
 
if __name__ == '__main__':
    args = parse_args()
    file_dir = args.file_dir
    log_path = args.file_dir + '.txt'
    log = open(log_path, 'w+')
    for root, _, files in os.walk(file_dir):
        for file in files:
            file_path = os.path.join(root, file)
            md5_value = generate_md5(file_path)
            print(file + " : " + md5_value)
            log.write(file + " : " + md5_value + '\n')

parser = argparse.ArgumentParser('對文件批量生成MD5值')
parser.add_argument('--file_dir',dest='fdir',type=str,help='指定文件所在目錄')
print(parser.print_help())
'''
usage: 對文件批量生成MD5值 [-h] [--file_dir FDIR]
optional arguments:
  -h, --help       show this help message and exit
  --file_dir FDIR  指定文件所在目錄
None
'''

#d.py
import argparse
import math
 
def perfect_square(s):
    v=int(s)
    vs=math.sqrt(v)
    if vs!=int(vs):
        msg="%r 不是一個完全平方數" %s
        raise argparse.ArgumentTypeError(msg)
    return v
parser=argparse.ArgumentParser(description='自定義類型')
parser.add_argument('hi',type=perfect_square,help='隻能是完全平方數')
args = parser.parse_args()
print(args.hi)

#c.py
parser=argparse.ArgumentParser(description='自定義選項')
parser.add_argument('sel',type=int,choices=range(1,9),help='請輸入[1-9)范圍的整數')
args = parser.parse_args()
print(args.sel)

parser=argparse.ArgumentParser(description='自定義選項')
parser.add_argument('--sel',type=int,choices=range(1,9),help='請輸入[1-9)范圍的整數')
print(parser.parse_args([]))

parser.add_argument('--sel',required=True,type=int,choices=range(1,9),help='請輸入[1-9)范圍的整數')
#執行就會報錯：
usage: ipykernel_launcher.py [-h] --sel {1,2,3,4,5,6,7,8}
ipykernel_launcher.py: error: the following arguments are required: --sel
An exception has occurred, use %tb to see the full traceback.
#這個時候的參數sel就是必選參數瞭
print(parser.parse_args(['--sel','2']))
Namespace(sel=2)

主要是set_defaults方法的用途，在多個解析器的時候顯得很重要。
#s.py
def f(v):
     r=v.x + v.y + v.z
     print('x+y+z=',r)
parser = argparse.ArgumentParser(description='求三個輸入值的和')
subparsers = parser.add_subparsers(help='子命令的用法')
parser_a = subparsers.add_parser('add', help='加法運算')
parser_a.add_argument('--x', type=int)
parser_a.add_argument('--y', type=int)
parser_a.add_argument('--z', type=int)
parser_a.set_defaults(func=f)#parser_a設置默認函數為f
args = parser.parse_args()
args.func(args)#運行函數