32 lines
1.0 KiB
Python
32 lines
1.0 KiB
Python
from apps.authentication.api.v1.serializers.serializer import (
|
|
CitySerializer,
|
|
ProvinceSerializer
|
|
)
|
|
from apps.core.mixins.soft_delete_mixin import SoftDeleteMixin
|
|
from apps.authentication.models import City, Province
|
|
from rest_framework.viewsets import ModelViewSet
|
|
from rest_framework.response import Response
|
|
from rest_framework import status
|
|
|
|
|
|
class CityViewSet(ModelViewSet, SoftDeleteMixin): # noqa
|
|
""" Crud operations for city model """ #
|
|
queryset = City.objects.all()
|
|
serializer_class = CitySerializer
|
|
|
|
def list(self, request, *args, **kwargs):
|
|
""" return list of cities by province """
|
|
|
|
serializer = self.serializer_class(
|
|
self.queryset.filter(
|
|
province_id=int(request.GET['province'])
|
|
), many=True
|
|
)
|
|
return Response(serializer.data, status=status.HTTP_200_OK)
|
|
|
|
|
|
class ProvinceViewSet(ModelViewSet, SoftDeleteMixin):
|
|
""" Crud operations for province model """ #
|
|
queryset = Province.objects.all()
|
|
serializer_class = ProvinceSerializer
|