python通过MD5文件校验来查找重复内容的文件

下面是编程之家 jb51.cc 通过网络收集整理的代码片段。

编程之家小编现在分享给大家，也给大家做个参考。

网上批量下载下来一堆文件，其中有重复的，人工一个一个比对太麻烦～～
所以想写个Python脚本，通过MD5值校验来比对重复文件，然后把重复的删除掉

#!/usr/bin/python
#encoding=utf-8
 
#import the modules
import os
import os.path
import sys
import hashlib
 
#define the functions
def findFile(rootPath,fileSeq,delSeq):
    dirs = os.listdir(rootPath)                             #list the directories under the root path
    for dir in dirs:                                        #traversal all the directories
        path = rootPath + os.sep + dir                      #complete the path of current file
        if os.path.isdir(path):
            findFile(path,delSeq)                 #if current file is a directory,recursive the function
        else:
            md5Check(path,delSeq)                 #if not a directory,check the md5
 
def md5Check(path,delSeq):
    f = file(path,'rb')                                    #open the file with 'read-only' and 'binary'
    md5obj = hashlib.md5()
    md5obj.update(f.read())                                 #calculate the md5
    if md5obj.hexdigest() in fileSeq:
        delSeq.append(path)                                 #if md5 of current file is in the fileSeq,put the file path into the delSeq
    else:
        fileSeq.append(md5obj.hexdigest())                  #if not in the fileSeq,put the md5 into the fileSeq
    f.close()                                               #close the file
 
def delList(delSeq):
    print 'These files are waiting to be removed:'
    for delFile in delSeq:
        print delFile                                       #list the file path in the delSeq
 
#the main program
fileSeq = []
delSeq = []
while True:
    if len(sys.argv) == 1:
        rootPath = raw_input('Enter the root path: ')       #one parameter means no parameter,ask the root path
    else:
        rootPath = sys.argv[1]                              #or get the second parameter as the root path
    try:
        findFile(rootPath,delSeq)                 #try if the root path is valid
    except(OSError):
        print 'The root path is invalid. Please enter again. '
        del sys.argv[1:]
        continue                                            #catch the except and delete all invalid parameters
    break
 
if len(delSeq) == 0 :
    print 'No duplicate file was found! '                   #if no files in delSeq,exit
else:
    delList(delSeq)                                         #or list the delSeq
    while True:
        answer = raw_input('Would you want to remove these files? Please answer yes(y) or no(n): ')
        answer.lower
        if answer in ('yes','y'):                          #if "yes"
            for delFile in delSeq:
                try:
                    os.remove(delFile)                      #remove all files in delSeq
                except(OSError):
                    print 'Warning! "%s" is not existed! ' % delFile
                    continue                                #ignore the files witch are not existed
            print 'All duplicate files have been removed! '
            break
        elif answer in ('no','n'):
            print 'Process has exited without any change! '
            break                                           #if "no",do nothing
        else:
            print 'Please enter yes(y) or no(n). '
sys.exit()        

以上是编程之家(jb51.cc)为你收集整理的全部代码内容，希望文章能够帮你解决所遇到的程序开发问题。

如果觉得编程之家网站内容还不错，欢迎将编程之家网站推荐给程序员好友。