时间戳

首页 > 解决方案 > 在 Django 中使用嵌套序列化程序创建 2 个相关模型的实例

python - 在 Django 中使用嵌套序列化程序创建 2 个相关模型的实例

问题描述

我是 Django 的新手，我的代码遇到了这个问题。我有一个自定义用户模型和一个帐户模型，它们与多对多字段相关。

在注册期间，系统会要求用户创建或不创建帐户（通过链接加入其他帐户）。

我在文档中阅读了有关嵌套序列化程序的信息，我认为这应该创建两个模型实例。如何使用嵌套序列化程序在一个视图中创建关系？解决问题的其他方法？

楷模

class Account(models.Model):
    AccountName = models.TextField(max_length=100, blank=False, null=False)

class User(AbstractBaseUser):
    AccountName = models.ManyToManyField(Account)
    CreateAccount = models.BooleanField(blank=False, null=False)
    EmailId = models.EmailField(max_length=128, blank=False, null=False, unique=True)

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

    USERNAME_FIELD = 'EmailId'
    REQUIRED_FIELDS = ['AccountName', 'CreateAccount',]

    # Implemented the other req. functions

    objects = MyAccountManager()

序列化器

class AccountCreationSerializer(ModelSerializer):
    class Meta:
        model = Account
        fields = ['AccountName']

class SignUpSerializer(ModelSerializer):

    AccountName = AccountCreationSerializer()

    class Meta:
        model = User
        fields = ['EmailId', 'AccountName', 'CreateAccount', 'password']
        extra_kwargs = {'password': {'write_only': True, 'required': True}}

    def create(self, validated_data):
        AccountName = validated_data.pop('AccountName')

        if validated_data['CreateAccount']: #Create only when this is True
            Account.objects.create(AccountName=AccountName, **AccountName)

        userAcc = User.objects.create_user(**validated_data)

        return userAcc

看法

class SignUpView(APIView):

    def post(request):
        # to edit
        signup_serializer = SignUpSerializer(data=request.data)
        
        # rest of the view

请求 // 忽略引号

EmailID: xyz@gmail.com
AccountName: TestAcc
CreateAccount: False
Password: ****

错误： Direct assignment to the forward side of a many-to-many set is prohibited. Use AccountName.set() instead.

自定义模型中的 Create_user

    def create_user(self, EmailId, AccountName, CreateAccount, password):
        if not EmailId:
            raise ValueError("Users must have an email")

        user = self.model(
            EmailId=self.normalize_email(EmailId),
            AccountName=AccountName,
            CreateAccount=CreateAccount,
        )

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

我很确定我在多线程领域犯了一些错误，但还没有找到解决方案。任何帮助都会对我有益。蒂亚！

标签： pythondjangodjango-rest-frameworkmany-to-many

解决方案

您不能将值直接保存到many-to-many字段。数据库不允许您这样做。它只允许您添加它们以关联两个表之间的关系（即 User，Account）。用以下代码替换Serializer文件的代码段。

class AccountCreationSerializer(ModelSerializer):
    class Meta:
        model = Account
        fields = ['AccountName']


class SignUpSerializer(ModelSerializer):
    AccountName = serializers.SerializerMethodField()

    class Meta:
        model = User
        fields = ['EmailId', 'AccountName', 'CreateAccount', 'password']
        extra_kwargs = {'password': {'write_only': True, 'required': True}}

    def validate(self, attrs):
        attrs = super(SignUpSerializer, self).validate(attrs=attrs)
        attrs.update({"AccountName": self.initial_data.get("AccountName")})
        return attrs

    def create(self, validated_data):
        AccountName = validated_data.pop('AccountName')
        acc = Account.objects.create(AccountName=AccountName) if "CreateAccount" in validated_data and validated_data['CreateAccount'] else None

        userAcc = User.objects.create_user(**validated_data)
        if acc:
            userAcc.AccountName.add(acc)

        return userAcc

最后，用SignUpView以下方式替换你的类：

class SignUpView(APIView):
    serializer_class = SignUpSerializer

    def post(self, request, *args, **kwargs):
        serializer = self.serializer_class(data=request.data)
        is_valid_serializer = serializer.is_valid(raise_exception=True)
        if is_valid_serializer:
            with transaction.atomic():
                serializer.save()
        # Rest of your code segments to finalize the response

更新

你的方法有问题create_user。您在这里传递了不应传递的many-to-many字段引用 ( AccountName )。正如我之前提到的，您不能直接保存many-to-many字段。您只需要关联它们之间的关系。省略它，它会工作！

遵循此方法的新定义 ( create_user)。

    def create_user(self, EmailId, CreateAccount, password):
        if not EmailId:
            raise ValueError("Users must have an email")

        user = self.model(EmailId=self.normalize_email(EmailId), CreateAccount=CreateAccount)

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

推荐阅读