Python用摘要算法生成token及检验token的示例代码

# 基础版，不依赖环境

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:
      # 有异常表明验证失败或者传入参数不合法
      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 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:
      # 有异常表明验证失败或者传入参数不合法
      return False

# 封装成Django模块，也依赖Django运行环境 
# 需要在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:
      # 有异常表明验证失败或者传入参数不合法
      return False

以上就是Python用摘要算法生成token及检验token的示例代码的详细内容，更多关于Python用摘要算法生成token的资料请关注其它相关文章！