我正在一个项目中工作,该项目依赖于Django用户模型和DjangoRestFramework下的TokenAuthentication
我被要求为每个用户获取最后一次登录日期时间,并且我已经意识到当我调用身份验证REST端点时,该字段没有得到更新。
这是一个已知的事实吗?为了在每次调用令牌认证时更新该字段,我是否遗漏了必须执行的操作?
谢谢
好吧,最后我从REST Framework TokenAuthentication继承,在urls文件中指向它
url(r'^api-token-auth/', back_views.TokenAuthenticationView.as_view()),
并且View处理请求并手动调用update_last_login,如下所示:
from django.contrib.auth.models import update_last_login
class TokenAuthenticationView(ObtainAuthToken):
"""Implementation of ObtainAuthToken with last_login update"""
def post(self, request):
result = super(TokenAuthenticationView, self).post(request)
try:
request_user, data = requests.get_parameters(request)
user = requests.get_user_by_username(data['username'])
update_last_login(None, user)
except Exception as exc:
return None
return result
@ F.D.F。答案很棒。这是另一种方法。
我们发送将调用user_logged_in
的update_last_login
信号:
user_logged_in.send(sender=user.__class__, request=request, user=user)
这是一个工作视图(基于使用电子邮件作为USERNAME_FIELD的自定义用户模型):
from rest_framework import parsers, renderers
from rest_framework.authtoken.models import Token
from rest_framework.response import Response
from rest_framework.views import APIView
from django.contrib.auth.signals import user_logged_in
from emailauth.serializers import AuthTokenSerializer, UserSerializer
class ObtainAuthToken(APIView):
throttle_classes = ()
permission_classes = ()
parser_classes = (parsers.FormParser, parsers.MultiPartParser, parsers.JSONParser,)
renderer_classes = (renderers.JSONRenderer,)
serializer_class = AuthTokenSerializer
def post(self, request, *args, **kwargs):
serializer = self.serializer_class(data=request.data)
serializer.is_valid(raise_exception=True)
user = serializer.validated_data['user']
token, created = Token.objects.get_or_create(user=user)
user_logged_in.send(sender=user.__class__, request=request, user=user)
return Response({'token': token.key, 'user': UserSerializer(user).data})
obtain_auth_token = ObtainAuthToken.as_view()
你可以在这里找到完整的源代码:Api View with last_login updated
希望这可以帮助。
更简洁的方法:
from django.contrib.auth.models import update_last_login
from rest_framework.authtoken.models import Token
from rest_framework.authtoken.views import ObtainAuthToken
class LoginToken(ObtainAuthToken):
def post(self, request, *args, **kwargs):
result = super().post(request, *args, **kwargs)
token = Token.objects.get(key=result.data['token'])
update_last_login(None, token.user)
return result
然后设置urls.py:
url(r'^api-token-auth/', views.LoginToken.as_view()),
我对Django==2.0.5
,django-rest-framework-social-oauth2==1.1.0
的回答
from django.contrib.auth import user_logged_in
from oauth2_provider.models import AccessToken
from rest_framework import status
from rest_framework_social_oauth2.views import TokenView
class MyTokenView(TokenView):
def post(self, request, *args, **kwargs):
response = super().post(request, *args, **kwargs)
if response.status_code == status.HTTP_200_OK:
token = AccessToken.objects.get(token=response.data['access_token'])
user = token.user
user_logged_in.send(sender=type(user), request=request, user=user)
return response
URLs.朋友:
from django.urls import path
urlpatterns = [
path('token', MyTokenView.as_view(), name='token'),
]
这是我使用ViewSet的解决方案:
views.朋友:
from rest_framework.authtoken.views import ObtainAuthToken
from rest_framework.authtoken.serializers import AuthTokenSerializer
from rest_framework import viewsets
from django.contrib.auth.models import update_last_login
class LoginViewSet(viewsets.ViewSet):
"""Checks email and password and returns an auth token."""
serializer_class = AuthTokenSerializer
def create(self, request):
"""Use the ObtainAuthToken APIView to validate and create a token."""
##update last_login
try:
user = models.User.objects.get(email = request.data['username'])
update_last_login(None, user)
except:
pass
return ObtainAuthToken().post(request)
现在只需将此视图集添加到urls.py:
router.register('login', views.LoginViewSet, base_name="login")