You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 
flyapps/fir_ser/api/utils/TokenManager.py

69 lines
2.3 KiB

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# project: 3月
# author: liuyu
# date: 2020/3/8
import logging
import random
import string
import time
import uuid
from django.core.cache import cache
from fir_ser.settings import CACHE_KEY_TEMPLATE
logger = logging.getLogger(__name__)
def make_token(release_id, time_limit=60, key='', force_new=False):
token_key = "_".join([key.lower(), CACHE_KEY_TEMPLATE.get("make_token_key"), release_id])
token = cache.get(token_key)
if token and not force_new:
logger.debug(
f"make_token cache exists get token:{token} release_id:{release_id} force_new:{force_new} token_key:{token_key}")
return token
else:
random_str = uuid.uuid1().__str__().split("-")[0:-1]
user_ran_str = uuid.uuid5(uuid.NAMESPACE_DNS, release_id).__str__().split("-")
user_ran_str.extend(random_str)
token = "".join(user_ran_str)
cache.set(token, {
"atime": time.time() + time_limit,
"data": release_id
}, time_limit)
cache.set(token_key, token, time_limit - 1)
logger.debug(
f"make_token cache not exists get token:{token} release_id:{release_id} force_new:{force_new} token_key:{token_key}")
return token
def verify_token(token, release_id):
try:
values = cache.get(token)
if values and release_id == values.get("data", None):
logger.debug(f"verify_token token:{token} release_id:{release_id} success")
return True
except Exception as e:
logger.error(f"verify_token token:{token} release_id:{release_id} failed Exception:{e}")
return False
logger.error(f"verify_token token:{token} release_id:{release_id} failed")
return False
def generate_token_for_medium(medium):
if medium == 'email':
return generate_alphanumeric_token_of_length(32)
elif medium == 'wechat':
return 'WeChat'
else:
return generate_numeric_token_of_length(6)
def generate_numeric_token_of_length(length, random_str=''):
return "".join([random.choice(string.digits + random_str) for _ in range(length)])
def generate_alphanumeric_token_of_length(length):
return "".join(
[random.choice(string.digits + string.ascii_lowercase + string.ascii_uppercase) for _ in range(length)])