修改檔案的md5碼

孙犯困發表於2024-07-28

可選資料夾 未按照檔案型別匹配

import os
import random
import sys
import hashlib
from tkinter import filedialog


def md5_change(file_name):
    file = open(file_name, 'a')
    file.write('###&&&')
    file.close()


_FILE_SLIM = 100*1024*1024


def md5_search(filepath):
    calltimes = 0
    hmd5 = hashlib.md5()
    fp = open(filepath, "rb")
    f_size = os.stat(filepath).st_size
    if f_size > _FILE_SLIM:
        while (f_size > _FILE_SLIM):
            hmd5.update(fp.read(_FILE_SLIM))
            f_size /= _FILE_SLIM
            calltimes += 1
        if (f_size > 0) and (f_size <= _FILE_SLIM):
            hmd5.update(fp.read())
    else:
        hmd5.update(fp.read())
    return hmd5.hexdigest()


# 獲取資料夾下所有的檔名稱，以列表的形式呈現
#file_name_list = os.listdir(path)
#print(file_name_list)

Folderpath = filedialog.askdirectory() #獲得選擇好的資料夾
file_name_list = os.listdir(Folderpath)

# 在列表元素個數範圍內取隨機數
random_x = random.randint(0, len(file_name_list))
# 獲取到資料夾內某個隨機檔案的絕對路徑（為結尾驗證做準備）
file_name_path = file_name_list[random_x]

# 獲取該隨機選取到的檔案的md5值（md5值修改之前）
md5 = md5_search(file_name_path)
print('"' + file_name_list[random_x] + '"原md5:  ' + md5)


# 開始遍歷含有檔名的列表
for x in range(0, len(file_name_list)):
    filename = file_name_list[x]
    # 剔除目標檔案之外的元素
    if filename == '!!!_md5_change.py':
        pass
    elif filename == '.DS_Store':
        pass
    else:
        # 對檔案進行md5修改
        md5_change(filename)

# 獲取該隨機選取到的檔案的md5值（md5值修改之後）
new_md5 = md5_search(file_name_path)
print('"' + file_name_list[random_x] + '"修改後的md5:  ' + new_md5)

# 對隨機挑選到的檔案進行修改前後比對驗證
if md5 != new_md5:
    print('md5值的修改驗證透過！')
else:
    print('"' + file_name_list[random_x] + '"' + '驗證未透過，該資料夾下所有檔案的md5值可能均修改失敗')

相關文章