python通过MD5文件校验来查找重复内容的文件,pythonmd5,#!/usr/bin/p


#!/usr/bin/python#encoding=utf-8#import the modulesimport osimport os.pathimport sysimport hashlib#define the functionsdef 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, fileSeq, delSeq)                 #if current file is a directory, recursive the function        else:            md5Check(path, fileSeq, delSeq)                 #if not a directory, check the md5def md5Check(path, fileSeq, 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 filedef 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 programfileSeq = []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, fileSeq, 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    breakif len(delSeq) == 0 :    print 'No duplicate file was found! '                   #if no files in delSeq, exitelse:    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()                                                  #exit

评论关闭