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

更新時間：2020年12月01日 15:37:37 作者：希希大隊長

這篇文章主要介紹了Python用摘要算法生成token及檢驗token的示例代碼，幫助大家更好的理解和學習python，感興趣的朋友可以了解下

# 基礎版，不依賴環(huán)境

import time
import base64
import hashlib
class 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 False
        else:
          # token驗證成功返回新的token
          return self.build_token(message_list.pop(-1))
    except BaseException as ex:
      # 有異常表明驗證失敗或者傳入?yún)?shù)不合法
      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運行環(huán)境，不可單獨測試，需運行Django環(huán)境，
# 需要在settings配置文件中配置 OUT_TIME = 時間 ，以秒為單位

import os
import time
import base64
import hashlib
import importlib
ENVIRONMENT_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 False
        else:
          # token驗證成功返回新的token
          return self.build_token(message_list.pop(-1))
    except BaseException as ex:
      # 有異常表明驗證失敗或者傳入?yún)?shù)不合法
      return False

# 封裝成Django模塊，也依賴Django運行環(huán)境 
# 需要在settings配置文件中配置 OUT_TIME = 時間 ， 秒為單位


import time
import base64
import hashlib
from django.conf import settings


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):
    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 False
        else:
          # token驗證成功返回新的token
          return self.build_token(message_list.pop(-1))
    except BaseException as ex:
      # 有異常表明驗證失敗或者傳入?yún)?shù)不合法
      return False

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

您可能感興趣的文章:

pytorch 把MNIST數(shù)據(jù)集轉換成圖片和txt的方法
這篇文章主要介紹了pytorch 把MNIST數(shù)據(jù)集轉換成圖片和txt的方法，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧
2018-05-05
Python模擬登陸實現(xiàn)代碼
本篇文章主要介紹了Python模擬登陸實現(xiàn)代碼，這里整理了詳細的代碼,具有一定的參考價值，感興趣的小伙伴們可以參考一下
2017-06-06
Python根據(jù)已知鄰接矩陣繪制無向圖操作示例
這篇文章主要介紹了Python根據(jù)已知鄰接矩陣繪制無向圖操作,涉及Python使用networkx、matplotlib進行數(shù)值運算與圖形繪制相關操作技巧,需要的朋友可以參考下
2018-06-06
使用python實現(xiàn)名片管理系統(tǒng)
這篇文章主要為大家詳細介紹了使用python實現(xiàn)名片管理系統(tǒng)，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下
2020-06-06
關于SSD目標檢測模型的人臉口罩識別
這篇文章主要介紹了關于SSD目標檢測模型的人臉口罩識別問題，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教
2022-11-11
Python 3 實現(xiàn)定義跨模塊的全局變量和使用教程
這篇文章主要介紹了Python 3 實現(xiàn)定義跨模塊的全局變量和使用,本文通過實例代碼相結合的形式給大家介紹的非常詳細，具有一定的參考借鑒價值，需要的朋友可以參考下
2019-07-07
Windows10+anacond+GPU+pytorch安裝詳細過程
這篇文章主要介紹了Windows10+anacond+GPU+pytorch安裝詳細過程,本文通過圖文并茂的形式給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下
2020-03-03
Python中operator模塊的操作符使用示例總結
operator模塊中包含了Python的各種內置操作符,諸如邏輯、比較、計算等,這里我們針對一些常用的操作符來作一個Python中operator模塊的操作符使用示例總結:
2016-06-06
Windows系統(tǒng)下安裝tensorflow的配置步驟
這篇文章主要介紹了Windows系統(tǒng)下安裝tensorflow,本文通過圖文并茂的形式給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下
2022-07-07
Python使用matplotlib繪制Logistic曲線操作示例
這篇文章主要介紹了Python使用matplotlib繪制Logistic曲線操作,結合實例形式詳細分析了Python基于matplotlib庫繪制Logistic曲線相關步驟與實現(xiàn)技巧,需要的朋友可以參考下
2019-11-11