#!/usr/bin/env python3
# -*- coding: UTF-8 -*-
import sys
import os
# 獲取文件類型庫
import imghdr
# MD5庫
import hashlib
# 文件操作庫
import shutil
# 設(shè)定默認(rèn)參數(shù)
sourceDir = './'
targetDir = './'
operation = 'move'
images = []
# 用戶操作說明
def Usage():
    print('USAGE: md5img [source dir] [target dir] [-m | -c]')
    sys.exit(0)
# 根據(jù)輸入的參數(shù)，進(jìn)行不同的處理
if len(sys.argv) == 1 or len(sys.argv) > 4:
    Usage()
if len(sys.argv) >= 2:
    sourceDir = sys.argv[1]
    targetDir = sys.argv[1]
if len(sys.argv) >= 3:
    targetDir = sys.argv[2]
if len(sys.argv) == 4:
    if sys.argv[3] == '-m':
        operation = 'move'
    elif sys.argv[3] == '-c':
        operation = 'copy'
    else:
        print('Operation method is not defined')
        Usage()
# 找到源目錄下所有的圖片
def findImage(sourceDir):
    # 如果源目錄不存在，則報(bào)錯退出
    if not os.path.exists(sourceDir):
        print('Source directory is not defined')
        Usage()
    # 循環(huán)目標(biāo)目錄中的文件
    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)
# 計(jì)算 MD5 值函數(shù)
def calcMD5(filepath):
    with open(filepath,'rb') as f:
        md5obj = hashlib.md5()
        md5obj.update(f.read())
        hash = md5obj.hexdigest()
        return hash
# 處理所有圖片
def md5img(images):
    # 如果目標(biāo)目錄不存在，則報(bào)錯退出
    if not os.path.exists(targetDir):
        print('Target directory is not defined')
        Usage()
    # 循環(huán)需要處理的圖片列表
    for img in images:
        # 根據(jù)圖片的真實(shí)后綴，來確定圖片的后綴，如果是 jpeg 則改成 jpg
        postfix = 'jpg' if imghdr.what(img) == 'jpeg' else imghdr.what(img)
        # os.rename(img, targetDir + '/' + calcMD5(img) + '.' + postfix)
        # 執(zhí)行復(fù)制或者移動操作
        if operation == 'move':
            shutil.move(img, targetDir + '/' + calcMD5(img) + '.' + postfix)
        elif operation == 'copy':
            shutil.copy(img, targetDir + '/' + calcMD5(img) + '.' + postfix)
# 找圖片
findImage(sourceDir)
# 處理圖片
md5img(images)
# 完成
print('Finish')

import time
writefile = int(time.time * 1000)
with open(path,"a") as f:
? ? f.write(str(writefile))