from django.core.cache import cache
from rest_framework.authentication import BaseAuthentication
from user.models import AXFUser
from utils.error import PramsException
class UserTokenAuthtication(BaseAuthentication):
def authenticate(self, request):
try:
token = request.query_params.get('token') if request.query_params.get('token') else request.data.get('token')
user_id = cache.get(token)
user = AXFUser.objects.filter(pk=user_id).first()
return user, token
except:
raise PramsException({'code':1009, 'msg':'用户没有登录,没有操作权限'})
网友评论