我想在Django Rest Framework中获取ForeignKey的url

时间:2018-06-01 22:06:52

标签: python django django-rest-framework

我有(venv) Kurts-MacBook-Pro-2:lucy-web kurtpeek$ python manage.py test lucy_web.tests.test_session_recommendation.SessionRecommendationTestCase.test_create_session_from_non_existent_session_type_title_triggers_error_logging Creating test database for alias 'default'... System check identified no issues (0 silenced). . ---------------------------------------------------------------------- Ran 1 test in 0.004s OK Destroying test database for alias 'default'... 的模型HyperlinkedModelSerializer,我希望将超链接返回到该字段中引用的实例,但我将整个对象嵌套在json中。

这些是我的模特:

ForeignKey

这是我的序列化器:

class Hotel(models.Model):

    ONE_STAR = '*'
    TWO_STARS = '**'
    THREE_STARS = '***'
    FOUR_STARS = '****'
    FIVE_STARS = '****'
    GRAND_TOURISM = 'GRAND_TOURISM'
    NA = 'NA'
    SPECIAL = 'SPECIAL'
    ECO = 'ECO'
    BOUTIQUE = 'BOUTIQUE'

    HOTEL_CATEGORY_CHOICES = (
        (ONE_STAR, _('*')),
        (TWO_STARS, _('**')),
        (THREE_STARS, _('***')),
        (FOUR_STARS, _('****')),
        (FIVE_STARS, _('*****')),
        (GRAND_TOURISM, _('Grand Tourism')),
        (NA, _('NA')),
        (SPECIAL, _('Special')),
        (ECO, _('Eco-Hotel')),
        (BOUTIQUE, _('Boutique-Hotel'))
    )

    company = models.OneToOneField(Company, on_delete=models.CASCADE, primary_key=True, verbose_name=_('Company'))
    code = models.CharField(max_length=10, verbose_name=_('Code'))
    city = models.ForeignKey(City, on_delete=models.PROTECT, related_name='hotels', verbose_name=_('City'))
    category = models.CharField(max_length=20, choices=HOTEL_CATEGORY_CHOICES, verbose_name=_('Category'))
    capacity = models.IntegerField(verbose_name=_('Capacity'))
    position = models.DecimalField(max_digits=11, decimal_places=2, default=0.00, verbose_name=_('Position'))
    in_pickup = models.BooleanField(default=False, verbose_name=_('In pickup?'))
    is_active = models.BooleanField(default=True, verbose_name=_('Is active?'))

    class Meta:
        verbose_name = _('Hotel')
        verbose_name_plural = _('Hotels')

    def __str__(self):
        return self.company.name

class Company(models.Model):
    name = models.CharField(max_length=40, verbose_name=_('Name'))
    legal_name = models.CharField(max_length=100, null=True, blank=True, verbose_name=_('Legal name'))
    tax_id = models.CharField(max_length=12, null=True, blank=True, verbose_name=_('Tax ID'))
    url = models.URLField(null=True, blank=True, verbose_name=_('URL'))
    address = models.TextField(max_length=400, null=True, blank=True, verbose_name=_('Address'))

    class Meta:
        verbose_name = _('Company')
        verbose_name_plural = _('Companies')

    def __str__(self):
        return "[{}]{}".format(self.id, self.name)

这是我的ViewSet:

class HotelProductsSerializer(serializers.HyperlinkedModelSerializer):
    company = CompanySerializer()
    products = serializers.SerializerMethodField()

    def get_products(self, instance):
        queryset = [product for product in instance.company.products.all()]
        return ProductSerializer(queryset, many=True, context=self.context).data

    class Meta:
        model = models.Hotel
        fields = ('company', 'products')

一切正常,但字段class HotelProductsViewSet(viewsets.ModelViewSet): permission_classes = (permissions.IsAuthenticated,) queryset = models.Hotel.objects.all() serializer_class = serializers.HotelProductsSerializer 会生成一个嵌套的Company对象,我希望得到一个指向该对象的超链接。

感谢您的帮助。

1 个答案:

答案 0 :(得分:0)

myFirstModule25字段被序列化为嵌套对象,因为您已使用嵌套序列化程序声明它。尝试更换行

company

有这样的东西

company = CompanySerializer()

您可能需要根据the docs调整选项。