Использование отложенной загрузки с DRF

Question

Я хотел бы сделать простой lazy loading с Django Rest Framework.У меня есть твердый фон с Laravel, где вы можете просто использовать его так:

Subscription::with('company')->paginate()

Но с DRF у меня проблемы.Одна компания может иметь одну подписку , и мои модели определены следующим образом:

Компания

class CompanyManager(BaseUserManager):
    def create_user(self, email, password=None):
        """
        Creates and saves a User with the given email and password
        """
        if not email:
            raise ValueError('Users must have an email address')

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

        user.set_password(password)

        user.save(using=self._db)

        return user


class Company(AbstractBaseUser):
    """
        Company refers to what we have referee in the official document as "Empresa Partner"

        Attributes:
            name: The name of the main company
            email: Email to contact the main company
            code: the code attributed to this company
    """
    id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
    email = models.EmailField(unique=True)
    username = models.CharField(blank=True, max_length=100)
    is_staff = models.BooleanField(blank=True, default=False)
    our_system_date = models.DateTimeField(auto_now=True)

    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = []

    objects = CompanyManager()

    def __str__(self):
        return self.email

Подписка

class Subscription(models.Model):
    """
    Subscriptions that references the plans and the companies
    :company:
        The owner of this subscription

    :plan:
        The plan this subscription is based on

    :starts:
        When the subscription starts

    :ends:
        When this subscription ends
    """
    company = models.ForeignKey(Company, on_delete=models.CASCADE)
    plan = models.ForeignKey(Plan, on_delete=models.CASCADE)
    starts = models.DateTimeField()
    ends = models.DateTimeField()
    created_at = models.DateTimeField(auto_now=True)

Я пытался использовать select_related, на который ссылается этот ответ В чем разница между select_related и prefetch_related в ORM Django? , но каждый раз, когда я делаю запрос к конечной точке, это не такпоказать связанный Company из Subscription

Как я использую select_related:

    def list(self, request, *args, **kwargs):
        queryset = Subscription.objects\
            .all().select_related('company')\
            .order_by('-created_at')
        page = self.paginate_queryset(queryset)

        if page is not None:
            serializer = SubscriptionSerializer(
                page,
                many=True,
            )

            return self.get_paginated_response(serializer.data)

В моем SubscriptionSerliazer:

class SubscriptionSerializer(ModelSerializer):
    class Meta:
        model = Subscription
        fields = (
            'company',
            'plan',
            'starts',
            'ends',
            'created_at',
        )

Answer 1

Если вы хотите вернуть поля связанной модели, а не только ее первичный ключ, вам нужно определить сериализатор для связанной модели (здесь: CompanySerializer) и указать родительскому сериализатору использовать его:

class SubscriptionSerializer(ModelSerializer):
    company = CompanySerializer()

    class Meta:
        model = Subscription
        fields = (
            'company',
            # ...
        )