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
ameduplicate'
if __name__ == '__main__':
    starttime = time.time()
    main(path)
    endtime = time.time()
    usetime = endtime - starttime
    print('总计用时:', usetime, 's')