在Django rest框架中处理一个请求头以获取头中传递的密钥并将其保存在用户表中？

Question

我有一个移动应用程序的 django 后端。

我正在使用 django rest 框架 (django-rest-auth)，但我仍在编写自己的视图和逻辑，因为 API 端点不是基于模型的。

发送给我的请求在标头中有一个密钥，该密钥与注册的各个设备相关联。除此之外，它还有包含设备 ID 和其他详细信息的发布数据。

一旦我使用 request.META 获得该标头，如何将其保存在用户模型中（每次用户登录时都会更新），因此稍后它可以用作获取所有详细信息（如用户和关联设备）的外键。

class CustomLoginView(LoginView):
   def get_response_serializer(self):

    if getattr(settings, 'REST_USE_JWT', False):
        response_serializer = JWTSerializer
    else:
        # print(self.request.data)
        response_serializer = serializers.TokenSerializer
        # response_serializer = {'data':response_serializer}
    return response_serializer
   def get_response(self):

    # raise APIException("There was a problem!")
    serializer_class = self.get_response_serializer()
    # print(serializer_class.data)
    try:
        if getattr(settings, 'REST_USE_JWT', False):
            data = {
                'user': self.user,
                'token': self.token
            }
            serializer = serializer_class(instance=data,
                                        context={'request': self.request})
        else:
            print("hello")
            serializer = serializer_class(instance=self.token,
                                        context={'request': self.request})
            print("BYE")
            print(serializer.data)
            # print(self.request.data)
            try:
                # print(serializer.data)
                if 'email' not in self.request.POST:
                    return Response({'data':'message'})

                return Response({'data':serializer.data,'status':1,'message':'Success'}, status=status.HTTP_200_OK)
            except ValidationError as exc:
                raise ValidationError({
                        'field_val1': exc.detail,
                    })
    except:
            print("in except")
            raise APIException("There was a problem!")


   def post(self, request, *args, **kwargs):
    self.request = request
    self.serializer = self.get_serializer(data=self.request.data,
                                          context={'request': request})
    print(self.request.data)

    if self.serializer.is_valid():
        print(request.META)
        self.login()
        return self.get_response()
    else:
        return Response({"data":[],"message":"Credentials are wrong",'status':0})

我的用户模型

class User(AbstractUser):
"""User model."""

username = None
email = models.EmailField(_('email address'), unique=True)
mobile_token = models.CharField(max_length=20,blank=True)

USERNAME_FIELD = 'email'
REQUIRED_FIELDS = []

objects = UserManager()

我的 UserManage()

类用户管理器（BaseUserManager）： """为没有用户名字段的用户模型定义模型管理器。"""

use_in_migrations = True

def _create_user(self, email, password, **extra_fields):
    """Create and save a User with the given email and password."""
    if not email:
        raise ValueError('The given email must be set')
    email = self.normalize_email(email)
    user = self.model(email=email, **extra_fields)
    user.set_password(password)
    user.save(using=self._db)
    return user

def create_user(self, email, password=None, **extra_fields):
    """Create and save a regular User with the given email and password."""
    extra_fields.setdefault('is_staff', False)
    extra_fields.setdefault('is_superuser', False)
    return self._create_user(email, password, **extra_fields)

def create_superuser(self, email, password, **extra_fields):
    """Create and save a SuperUser with the given email and password."""
    extra_fields.setdefault('is_staff', True)
    extra_fields.setdefault('is_superuser', True)

    if extra_fields.get('is_staff') is not True:
        raise ValueError('Superuser must have is_staff=True.')
    if extra_fields.get('is_superuser') is not True:
        raise ValueError('Superuser must have is_superuser=True.')

    return self._create_user(email, password, **extra_fields)

Answer 1

定义一个instance method 并将其在CustomLoginView中的任何位置用作

from rest_framework.exceptions import ValidationError


class CustomLoginView(LoginView):
    def save_header(self):
        headr = self.request.META.get('HTTP_MYHEADER', None)
        if headr is None:
            raise ValidationError('"MYHEADER" is missing in headers"')
        current_user = self.user
        current_user.mobile_token = headr
        current_user.save()

    # your code

更新
如下更改您的视图

from rest_auth.views import LoginView
from rest_framework.exceptions import ValidationError


class CustomLoginView(LoginView):
    def save_header(self):
        headr = self.request.META.get('HTTP_MYHEADER', None)
        if headr is None:
            raise ValidationError('"MYHEADER" is missing in headers"')
        current_user = self.user
        current_user.mobile_token = headr
        current_user.save()

    def post(self, request, *args, **kwargs):
        response = super().post(request, *args, **kwargs)
        if response.status_code == 200:
            self.save_header()
        return response

你的有效载荷应该是

{
    "email":"myemail@email.com",
    "password":"mypassword"
}

截图
1.如何添加HEADER

2.如何在POSTMAN中发送数据