Wine regions route

This commit is contained in:
Kuroshini 2019-10-16 14:11:07 +03:00
parent cb1366b57b
commit f8c9a5c830
4 changed files with 26 additions and 0 deletions

View File

@ -169,6 +169,9 @@ class Address(models.Model):
class WineRegionQuerySet(models.QuerySet):
"""Wine region queryset."""
def with_sub_region_related(self):
return self.prefetch_related('wine_sub_region')
class WineRegion(models.Model, TranslatedFieldsMixin):
"""Wine region model."""
@ -207,6 +210,7 @@ class WineSubRegion(models.Model):
"""Wine sub region model."""
name = models.CharField(_('name'), max_length=255)
wine_region = models.ForeignKey(WineRegion, on_delete=models.PROTECT,
related_name='wine_sub_region',
verbose_name=_('wine sub region'))
old_id = models.PositiveIntegerField(_('old id'), default=None,
blank=True, null=True)

View File

@ -189,3 +189,14 @@ class WineSubRegionBaseSerializer(serializers.ModelSerializer):
'id',
'name',
]
class WineRegionSerializer(WineRegionBaseSerializer):
"""Wine region w/ subregion serializer"""
wine_sub_region = WineSubRegionBaseSerializer(allow_null=True, many=True)
class Meta(WineRegionBaseSerializer.Meta):
fields = WineRegionBaseSerializer.Meta.fields + [
'wine_sub_region'
]

View File

@ -17,4 +17,6 @@ urlpatterns = [
path('regions/', views.RegionListView.as_view(), name='region-list'),
path('regions/<int:pk>/', views.RegionRetrieveView.as_view(), name='region-retrieve'),
path('wine-regions/', views.WineRegionListView.as_view(), name='wine-region-list'),
]

View File

@ -63,6 +63,15 @@ class RegionListView(RegionViewMixin, generics.ListAPIView):
serializer_class = serializers.CountrySerializer
class WineRegionListView(generics.ListAPIView):
"""List view for model WineRegion"""
pagination_class = None
model = models.WineRegion
permission_classes = (permissions.AllowAny,)
queryset = models.WineRegion.objects.with_sub_region_related().all()
serializer_class = serializers.WineRegionSerializer
class RegionDestroyView(RegionViewMixin, generics.DestroyAPIView):
"""Destroy view for model Country"""
serializer_class = serializers.CountrySerializer