问题描述
我是 Django 新手。我想创建注册资料。我找到了一些代码,但它对我不起作用。当我想进行迁移时,我总是收到此错误 AttributeError: Manager is not available; 'auth.User' 已替换为 'user.User' 我读到我可以用 User = get_user_model() 修复它,但它看起来对我不起作用。
我的模型.py
import random
from django.contrib.auth import get_user_model
from django.db import models
from django.db.models.signals import post_save
from django.dispatch import receiver
User = get_user_model()
def code_generator(length=5):
numbers = '0123456789'
return ''.join(random.choice(numbers) for _ in range(length))
class RegistrationProfile(models.Model):
code = models.CharField(default=code_generator,max_length=5)
user = models.OnetoOneField(to=User,on_delete=models.CASCADE,related_name='registration_profile',primary_key=True)
@receiver(post_save,sender=User)
def create_registration_profile(sender,instance,**kwargs):
profile,created = RegistrationProfile.objects.get_or_create(user=instance)
if created:
profile.save()
serializer.py
from rest_framework import serializers
from rest_framework.validators import UniqueValidator
from django.contrib.auth.password_validation import validate_password
from .models import User
class RegisterSerializer(serializers.ModelSerializer):
email = serializers.EmailField(
required=True,validators=[UniqueValidator(queryset=User.objects.all())]
)
password = serializers.CharField(write_only=True,required=True,validators=[validate_password])
password2 = serializers.CharField(write_only=True,required=True)
class Meta:
model = User
fields = ('username','password','password2','email','first_name','last_name')
extra_kwargs = {
'first_name': {'required': True},'last_name': {'required': True}
}
def validate(self,attrs):
if attrs['password'] != attrs['password2']:
raise serializers.ValidationError({"password": "Password fields didn't match."})
return attrs
def create(self,validated_data):
user = User.objects.create(
username=validated_data['username'],email=validated_data['email'],first_name=validated_data['first_name'],last_name=validated_data['last_name']
)
user.set_password(validated_data['password'])
user.save()
return user
和views.py
from django.contrib.auth.models import User
from rest_framework import generics
from rest_framework.permissions import AllowAny
from .serializers import RegisterSerializer
class RegisterView(generics.CreateAPIView):
queryset = User.objects.all()
permission_classes = (AllowAny,)
serializer_class = RegisterSerializer
最后,我想将代码发送到电子邮件进行验证。任何想法如何以最有效的方式做到这一点?
解决方法
你可以这样做
from django.contrib.auth.models import User
class RegistrationProfile(models.Model):
code = models.CharField(default=code_generator,max_length=5)
user = models.ForeignKey(to=User,on_delete=models.CASCADE,related_name='registration_profile')
@receiver(post_save,sender=User)
def create_registration_profile(sender,instance,**kwargs):
profile,created = RegistrationProfile.objects.get_or_create(user=instance)
if created:
profile.save()