[英]Creating multiple types users in django rest framework, got an unexpected keyword argument 'last_login'
[英]last_login field is not updated when authenticating using Tokenauthentication in Django Rest Framework
我正在一个项目中工作,该项目依赖于 DjangoRestFramework 下的 Django 用户 model 和 TokenAuthentication
我被要求获取每个用户的上次登录日期时间,并且我意识到当我调用身份验证 REST 端点时,该字段没有得到更新。
这是众所周知的事实吗? 为了在每次调用令牌身份验证时更新该字段,我是否遗漏了一些我必须做的事情?
谢谢
好吧,最后我继承自 REST Framework TokenAuthentication,在 urls 文件中指向它
url(r'^api-token-auth/', back_views.TokenAuthenticationView.as_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
@FDF 的回答很棒。 这是另一种方法。
我们发送将调用update_last_login
user_logged_in
信号:
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()
您可以在此处找到完整的源代码: 更新了 last_login 的 Api View
希望这可以帮助。
一种更清洁的方法:
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
网址.py:
from django.urls import path
urlpatterns = [
path('token', MyTokenView.as_view(), name='token'),
]
这是我使用 ViewSet 的解决方案:
视图.py:
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")
这是 django 3.0.8 的最新代码。
:)
谢谢 FDF!
from django.contrib.auth.models import update_last_login
from rest_framework.authtoken.views import ObtainAuthToken
from django.contrib.auth import get_user_model
class TokenAuthenticationView(ObtainAuthToken):
"""Implementation of ObtainAuthToken with last_login update"""
def post(self, request):
result = super(TokenAuthenticationView, self).post(request)
currentUserModel = get_user_model()
try:
user = currentUserModel.objects.get(username=request.data['username'])
update_last_login(None, user)
except Exception as exc:
return None
return result
通过信号实现
from django.dispatch import receiver
from django.db.models.signals import post_save
from django.contrib.auth import user_logged_in
from oauth2_provider.models import AccessToken
@receiver(post_save, sender=AccessToken)
def post_save_access_token(instance, created, raw, **kwargs):
if not created or raw:
return
user_logged_in.send(sender=instance.user.__class__, user=instance.user)
对于使用 knox 对用户进行身份验证的任何人,则需要编辑 api.py 文件并导入 user_logged_in,如下所示
from django.contrib.auth.signals import user_logged_in
之后,在同一 api.py 文件中的 LoginAPI 类中,在 _, token = AuthToken.objects.create(user) 之后添加以下行,如下所示
user_logged_in.send(sender=user.__class__, request=request, user=user)
如果您将 Simple JWT 用于 Django,您可以通过以下链接轻松完成。 你只需要添加
SIMPLE_JWT = {
...,
'UPDATE_LAST_LOGIN': True,
...,
}
有时您不想要确切的登录时间,因为前端存储令牌并使用它而无需再次登录。
您可以通过从rest_framework
APIView
来保存用户通过身份验证和授权的每个请求中的当前时间
from django.contrib.auth.models import update_last_login
from rest_framework.views import APIView
class CustomAPIView(APIView):
def check_permissions(self, request):
super().check_permissions(request)
# here user is authorized (otherwise an exception would have been raised)
update_last_login(None, request.user)
class my_endpoint(CustomAPIView):
permission_classes = [IsAuthenticated]
def get(self, request, company_id=None):
...
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.