如何基於父 model 在另一個 model 中創建 object？

Question

我在嘗試實施（實際上是 map 出來）一個新的 model object 基於父 Z20F35E630DAF49DB88CZ object 時遇到了麻煩。

下面的Bucket model 有一個基於兩個category_options的category字段。

class Bucket(models.Model):

    category_options = (
        ('personal', 'Personal'),
        ('social', 'Social'),
    )


    class BucketObjects(models.Manager):
        def get_queryset(self):
            return super().get_queryset()

    owner = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE, related_name='buckets')
    admin_user = models.ManyToManyField(settings.AUTH_USER_MODEL, related_name='admin_user')
    guest_user = models.ManyToManyField(settings.AUTH_USER_MODEL, related_name='guest_user', blank=True)
    category = models.CharField(max_length=30, choices=category_options)

    ...

    objects = models.Manager()
    bucketobjects = BucketObjects()

    class Meta:
        ordering = ('-created',)

    def save(self, *args, **kwargs):
        if self.stock_list:
            self.stock_count = len(self.stock_list)
            super().save(*args, **kwargs)
        else:
            super().save(*args, **kwargs)

我想在一個完全不同的 model, SocialBucket ，當Bucket model 實例被選擇為基於上述category字段的social類別時：

class SocialBucket(models.Model):
    bucket = models.ForeignKey(Bucket.objects.id, on_delete=models.CASCADE, related_name='social_buckets)

我怎樣才能 go 關於填充我的SocialBucket model 與基於它的父對象， Bucket ， model 的新對象？

編輯：根據要求，這是我將使用的視圖：

view.py

class CreateBucket(generics.CreateAPIView):
    permission_classes = [IsAuthenticated]
    serializer_class = BucketCreateSerializer
    queryset = Bucket.objects.all()

serializer.py

class BucketCreateSerializer(serializers.ModelSerializer):
    class Meta:
        model = Bucket
        fields = ('owner','category','name','about')
        read_only_fields = ['owner']
    
    def create(self, validated_data):
        user = self.context['request'].user
        bucket = Bucket.objects.create(
         owner=user, 
         **validated_data
        )
        bucket.save()
        return bucket

Answer 1

嘗試將models.Model更改為Bucket：

class SocialBucket(Bucket)
    bucket = models.ForeignKey(Bucket.objects.id, on_delete=models.CASCADE, related_name='social_buckets)

Answer 2

我認為最好的方法是使用post_save信號，例如：

from django.db.models.signals import post_save
from django.dispatch import receiver


class Bucket(models.Model):
    ....

class SocialBucket(models.Model):
     bucket = models.ForeignKey(Bucket, on_delete=models.CASCADE, related_name='social_buckets')

@receiver(post_save, sender=Bucket)
def create_social_bucket(sender, instance, created, **kwargs):
    if created and instance.category == 'social':
        SocialBucket.objects.create(bucket=instance)

此外，還有另一種實現方式。 它是從 Bucket model 中覆蓋保存方法：

class SocialBucket(models.Model):
    bucket = models.ForeignKey('Bucket', on_delete=models.CASCADE, related_name='social_buckets')  # putting the class of model as a string value to avoid import error

class Bucket(models.Model):
    ....

    def create_social_bucket(self, create_social_bucket=False):
       if create_social_bucket:
           Social.objects.create(bucket_id=self.id)

   def save(self, *args, **kwargs):
       if self.stock_list:
           self.stock_count = len(self.stock_list)
       create_social_bucket = False
       if self.pk is None and self.category == 'social':
           create_social_bucket = True
       super().save(*args, **kwargs)
       self.create_social_bucket(create_social_bucket)

Answer 3

雖然不建議這樣做。 您可以破解和硬編碼：

class BucketCreateSerializer(serializers.ModelSerializer):
    class Meta:
        model = Bucket
        fields = ('owner','category','name','about')
        read_only_fields = ['owner']
    
    def create(self, validated_data):
        user = self.context['request'].user
        bucket = Bucket.objects.create(
         owner=user, 
         **validated_data
        )
        bucket.save()
        # creating SocialBucket if category is social. You can later update it.
        if bucket.category == 'social':
            social_bucket = SocialBucket()
            social_bucket.bucket = bucket
            social_bucket.save()
        return bucket