import os  # 應(yīng)用文件操作
import hashlib  # 文件對比操作
import logging  # 日志函數(shù)操作
import sys  # 系統(tǒng)應(yīng)用操作

logger = logging.getLogger('系統(tǒng)文件去重')
logging.basicConfig(format='%(asctime)s %(levelname)-8s: %(message)s')
logger.setLevel(logging.DEBUG)

diretory = input('請輸入需要整理的文件目錄: \n')  # 去重的文件夾路徑

if os.path.isdir(diretory):
    logger.info('當(dāng)前目錄[' + diretory + ']校驗成功！')
    md5s = []
    for file_path, dir_names, file_names in os.walk(r'' + diretory):
        for file_name in file_names:
            try:
                file_name_path = os.path.join(file_path, file_name)
                logger.info('當(dāng)前比對路徑: '+ file_name_path)
                md5 = hashlib.md5()
                file = open(file_name_path, "rb")
                md5.update(file.read())
                file.close()
                md5_value = md5.hexdigest()
                if md5_value in md5s:
                    os.remove(file_name_path)
                    logger.info('[' + file_name_path + ']出現(xiàn)重復(fù)已經(jīng)移除！')
                else:
                    md5s.append(md5_value)
            except:
                logger.error('[' + file_name_path + ']對比發(fā)生異常，執(zhí)行下一個！')

else:
    logger.error('輸入的文件夾或者目錄不存在！')

#!/usr/bin/env python
#-*- coding:utf-8 -*-
"""
功能：
    登錄驗證模塊
詳細(xì)說明：
    1.密碼文件為passwd
    2.passwd未創(chuàng)建或丟失，會提示：密碼文件不存在，建議重新注冊??！
    3.未注冊用戶登錄會提示：用戶名不存在，請您先進行注冊！
    4.已注冊用戶登錄時，忘記密碼，嘗試3次后密碼還不正確則退出驗證，等一會兒則可以重新登錄
    5.作為裝飾器進行登錄驗證
"""
import json
import hashlib
import os
pwd = os.getcwd()
fileName = os.path.join(pwd,"passwd")

#將明文密碼通過md5進行加密,返回一個加密后的md5的值
def calc_md5(passwd):
    md5 = hashlib.md5("haliluya")
    md5.update(passwd)
    ret = md5.hexdigest()
    return ret


#新用戶注冊模塊
def register():
    #判斷密碼文件passwd是否存在，存在則載入列表，不存在就重新創(chuàng)建一個空字典
    if os.path.exists(fileName):
        #載入用戶列表，數(shù)據(jù)結(jié)構(gòu)為字典，k=userName,v=passwdMd5
        with open("passwd", "r+") as loadsFn:
            userDB = json.loads(loadsFn.read())
    else:
        userDB = {}
    #讓用戶輸入用戶名
    userName = raw_input("姓名：")
    #標(biāo)志位：控制循環(huán)跳出
    flag = True
    while flag:
        #用戶注冊時，需輸入兩次密碼
        passwd1 = raw_input("密碼：")
        passwd2 = raw_input("確認(rèn)密碼：")
        #如果兩次密碼不一致，則不執(zhí)行下一步，再次輸入密碼并進行確認(rèn)
        if not passwd1 == passwd2:
            continue
        else:
            #兩次輸入密碼一致，標(biāo)志位置為False，下次跳出循環(huán)
            flag = False
        #調(diào)用calc_md5函數(shù)將明文密碼轉(zhuǎn)為對應(yīng)的md5值，用于保存
        passwdMd5 = calc_md5(passwd1)
    #將用戶名與密碼對應(yīng)存入字典userDB中
    userDB[userName] = passwdMd5
    #將用戶名和密碼存入文件
    with open(fileName, "w") as dumpFn:
        dumpFn.write(json.dumps(userDB))


#用戶登錄驗證,裝飾器
def login(func):
    def decorater(*args,**kwargs):
        #判斷passwd文件是否存在，存在則載入userDB（用戶：密碼），否則就重新注冊新的passwd文件并返回
        if os.path.exists(fileName):
            with open("passwd", "r+") as loadsFn:
                userDB = json.loads(loadsFn.read())
        else:
            print "密碼文件不存在，建議重新注冊?。?
            register()
            return

        name = raw_input("用戶名：")
        #用戶名是否存在，存在就繼續(xù)輸入密碼，不存在則進行注冊
        if name in userDB.keys():
            flag = True
            counter = 0
            #循環(huán)輸入密碼，密碼正確，flag=False（下次直接跳出循環(huán)）并執(zhí)行函數(shù)，密碼錯誤則允許嘗試3次，超過3次驗證失敗，退出驗證
            while flag:
                passwd = raw_input("密碼：")
                passwdMd5 = calc_md5(passwd)
                if passwdMd5 == userDB[name]:
                    flag = False
                    func(*args,**kwargs)
                elif counter > 2:
                    print "您已經(jīng)嘗試了3次，請過會兒再試??！"
                    return
                else:
                    counter += 1
        else:
            print "用戶名不存在，請您先進行注冊！"
            register()
    return decorater

if __name__ == "__main__":
    @login
    def hello():
        print "Hello world!"
    hello()