Python用摘要算法生成token及檢驗token的示例代碼

# 基礎版，不依賴環境import timeimport base64import hashlibclass Token_hander(): def __init__(self,out_time): self.out_time = out_time self.time = self.timer pass def timer(self): return time.time() def hax(self,str): ''' 摘要算法加密 :param str: 待加密字符串 :return: 加密后的字符串 ''' if not isinstance(str,bytes): # 如果傳入不是bytes類型，則轉為bytes類型 try:str = bytes(str,encoding='utf8') except BaseException as ex:raise ValueError('’%s’不可被轉換為bytes類型'%str) md5 = hashlib.md5() md5.update('天王蓋地虎erafe23'.encode(encoding=’utf-8’)) md5.update(str) md5.update('992ksd上山打老虎da'.encode(encoding=’utf-8’)) return md5.hexdigest() def build_token(self,message): ''' hax_message: 待加密字符串內容 格式： ’當前時間戳：message：過期時間戳’ :param message: 需要生成token的字符串 :param time: 過期時間 :return: token ''' hax_message = '%s:%s:%s'%(str(self.time()),message,str(float(self.time())+float(self.out_time))) hax_res = self.hax(hax_message) token = base64.urlsafe_b64encode(('%s:%s'%(hax_message,hax_res)).encode(encoding=’utf-8’)) return token.decode('utf-8') def check_token(self,token): ''' :param token: 待檢驗的token :return: False or new token ''' try: hax_res = base64.urlsafe_b64decode(token.encode('utf8')).decode('utf-8') message_list = hax_res.split(':') md5 = message_list.pop(-1) message = ’:’.join(message_list) if md5 != self.hax(message):# 加密內容如果與加密后的結果不符即token不合法return False else:if self.time() - float(message_list.pop(-1)) >0: # 超時返回False return Falseelse: # token驗證成功返回新的token return self.build_token(message_list.pop(-1)) except BaseException as ex: # 有異常表明驗證失敗或者傳入參數不合法 return False# 測試if __name__ == ’__main__’: token_hand = Token_hander(5) token = token_hand.build_token(b’dxxx’) print(token_hand.check_token(token)) time.sleep(5) print(token_hand.check_token(token))

# 封裝成Django源碼版# 依賴Django運行環境，不可單獨測試，需運行Django環境，# 需要在settings配置文件中配置 OUT_TIME = 時間 ，以秒為單位import osimport timeimport base64import hashlibimport importlibENVIRONMENT_VARIABLE = 'DJANGO_SETTINGS_MODULE'class Token_hander(): def __init__(self): self.out_time = self.getOutTime() self.time = self.timer pass def timer(self): return time.time() def getOutTime(self): module = importlib.import_module(os.environ.get(ENVIRONMENT_VARIABLE)) return getattr(module, 'OUT_TIME',60) # 在settings配置文件中找 OUT_TIME 變量，如果沒有，默認60秒 def hax(self,str): ''' 摘要算法加密 :param str: 待加密字符串 :return: 加密后的字符串 ''' if not isinstance(str,bytes): # 如果傳入不是bytes類型，則轉為bytes類型 try:str = bytes(str,encoding='utf8') except BaseException as ex:raise ValueError('’%s’不可被轉換為bytes類型'%str) md5 = hashlib.md5() md5.update('天王蓋地虎erafe23'.encode(encoding=’utf-8’)) md5.update(str) md5.update('992ksd上山打老虎da'.encode(encoding=’utf-8’)) return md5.hexdigest() def build_token(self,message): ''' hax_message: 待加密字符串內容 格式： ’當前時間戳：message：過期時間戳’ :param message: 需要生成token的字符串 :param time: 過期時間 :return: token ''' hax_message = '%s:%s:%s'%(str(self.time()),message,str(float(self.time())+float(self.out_time))) hax_res = self.hax(hax_message) token = base64.urlsafe_b64encode(('%s:%s'%(hax_message,hax_res)).encode(encoding=’utf-8’)) return token.decode('utf-8') def check_token(self,token): ''' :param token: 待檢驗的token :return: False or new token ''' try: hax_res = base64.urlsafe_b64decode(token.encode('utf8')).decode('utf-8') message_list = hax_res.split(':') md5 = message_list.pop(-1) message = ’:’.join(message_list) if md5 != self.hax(message):# 加密內容如果與加密后的結果不符即token不合法return False else:if self.time() - float(message_list.pop(-1)) >0: # 超時返回False return Falseelse: # token驗證成功返回新的token return self.build_token(message_list.pop(-1)) except BaseException as ex: # 有異常表明驗證失敗或者傳入參數不合法 return False

# 封裝成Django模塊，也依賴Django運行環境 # 需要在settings配置文件中配置 OUT_TIME = 時間 ， 秒為單位import timeimport base64import hashlibfrom django.conf import settingsclass Token_hander(): def __init__(self): self.out_time = self.getOutTime() self.time = self.timer pass def timer(self): return time.time() def getOutTime(self): try: return settings.__getattr__('OUT_time') # 在導入的settings中找 OUT_TIME 變量 except BaseException: return 60 # 找不到默認60 也可以設置直接拋異常 def hax(self,str): ''' 摘要算法加密 :param str: 待加密字符串 :return: 加密后的字符串 ''' if not isinstance(str,bytes): # 如果傳入不是bytes類型，則轉為bytes類型 try:str = bytes(str,encoding='utf8') except BaseException as ex:raise ValueError('’%s’不可被轉換為bytes類型'%str) md5 = hashlib.md5() md5.update('天王蓋地虎erafe23'.encode(encoding=’utf-8’)) md5.update(str) md5.update('992ksd上山打老虎da'.encode(encoding=’utf-8’)) return md5.hexdigest() def build_token(self,message): ''' hax_message: 待加密字符串內容 格式： ’當前時間戳：message：過期時間戳’ :param message: 需要生成token的字符串 :param time: 過期時間 :return: token ''' hax_message = '%s:%s:%s'%(str(self.time()),message,str(float(self.time())+float(self.out_time))) hax_res = self.hax(hax_message) token = base64.urlsafe_b64encode(('%s:%s'%(hax_message,hax_res)).encode(encoding=’utf-8’)) return token.decode('utf-8') def check_token(self,token): ''' :param token: 待檢驗的token :return: False or new token ''' try: hax_res = base64.urlsafe_b64decode(token.encode('utf8')).decode('utf-8') message_list = hax_res.split(':') md5 = message_list.pop(-1) message = ’:’.join(message_list) if md5 != self.hax(message):# 加密內容如果與加密后的結果不符即token不合法return False else:if self.time() - float(message_list.pop(-1)) >0: # 超時返回False return Falseelse: # token驗證成功返回新的token return self.build_token(message_list.pop(-1)) except BaseException as ex: # 有異常表明驗證失敗或者傳入參數不合法 return False

以上就是Python用摘要算法生成token及檢驗token的示例代碼的詳細內容，更多關于Python用摘要算法生成token的資料請關注好吧啦網其它相關文章！