Python3 初学实践案例（3）argparse 命令行参数库的使用
Python3 初学实践案例（3）argparse 命令行参数库的使用
Python3 初学实践案例（3）argparse 命令行参数库的使用
在前面一篇博文中 http://blog.csdn.net/fungleo/article/details/78754419， 我实现了一个程序——将源目录中的图片用MD5重命名后移动或复制到目标文件夹。
虽然实现了效果，但是其中处理命令行传入的参数，是我自己手写判断的。这样做还是有缺点的：
- 笨~用
python语言就是用它各种各样牛逼的库 - 参数的位置是固定的，不能放在前面
- 如果要实现上一条，我的手写的代码部分的复杂度将要大大提高
- 想要更多功能，就得更复杂，我表示无能为力
所以，我找一下有没有合适的库来做这个，有好些的还是。最终我选择了自带的比较新的 argparse 库。还有第三方的更简洁的库，不过我没有使用。
argparse 使用简单说明
argparse 库功能比较强大，官方中文版资料见：http://python.usyiyi.cn/translate/python_278/library/argparse.html
需要注意的是，教程里面的语法是 python2 的，而我们现在一般学习的都是 python3 吧~
引用 argparse 库
import argparse就这样就可以直接引用了。非常简单。
首先，我们定义一个分析器
parser = argparse.ArgumentParser()官方教程就是这么写的，我粗浅的认为，这是为了让代码变得短一些。
定义分析器之后，我们就可以往里面添加我们需要的参数。
为我们的脚本添加一个说明描述
首先呢，我们的脚本是干啥的，得先描述一下。官方的代码如下：
parser = argparse.ArgumentParser(description='这里写点描述信息')我不太喜欢这种写法。于是，我是这样写的。
parser = argparse.ArgumentParser()
parser.description='这里写点描述信息'效果没差。
设置 -v 为输出程序版本号
好，描述好了，我们的程序应该是会版本升级的。所以呢，版本号很有必要。
parser.add_argument("-v", "--version",action='version', version='%(prog)s 1.0')如上，配置参数 action 的值为 version 就可以了。后面的 %(prog)s 代表你自己的脚本的名称。这个是可变的。并且是可以设置的。
添加一个必填的参数
我们用 parser.add_argument 来往分析器里面添加参数，上面我们已经添加了一个特殊的版权输出的参数。那么正常的添加普通参数，就是下面的方法了。
parser.add_argument('sourceDir', help='Select source directory')如上，添加了一个必填的参数 sourceDir ，后面的 help 是帮助信息，默认的参数格式是字符串，也就是 str 如果你需要参数是其他内容，需要指定 type。例如，你需要的参数是数字，就需要用 type=int 来指定类型。
添加一个不带-前缀的选填的参数
parser.add_argument('targetDir', help='Select target directory', nargs='?')如上，使用 nargs='?' 就可以使这个参数是选填的。它还有其他的值，可以参考官方文档。我这里使用问号表示可以接收0个（也就是不填）或者1个（也就是可以填）值。
添加一个带 - 前缀的选填参数
parser.add_argument("-a", "--add", help="add something")代码如上，只需要加上 - 前缀，那么就表示这个参数是选填了的。并且可以同时指定简写或者双横线的全拼，也可以只指定一个。根据你自己的需要来进行设置。
添加一个互相排斥的二选一的选填参数
上面添加的这些参数是可以满足很多场景，但是一个场景满足不了，比如，我需要 -m | -c 这两个参数只能出现一个，如果同时出现两个就不行。那么上面的方法就不够用了。
我们用下面的代码来进行实现。
group = parser.add_mutually_exclusive_group()
group.add_argument("-m", "--move", help="The way to operate the file is to move", action="store_true")
group.add_argument("-c", "--copy", help="The way to operate the file is to copy", action="store_true")如上，定义一个参数组 group 然后往这个组里添加不同功能的参数即可。
action="store_true" 这个配置项代表这个不需要填写值，直接写参数就可以了。输出的时候为布尔值。
获取所有的参数
args = parser.parse_args()
print(args)好，如上代码，我们就把参数定义为变量 args 了，然后，我们就可以用这个对象来获取我们的各种参数了。
打印帮助信息
我们在程序后面加上 -h 这个参数，就可以得到一个我们的程序的详细的帮助信息。这个是 argparse 自带的牛逼功能。省得我们自己去构建帮助信息，我表示非常喜欢。
但是如果只有这样才能获取帮助信息，那么有点索然无味了。其实， argparse 内置了一些方法，可以很方便的让我们调用帮助信息
# 打印 usage
parser.print_usage()
# 打印完整的 help 信息
parser.print_help()
# 输出 usage
parser.format_usage()
# 输出 help
parser.format_help()打印和输出的区别是，打印就直接打出来了。输出的话，只是输出，你可以拿来做任何事情，例如，你想输出了之后再打印出来就可以如下执行：
print(parser.format_help())好吧，有点脱裤子放屁的意思。不过你明白我想表达什么了。
优化过的“将源目录中的图片用MD5重命名后移动或复制到目标文件夹”源代码
#!/usr/bin/env python3
# -*- coding: UTF-8 -*-
import os
# 获取文件类型库
import imghdr
# MD5库
import hashlib
# 文件操作库
import shutil
# 命令行分析库
import argparse
# 处理命令行参数，使用 argparse 库
parser = argparse.ArgumentParser()
# 定义脚本描述信息
parser.description='Move or copy the images in the source directory with the name of MD5 into the target directory'
# 定义脚本版本
parser.add_argument("-v", "--version",action='version', version='%(prog)s 1.0')
# 添加源目录参数，必填
parser.add_argument('sourceDir', help='Select source directory')
# 添加目标目录参数，选填
parser.add_argument('targetDir', help='Select target directory', nargs='?')
# 定义一个互相排斥的参数，copy or move 不可同时存在
group = parser.add_mutually_exclusive_group()
group.add_argument("-m", "--move", help="The way to operate the file is to move", action="store_true")
group.add_argument("-c", "--copy", help="The way to operate the file is to copy", action="store_true")
# 将参数命名为变量 args
args = parser.parse_args()
# 设定默认参数
sourceDir = args.sourceDir
targetDir = args.targetDir or args.sourceDir
operation = 'copy' if args.copy == True else 'move'
images = []
# 定义出错函数
def errUsage(res):
print('error: ' + res)
parser.print_usage()
exit()
# 找到源目录下所有的图片
def findImage(sourceDir):
# 如果源目录不存在，则报错退出
if not os.path.exists(sourceDir):
errUsage('Source directory is not defined')
# 循环目标目录中的文件
for fil in os.listdir(sourceDir):
# 取得文件的路径
filPath = os.path.join(sourceDir, fil)
# 判断文件是否为目录
if not os.path.isdir(filPath):
# 判断文件是否为图片
if imghdr.what(filPath) != None:
# 将图片插入需要处理的列表
images.append(filPath)
# 计算 MD5 值函数
def calcMD5(filepath):
with open(filepath,'rb') as f:
md5obj = hashlib.md5()
md5obj.update(f.read())
hash = md5obj.hexdigest()
return hash
# 处理所有图片
def md5img(images):
# 如果目标目录不存在，则报错退出
if not os.path.exists(targetDir):
errUsage('Target directory is not defined')
method = {
"move": shutil.move,
"copy": shutil.copy
}
# 循环需要处理的图片列表
for img in images:
# 根据图片的真实后缀，来确定图片的后缀，如果是 jpeg 则改成 jpg
postfix = 'jpg' if imghdr.what(img) == 'jpeg' else imghdr.what(img)
# 执行复制或者移动操作
method[operation](img, targetDir + '/' + calcMD5(img) + '.' + postfix)
# 找图片
findImage(sourceDir)
# 处理图片
md5img(images)
# 完成
print('Finish')代码中已经添加了很多的注释了，应该来说可以非常简单的看懂，这里就不赘述了。
程序执行效果如下图所示：
OK，终于把这个脚本给写得相对来说满意了。