import sys, os, filecmp
import time
from win32com.shell import shell,shellcon



def deltorecyclebin(filename):
    #print('deltorecyclebin', filename)
    # os.remove(filename) #直接刪除文件，不經(jīng)過回收站
    res= shell.SHFileOperation((0,shellcon.FO_DELETE,filename,None, shellcon.FOF_SILENT | shellcon.FOF_ALLOWUNDO | shellcon.FOF_NOCONFIRMATION,None,None))  #刪除文件到回收站
    #print(res)
        # if not res[1]:
        #     os.system('del '+filename)

#清理重復(fù)文件
def md5(fname, chunk_size=1024):
    import hashlib
    hash = hashlib.md5()
    with open(fname, "rb") as f:
        chunk = f.read(chunk_size)
        while chunk:
            hash.update(chunk)
            chunk = f.read(chunk_size)
    return hash.hexdigest()

def check_for_duplicates(dufilelist):    
    sizes = {}
    hashes = {}
    duplicates = []
    # Traverse all the files and store their size in a reverse index map
    # 遍歷所有文件并將其大小存儲(chǔ)在反向索引映射中
    files_count = 0
    #print(dufilelist)
    for s in dufilelist:
        for root, _dirs, files in os.walk(s, topdown=True, onerror=None, followlinks=False):
            files_count += len(files)
            for f in files:
                file = os.path.join(root, f)
                size = os.stat(file).st_size
                if size not in sizes:
                    sizes[size] = []
                sizes[size].append(file)
    #print("Traversed {} files".format(files_count))

    # Remove empty files from the size map
    # 從大小映射中刪除空文件
    if 0 in sizes:
        del sizes[0]

    # Remove files with unique sizes from the size map
    # 從大小映射中刪除具有唯一大小的文件
    for (key, value) in list(sizes.items()):
        if len(value) == 1:
            del sizes[key]

    # Traverse the size map and enrich it with hashes
    # 遍歷大小映射并用哈希值豐富它
    for (size, files) in sizes.items():
        for file in files:            
            try:
                hash = md5(file)
            except:
                continue 
            tuple = (size, hash)
            if tuple not in hashes:
                hashes[tuple] = []
            hashes[tuple].append(file)

    # Remove files with unique (size, hash) tuple in hash map
    # 刪除哈希映射中具有唯一（大小，哈希）元組的文件
    for (key, value) in list(hashes.items()):
        if len(value) == 1:
            del hashes[key]

    # Compare file pairs
    # 比較文件對(duì)    
    for possible_list in hashes.values():
        #print(possible_list)
        #這里把文件按著時(shí)間排序:
        if possible_list:
            # 注意，這里使用lambda表達(dá)式，將文件按照最后修改時(shí)間順序升序排列
            # os.path.getmtime() 函數(shù)是獲取文件最后修改時(shí)間
            # os.path.getctime() 函數(shù)是獲取文件最后創(chuàng)建時(shí)間
            possible_list = sorted(possible_list,key=lambda x: os.path.getctime(x))
        

        while possible_list:
            first = possible_list[0]
            copy = [first]
            for other in possible_list[1:]:
                if filecmp.cmp(first, other, shallow=False):
                    copy.append(other)
            for c in copy:
                possible_list.remove(c)
            duplicates.append(copy)

    #print(duplicates)
    # Print duplicates
    # 打印相同  
    groupready=[]
    for _i, group in enumerate(duplicates):
        print("第 " + str(int(_i) + 1) + " 組")
        assert len(group) > 1
        #print("%r:" % (i + 1))
        if onlyprint: 
            for d in group:
                pass
                print("發(fā)現(xiàn)相同文件:  %r" % (d))
            for d in group[1:]:
                groupready.append(d)

            #print("全部要?jiǎng)h除的文件: "+str(groupready))

        else:
            if filedelete:
                for d in group[1:]:
                    os.remove(d)
                    print("直接刪除重復(fù)文件%r" % (d))
            else:
                for d in group[1:]:
                    deltorecyclebin(d)
                    print("回收重復(fù)文件%r" % (d))
                


    if not duplicates:
        print("目錄里沒有找到重復(fù)文件")


    if len(groupready)>0:
        print("--------------------------------分割線------------------------------------------")
        print("下面列出重復(fù)的文件:")
        for num in groupready:
            print(num)

        reback=input("輸入d直接刪除以上的重復(fù)文件, ,輸入r將以上重復(fù)文件放入回收站,,取消請(qǐng)任意輸入n或者關(guān)閉本窗口.請(qǐng)輸入: d/r/n:")
        if reback=="d":
            for num in groupready:
                os.remove(num)
                print("直接刪除重復(fù)文件%r" % (num)) 
        elif reback=="r":
            for num in groupready:
                deltorecyclebin(num)
                print("回收重復(fù)文件%r" % (num)) 
        else: 
            print("取消操作")







if __name__ == "__main__":        
    loopinfo=input("是否循環(huán)檢測重復(fù)文件? y/n:")
    if loopinfo=="y":
        loopinfoable=-1
    else:
        loopinfoable=1

    onlyprintAble=input("只檢測重復(fù)文件請(qǐng)輸入y,檢測并且自動(dòng)刪除重復(fù)文件請(qǐng)輸入n y/n:")
    if onlyprintAble=="y":
        onlyprint=True   
    else:
        onlyprint=False   
        filedeleteAble=input("重復(fù)文件放入回收站請(qǐng)輸入y ,直接刪除重復(fù)文件請(qǐng)輸入n y/n:")
        if filedeleteAble=="y":
            filedelete=False   #是否強(qiáng)制刪除
        else:
            filedelete=True   #是否強(qiáng)制刪除



    filepath=input("請(qǐng)輸入要檢測重復(fù)文件的目錄:")    

    time_start=time.time()

    while loopinfoable: 
        loopinfoable=loopinfoable-1   
        
        
        print("程序開始...請(qǐng)等待一會(huì)...我正在努力為主人干活中o(′^｀)o...")
        dufilelist=[filepath]    
        check_for_duplicates(dufilelist)


        
        #下面是計(jì)算時(shí)間
        time_end=time.time()
        seconds=time_end-time_start
        m, s = divmod(seconds, 60)
        h, m = divmod(m, 60)
        print('當(dāng)前程序總共用時(shí):%02d:%02d:%02d' % (h, m, s))        
        if loopinfoable!=0: #不循環(huán)
            time.sleep(3)

    input("程序結(jié)束,按回車退出")