def md5(fileMd5): import hashlib md5_value = hashlib.md5() with open(fileMd5,'rb') as f: while True:
#每次讀取2048個字節數據 data = f.read(2048) if not data: brea
#計算md5值 md5_value.update(data) return md5_value.hexdigest()
本例我一個登陸驗證的裝飾器python
#!/usr/bin/env python #-*- coding:utf-8 -*- """ 功能: 登陸驗證模塊 詳細說明: 1.密碼文件爲passwd 2.passwd未建立或丟失,會提示:密碼文件不存在,建議從新註冊!! 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是否存在,存在則載入列表,不存在就從新建立一個空字典 if os.path.exists(fileName): #載入用戶列表,數據結構爲字典,k=userName,v=passwdMd5 with open("passwd", "r+") as loadsFn: userDB = json.loads(loadsFn.read()) else: userDB = {} #讓用戶輸入用戶名 userName = raw_input("姓名:") #標誌位:控制循環跳出 flag = True while flag: #用戶註冊時,需輸入兩次密碼 passwd1 = raw_input("密碼:") passwd2 = raw_input("確認密碼:") #若是兩次密碼不一致,則不執行下一步,再次輸入密碼並進行確認 if not passwd1 == passwd2: continue else: #兩次輸入密碼一致,標誌位置爲False,下次跳出循環 flag = False #調用calc_md5函數將明文密碼轉爲對應的md5值,用於保存 passwdMd5 = calc_md5(passwd1) #將用戶名與密碼對應存入字典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("用戶名:") #用戶名是否存在,存在就繼續輸入密碼,不存在則進行註冊 if name in userDB.keys(): flag = True counter = 0 #循環輸入密碼,密碼正確,flag=False(下次直接跳出循環)並執行函數,密碼錯誤則容許嘗試3次,超過3次驗證失敗,退出驗證 while flag: passwd = raw_input("密碼:") passwdMd5 = calc_md5(passwd) if passwdMd5 == userDB[name]: flag = False func(*args,**kwargs) elif counter > 2: print "您已經嘗試了3次,請過會兒再試!!" return else: counter += 1 else: print "用戶名不存在,請您先進行註冊!" register() return decorater if __name__ == "__main__": @login def hello(): print "Hello world!" hello()