Django Rest框架中创建用户的一种好方法
我定制模型如下Django Rest框架中创建用户的一种好方法,django,api,django-rest-framework,Django,Api,Django Rest Framework,我定制模型如下 from django.db import models from django.contrib.auth.models import ( BaseUserManager, AbstractBaseUser, _user_has_perm ) from django.core import validators from django.utils.translation import ugettext_lazy as _ from django.utils import
from django.db import models
from django.contrib.auth.models import (
BaseUserManager, AbstractBaseUser, _user_has_perm
)
from django.core import validators
from django.utils.translation import ugettext_lazy as _
from django.utils import timezone
class AccountManager(BaseUserManager):
def create_user(self, request_data, **kwargs):
now = timezone.now()
if not request_data['email']:
raise ValueError('Users must have an email address.')
if request_data['profile']:
profile = request_data['profile']
else:
profile = None
user = self.model(
username=request_data['username'],
email=self.normalize_email(request_data['email']),
is_active=True,
last_login=now,
date_joined=now,
profile=profile
)
user.set_password(request_data['password'])
user.save(using=self._db)
return user
def create_superuser(self, username, email, password, **extra_fields):
request_data = {
'username': username,
'email': email,
'password': password
}
user = self.create_user(request_data)
user.is_active = True
user.is_staff = True
user.is_admin = True
# user.is_superuser = True
user.save(using=self._db)
return user
class Account(AbstractBaseUser):
username = models.CharField(_('username'), max_length=30, unique=True)
first_name = models.CharField(_('first name'), max_length=30, blank=True)
last_name = models.CharField(_('last name'), max_length=30, blank=True)
email = models.EmailField(verbose_name='email address', max_length=255, unique=True)
profile = models.CharField(_('profile'), max_length=255, blank=True)
is_active = models.BooleanField(default=True)
is_staff = models.BooleanField(default=False)
is_admin = models.BooleanField(default=False)
date_joined = models.DateTimeField(_('date joined'), default=timezone.now)
delete = models.BooleanField(default=0)
objects = AccountManager()
USERNAME_FIELD = 'email'
REQUIRED_FIELDS = ['username']
def user_has_perm(user, perm, obj):
return _user_has_perm(user, perm, obj)
def has_perm(self, perm, obj=None):
return _user_has_perm(self, perm, obj=obj)
def has_module_perms(self, app_label):
return self.is_admin
def get_short_name(self):
return self.first_name
@property
def is_superuser(self):
return self.is_admin
class Meta:
db_table = 'api_user'
swappable = 'AUTH_USER_MODEL'
与默认用户模式的不同之处在于使用电子邮件进行登录,并添加了profile
字段。同时,
python manage.py createsuperuser
可以毫无问题地执行。另外,
views.py
实现如下。
Serializer.py
实现如下
from django.contrib.auth import update_session_auth_hash
from rest_framework import serializers
from .models import Account, AccountManager
class AccountSerializer(serializers.ModelSerializer):
class Meta:
model = Account
fields = ('id', 'username', 'email', 'profile', 'password')
def create(self, validated_data):
return AccountManager.create_user(**validated_data)
如果AuthRegister中有POST请求,我们希望通过AccountManager实现用户创建。通过AccountSerializer创建用户。但是,当使用上述代码执行POST时,会出现“TypeError:create_user()缺少2个必需的位置参数:”self“和“request_data”错误。
如何改进代码?
另外,在Django Rest框架中,请告诉我您是否有更好的用户创建API实现方法。
注意。请尝试:
返回AccountManager.create\u user(请求\u数据=验证的\u数据)
@Ykh我收到一个错误类型错误:create\u user()缺少1个必需的位置参数:“self”
。。。谢谢。好的,忘记使用AccountManager
,只使用User.objects。创建用户(请求数据=验证数据)
将fine@Ykh太好了return Account.objects.create\u user(request\u data=validated\u data)
工作正常。但是,我不明白为什么它工作正常,所以我会学习。我也不知道。我会在业余时间找出答案,然后告诉你。
from django.contrib.auth import update_session_auth_hash
from rest_framework import serializers
from .models import Account, AccountManager
class AccountSerializer(serializers.ModelSerializer):
class Meta:
model = Account
fields = ('id', 'username', 'email', 'profile', 'password')
def create(self, validated_data):
return AccountManager.create_user(**validated_data)