!!token参考：RESTful API---自定义过期的token2

最近项目中使用django-rest-framework作为后台框架，给客户端返回json结果。

在用户验证方面用到token验证，这是一种安卓/iso/..手机客户端常用的，方便的验证方式。

原理是客户端给我发一段字符串,这段字符串是用户在注册，登入的时候、服务器生成的，并关联到用户。保存到数据库，然后返回给客户端，客户端之后呢，就可以凭借这个字符串来确认“我是我，不是别人”。而不用每次验证都要通过账号密码。 _ _ _

django-rest-framework 有一套默认的token验证机制dfs token验证 具体用法不再细讲了，官方文档写得很清楚。

但是笔者发现一个问题，这个token验证机制存的token，一旦生成就保持不变。这样就引发一些问题，万一某人拿到你的token不就为所欲为了吗，就像别人拿到你的密码一样。

解决方案： 给token设置过期时间，超过存活时间，这段token不再具有验证功能，每次用户重新登入，刷新token(这段新token的有存活时间)。这样，重新登入后，你的token更新了，某些居心不良的人即便拿着之前抢来的token也没用。stackoverflow上已经有了token过期时间的讨论。 参考他们的代码我这样写。

改进

#coding=utf-8 auth.py from django.utils.translation import ugettext_lazy as _ from django.core.cache import cache import datetime from rest_framework.authentication import BaseAuthentication from rest_framework import exceptions from account.models import Token from rest_framework import HTTP_HEADER_ENCODING def get_authorization_header(request): """ Return request's 'Authorization:' header, as a bytestring. Hide some test client ickyness where the header can be unicode. """ auth = request.META.get('HTTP_AUTHORIZATION', b'') if isinstance(auth, type('')): # Work around django test client oddness auth = auth.encode(HTTP_HEADER_ENCODING) return auth class ExpiringTokenAuthentication(BaseAuthentication): model = Token def authenticate(self, request): auth = get_authorization_header(request) if not auth: return None try: token = auth.decode() except UnicodeError: msg = _('Invalid token header. Token string should not contain invalid characters.') raise exceptions.AuthenticationFailed(msg) return self.authenticate_credentials(token) def authenticate_credentials(self, key): try: token = self.model.objects.get(key=key) except self.model.DoesNotExist: raise exceptions.AuthenticationFailed('认证失败') if not token.user.is_active: raise exceptions.AuthenticationFailed('用户被禁止') utc_now = datetime.datetime.utcnow() if token.created < utc_now - datetime.timedelta(hours=24 * 14): raise exceptions.AuthenticationFailed('认证信息过期') def authenticate_header(self, request): return 'Token' 

还要配置settings文件

REST_FRAMEWORK = {
    'DEFAULT_AUTHENTICATION_CLASSES': ( 'yourmodule.auth.ExpiringTokenAuthentication', ), } 

再改进 使用了cache缓存对token和关联的用户进行了缓存，因为token验证经常需要从数据库读取，加入缓存，大幅提高速度。

def authenticate_credentials(self, key): token_cache = 'token_' + key cache_user = cache.get(token_cache) if cache_user: return cache_user # 首先查看token是否在缓存中，若存在，直接返回用户 try: token = self.model.objects.get(key=key) except self.model.DoesNotExist: raise exceptions.AuthenticationFailed('认证失败') if not token.user.is_active: raise exceptions.AuthenticationFailed('用户被禁止') utc_now = datetime.datetime.utcnow() if token.created < utc_now - datetime.timedelta(hours=24 * 14): # 设定存活时间 14天 raise exceptions.AuthenticationFailed('认证信息过期') if token: token_cache = 'token_' + key cache.set(token_cache, token.user, 24 * 7 * 60 * 60) # 添加 token_xxx 到缓存 return (token.user, token) 

我的login函数是这样写的

@api_view(['POST']) def login_views(request): receive = request.data if request.method == 'POST': username = receive['username'] password = receive['password'] user = auth.authenticate(username=username, password=password) if user is not None and user.is_active: # update the token token = Token.objects.get(user=user) token.delete() token = Token.objects.create(user=user) user_info = UserInfo.objects.get(user=user) serializer = UserInfoSerializer(user_info) response = serializer.data response['token'] = token.key return json_response({ "result": 1, "user_info":response, # response contain user_info and token }) else: try: User.objects.get(username=username) cause = u'密码错误' except User.DoesNotExist: cause = u'用户不存在' return json_response({ "result": 0, "cause":cause, })

作者：不_一
链接：https://www.jianshu.com/p/e0a206212df4
來源：简书
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。