使用AbstractBaseUser和BaseUserManager在Django RestFramework中定制模型。

huangapple
go评论94阅读模式
英文:

customize models in Django RestFramework using AbstractBaseUser, BaseUserManager

问题

我想自定义模型，所以我使用了AbstractBaseUser和BaseUserManager。我使用**extra_field来添加'point'属性。

我访问了http://127.0.0.1:8000/account/signup/并输入了我的数据，但出现了错误。

我的代码有什么问题？我是初学者，请帮帮我。

英文:

i wanna customize models. so i used AbstractBaseUser, BaseUserManager
i use **extra_field to add 'point' attribute.

i enterd http://127.0.0.1:8000/account/signup/
and write my data
but the errors occured

enter image description here

what's wrong with my code?
i'm a beginner, plz help me

  1. # serializers.py
  3. from .models import User
  4. from rest_framework import serializers
  6. class UserSerializer(serializers.ModelSerializer):
  7.     def create(self, validated_data):
  8.         user = User.objects.create_user(
  9.             email = validated_data['email'],
  10.             nickname = validated_data['nickname'],
  11.             name = validated_data['name'],
  12.             password = validated_data['password'],
  13.             point = validated_data['point']
  14.         )
  15.         return user
  16.     class Meta:
  17.         model = User
  18.         fields = ['name', 'nickname', 'email', 'password']
  1. # models.py
  3. from django.db import models
  4. from django.contrib.auth.models import AbstractBaseUser, BaseUserManager
  6. class UserManager(BaseUserManager):
  8.     def create_user(self, email, nickname, name, password=None, **point):
  9.         if not email:
  10.             raise ValueError('must have user email')
  11.         if not nickname:
  12.             raise ValueError('must have user nickname')
  13.         if not name:
  14.             raise ValueError('must have user name')
  15.         user = self.model(
  16.             email = self.normalize_email(email),
  17.             nickname = nickname,
  18.             name = name,
  19.             point = point
  20.         )
  21.         user.set_password(password)
  22.         user.save(using=self._db)
  23.         return user
  25.     def create_superuser(self, email, nickname, name, password=None):
  26.         user = self.create_user(
  27.             email,
  28.             password = password,
  29.             nickname = nickname,
  30.             name = name
  31.         )
  32.         user.is_admin = True
  33.         user.save(using=self._db)
  34.         return user
  36. class User(AbstractBaseUser):
  37.     id = models.AutoField(primary_key=True)
  38.     email = models.EmailField(default='', max_length=100, null=False, blank=False, unique=True)
  39.     nickname = models.CharField(default='', max_length=100, null=False, blank=False, unique=False)
  40.     name = models.CharField(default='', max_length=100, null=False, blank=False)
  41.     point = models.IntegerField(default=500, max_length=10000000)
  43.     is_active = models.BooleanField(default=True)    
  44.     is_admin = models.BooleanField(default=False)
  46.     objects = UserManager()
  48.     USERNAME_FIELD = 'email'
  50.     REQUIRED_FIELDS = ['email', 'name', 'nickname', 'password']
  52.     def __str__(self):
  53.         return self.name
  1. # views.py
  3. from django.shortcuts import render
  4. from .serializers import UserSerializer
  5. from .models import User
  6. from rest_framework import generics
  8. # 회원가입
  9. class UserCreate(generics.CreateAPIView):
  10.     queryset = User.objects.all()
  11.     serializer_class = UserSerializer
  1. # urls.py
  3. from django.urls import path, include
  4. from . import views
  5. from rest_framework import urls
  7. urlpatterns =[
  8.     path('signup/', views.UserCreate.as_view()),
  9.     path('api-auth/', include('rest_framework.urls')),
  10.  ]

i wanna customize models. so i used AbstractBaseUser, BaseUserManager
i use **extra_field to add 'point' attribute.

i enterd http://127.0.0.1:8000/account/signup/
and write my data
but the errors occured

what's wrong with my code?
i'm a beginner, plz help me

答案1

得分: 0

当你使用 ** 解包值时，你应该这样访问这些值：

  1. def create_user(self, email, nickname, name, password=None, **kwargs):
  2.         if not email:
  3.             raise ValueError('必须提供用户邮箱')
  4.         if not nickname:
  5.             raise ValueError('必须提供用户昵称')
  6.         if not name:
  7.             raise ValueError('必须提供用户姓名')
  8.         user = self.model(
  9.             email = self.normalize_email(email),
  10.             nickname = nickname,
  11.             name = name,
  12.             point = kwargs['point']
  13.         )
  14.         user.set_password(password)
  15.         user.save(using=self._db)
  16.         return user

注意我如何获取 point 的值。问题在于你对 point 的定义是 { point: 500 }。所以它是一个字典。你需要从该字典中获取 point 的值。

英文:

When you unpack values using ** then you should access the values like this:

  1. def create_user(self, email, nickname, name, password=None, **kwargs):
  2.         if not email:
  3.             raise ValueError('must have user email')
  4.         if not nickname:
  5.             raise ValueError('must have user nickname')
  6.         if not name:
  7.             raise ValueError('must have user name')
  8.         user = self.model(
  9.             email = self.normalize_email(email),
  10.             nickname = nickname,
  11.             name = name,
  12.             point = kwargs['point']
  13.         )
  14.         user.set_password(password)
  15.         user.save(using=self._db)
  16.         return user

Check how I get the value of point. The thing is that your definition of point is { point: 500 }. So is a dict. You need to get the value of point from that dict.

huangapple
  • 本文由 发表于 2023年2月13日 23:49:31
  • 转载请务必保留本文链接：https://go.coder-hub.com/75438186.html
  • django
  • django-models
  • django-rest-framework
  • python
  • web-applications
匿名

发表评论

匿名网友

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定