python之讀取文件的MD5碼並重命名文件

由於自己的手機經常備份,備份后原來的圖片視頻沒有刪除,下次再備份的時候移動硬盤上又多了很多重復圖片,於是想着能不能用提取MD5校驗碼的方式識別出重復的文件,然后處理下硬盤里已經重復的內容.

考慮到最近在學python,於是通過上網查資料和內容,借很多大牛的輪子來試了一下,搞了兩個晚上,終於算是階段性完成,能達到目標的方法.主要難點有三個:

1. 獲取文件的MD5碼;
2. 不能更改文件的后綴名(例如原來是jpg的保留jpg,MP4的保留MP4);
3. 遍歷目標文件夾內所有文件;

不多說了,上代碼,各位如有適合請拿去不謝.如有幫助,請記得留言,謝謝.

# !/usr/bin/env python37
# -*- coding: utf-8 -*-
# @File  : renamefile.py
# @Author: Frank
# @Date  : 2018-09-22

import hashlib
import os
import time

# 1.獲取文件MD5值
def get_md5(file_name,path):
    with open(os.path.join(path,file_name),'rb') as f:
        md5obj = hashlib.md5()
        md5obj.update(f.read())
        hash = md5obj.hexdigest()
    #print(hash,type(hash))
    return hash

#2.找出當前的文件的后綴名
def file_type(file):
    filename = file.split('.')[0]
    filetype = file.split('.')[-1]
    #print(filename,filetype)
    return filetype


#3.主函數:查找目標目錄下的所有文件,並使用MD5值及后綴名重命名當前文件
def main(path):
    winerror = []
    for root,dirlist,filelist in os.walk(path):
        for file in filelist:
            newname =  '{0}.{1}'.format(get_md5(file,path), file_type(file))
            print(newname)
            try:
                if os.path.join(path,newname) == os.path.join(path,file):
                    pass
                else:
                    print('Now Renaming:', file, 'To', newname)
                    os.rename(os.path.join(path,file),os.path.join(path,newname))
            except WindowsError:
                nickname = '{0}.{1}'.format(str(len(winerror)),file_type(file))
                print('WindowsError for:',file, 'Renaming to:', nickname)
                winerror.append(file)
                os.rename(os.path.join(path,file),os.path.join(path,nickname))
    print(winerror)
    print(len(winerror))



#4.執行
path = r'D:\iPhoneshowdup\nameduplicate'
if __name__ == '__main__':
    starttime = time.time()
    main(path)
    endtime = time.time()
    usetime = endtime - starttime
    print('總計用時:', usetime, 's')