使用Python編寫文件重復檢查器的完整代碼

更新時間：2024年08月06日 09:54:40 作者：winfredzhang

在日常工作中,我們經(jīng)常需要處理大量文件,但有時候會遇到文件重復的情況,為了有效管理文件并避免重復占用存儲空間,我們可以編寫一個簡單的Python程序來檢查文件夾中是否存在重復文件,本文將介紹如何使用Python和其庫來編寫一個文件重復檢查器

介紹

本文將介紹如何使用Python和其庫來編寫一個文件重復檢查器。我們將利用wxPython模塊來構建圖形用戶界面，使用hashlib庫生成文件的MD5哈希值，并借助sqlite3庫將文件的MD5碼存儲在SQLite數(shù)據(jù)庫中，最后找出存在重復文件的情況。

全部代碼

import wx
import os
import hashlib
import sqlite3

class FileDuplicateChecker(wx.Frame):

    def __init__(self, parent, title):
        super(FileDuplicateChecker, self).__init__(parent, title=title, size=(400, 300))

        self.InitUI()
        self.Centre()
        self.Show()

    def InitUI(self):
        panel = wx.Panel(self)

        vbox = wx.BoxSizer(wx.VERTICAL)

        self.dir_picker = wx.DirPickerCtrl(panel, message="Select a folder", style=wx.DIRP_DIR_MUST_EXIST)
        self.dir_picker.Bind(wx.EVT_DIRPICKER_CHANGED, self.OnDirSelect)
        vbox.Add(self.dir_picker, flag=wx.ALL | wx.EXPAND, border=10)

        self.result_text = wx.TextCtrl(panel, style=wx.TE_MULTILINE | wx.TE_READONLY)
        vbox.Add(self.result_text, proportion=1, flag=wx.EXPAND | wx.ALL, border=10)

        panel.SetSizer(vbox)

    def OnDirSelect(self, event):
        selected_dir = self.dir_picker.GetPath()
        files_md5 = self.get_files_md5(selected_dir)
        duplicates = self.find_duplicates(files_md5)

        result = "Duplicate Files:\n"
        for file_path in duplicates:
            result += f"{file_path}\n"

        self.result_text.SetValue(result)

    def get_files_md5(self, folder):
        files_md5 = {}
        for root, _, files in os.walk(folder):
            for file in files:
                file_path = os.path.join(root, file)
                with open(file_path, "rb") as f:
                    content = f.read()
                    md5_hash = hashlib.md5(content).hexdigest()
                    if md5_hash in files_md5:
                        files_md5[md5_hash].append(file_path)
                    else:
                        files_md5[md5_hash] = [file_path]
        return files_md5

    def find_duplicates(self, files_md5):
        duplicates = []
        for md5_hash, file_paths in files_md5.items():
            if len(file_paths) > 1:
                duplicates.extend(file_paths)
        return duplicates

if __name__ == '__main__':
    app = wx.App()
    FileDuplicateChecker(None, title='File Duplicate Checker')
    app.MainLoop()