Question

我正在做一个简单的Django rest框架项目。这只是创建一个新用户并登录。当我使用django基本身份验证用户模型时，一切工作正常。但是在将基本用户模型更改为自定义用户后， dict对象没有属性“ pk” 创建新用户时出现此错误。自定义用户模型参考django文档。（https://docs.djangoproject.com/ko/2.1/topics/auth/customizing/）错误提示

文件 “ /home/seokchan/server/mdocker/lib/python3.5/site-packages/django/contrib/auth/init.py”，第100行，登录如果_get_user_session_key（request）！= user.pk或（（AttributeError：'dict'对象没有属性'pk'

这似乎表明用户模型没有pk，但我不知道。

models.py
class MyUserManager(BaseUserManager):
        def create_user(self, username, email, password=None):

        if not email:
            raise ValueError('Users must have an email address')
        if not username:
            raise ValueError('Users must have an user name')

        user = self.model(
            email=self.normalize_email(email),
            username = username
        )

        user.set_password(password)
        user.save(using=self._db)
        return user

    def create_superuser(self, username, email, password):

        user = self.create_user(
            username,
            password=password,
            email = email,
        )
        user.is_admin = True
        user.save(using=self._db)
        return user


class MyUser(AbstractBaseUser):
    id          = models.AutoField(primary_key=True)

    username = models.CharField(
        verbose_name='user name',
        max_length=30,
        unique=True,
    )

    email = models.EmailField(
        verbose_name='email address',
        max_length=255,
        unique=True,
    )

    is_active = models.BooleanField(default=True)
    is_admin = models.BooleanField(default=False)

    sth_test = models.TextField(blank = True)

    objects = MyUserManager()

    USERNAME_FIELD = 'username'
    REQUIRED_FIELDS = ['email']

    def __str__(self):
        return self.username

    def has_perm(self, perm, obj=None):
        "Does the user have a specific permission?"
        # Simplest possible answer: Yes, always
        return True

    def has_module_perms(self, app_label):
        "Does the user have permissions to view the app `app_label`?"
        # Simplest possible answer: Yes, always
        return True

    @property
    def is_staff(self):
        "Is the user a member of staff?"
        # Simplest possible answer: All admins are staff
        return self.is_admin

serializers.py

class CreateUserSerializer(serializers.ModelSerializer):
    class Meta:
        model = get_user_model
        fields = ('id', 'username', 'email', 'password', 'is_active')

    email = serializers.EmailField(
                required=True,
                validators=[UniqueValidator(queryset=User.objects.all())]
            )
    username = serializers.CharField(
                max_length=32,
                validators=[UniqueValidator(queryset=User.objects.all())]
            )
    password = serializers.CharField(min_length=8, write_only=True)

    def validate_email(self,value):
        if User.objects.filter(email=value).exists():
            raise serializers.ValidationError("err.")
        return value

    def create(self, validated_data):
        user = User.objects.create_user(
            validated_data['username'],
            validated_data['email'],
            validated_data['password'],
            )
        user.is_active = False
        user.save()


message=render_to_string('accounts/account_activate_email.html',{
            'user':user,
            'domain':'localhost:8000/api/accounts/activate', 
            'uid':urlsafe_base64_encode(force_bytes(user.pk)).decode('utf-8'),  
            'token':account_activation_token.make_token(user)  
        })

        mail_subject = 'Bplus'
        to_email = user.email 
        AuthEmail = EmailMessage(mail_subject, message, to=[to_email]) 
        AuthEmail.send()

        return validated_data

views.py

class UserCreateAPI(generics.GenericAPIView):
    serializer_class = CreateUserSerializer

    def post(self, request, *args, **kwargs):
        if len(request.data["username"]) < 4 or len(request.data["password"]) < 8:
            body = {"message":"short field"}
            return Response(body, status = 400) 

        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        user = serializer.save()
        user_for_auth = User.objects.get(username=user['username'])  
        login(request, user)

        return Response(
            {
                "user":UserSerializer(
                    user, context=self.get_serializer_context()
                ).data,
                "token":AuthToken.objects.create(user_for_auth),
            }  
        )

如何解决此错误？谢谢。

enter image description here

Answer 1

您的序列化器create方法返回经过验证的数据，而不是创建的对象。由于这是一个命令，因此最终您将传递给登录功能。

您应该使用return user而不是return validated_data。

django AttributeError：dict对象没有属性“ pk”

1 个答案: