# 基礎(chǔ)版，不依賴環(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類型，則轉(zhuǎn)為bytes類型
      try:
        str = bytes(str,encoding="utf8")
      except BaseException as ex:
        raise ValueError("'%s'不可被轉(zhuǎn)換為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: 待加密字符串內(nèi)容 格式： '當(dāng)前時(shí)間戳：message：過(guò)期時(shí)間戳'
    :param message: 需要生成token的字符串
    :param time: 過(guò)期時(shí)間
    :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: 待檢驗(yàn)的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):
        # 加密內(nèi)容如果與加密后的結(jié)果不符即token不合法
        return False
      else:
        if self.time() - float(message_list.pop(-1)) >0:
          # 超時(shí)返回False
          return False
        else:
          # token驗(yàn)證成功返回新的token
          return self.build_token(message_list.pop(-1))
    except BaseException as ex:
      # 有異常表明驗(yàn)證失敗或者傳入?yún)?shù)不合法
      return False

# 測(cè)試
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運(yùn)行環(huán)境，不可單獨(dú)測(cè)試，需運(yùn)行Django環(huán)境，
# 需要在settings配置文件中配置 OUT_TIME = 時(shí)間 ，以秒為單位

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 變量，如果沒有，默認(rèn)60秒

  def hax(self,str):
    """
    摘要算法加密
    :param str: 待加密字符串
    :return: 加密后的字符串
    """
    if not isinstance(str,bytes): # 如果傳入不是bytes類型，則轉(zhuǎn)為bytes類型
      try:
        str = bytes(str,encoding="utf8")
      except BaseException as ex:
        raise ValueError("'%s'不可被轉(zhuǎn)換為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: 待加密字符串內(nèi)容 格式： '當(dāng)前時(shí)間戳：message：過(guò)期時(shí)間戳'
    :param message: 需要生成token的字符串
    :param time: 過(guò)期時(shí)間
    :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: 待檢驗(yàn)的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):
        # 加密內(nèi)容如果與加密后的結(jié)果不符即token不合法
        return False
      else:
        if self.time() - float(message_list.pop(-1)) >0:
          # 超時(shí)返回False
          return False
        else:
          # token驗(yàn)證成功返回新的token
          return self.build_token(message_list.pop(-1))
    except BaseException as ex:
      # 有異常表明驗(yàn)證失敗或者傳入?yún)?shù)不合法
      return False

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


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")  # 在導(dǎo)入的settings中找 OUT_TIME 變量
    except BaseException:
      return 60  # 找不到默認(rèn)60 也可以設(shè)置直接拋異常

  def hax(self,str):
    """
    摘要算法加密
    :param str: 待加密字符串
    :return: 加密后的字符串
    """
    if not isinstance(str,bytes): # 如果傳入不是bytes類型，則轉(zhuǎn)為bytes類型
      try:
        str = bytes(str,encoding="utf8")
      except BaseException as ex:
        raise ValueError("'%s'不可被轉(zhuǎn)換為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: 待加密字符串內(nèi)容 格式： '當(dāng)前時(shí)間戳：message：過(guò)期時(shí)間戳'
    :param message: 需要生成token的字符串
    :param time: 過(guò)期時(shí)間
    :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: 待檢驗(yàn)的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):
        # 加密內(nèi)容如果與加密后的結(jié)果不符即token不合法
        return False
      else:
        if self.time() - float(message_list.pop(-1)) >0:
          # 超時(shí)返回False
          return False
        else:
          # token驗(yàn)證成功返回新的token
          return self.build_token(message_list.pop(-1))
    except BaseException as ex:
      # 有異常表明驗(yàn)證失敗或者傳入?yún)?shù)不合法
      return False