42 lines
1.5 KiB
Python
42 lines
1.5 KiB
Python
from rest_framework import generics
|
|
from rest_framework.permissions import IsAuthenticated
|
|
|
|
from core.apps.shared.serializers.region.district import DistrictSerializer, VillageSerializer, RegionSerializer
|
|
from core.apps.shared.models import RegionModel, VillageModel, DistrictModel
|
|
|
|
|
|
class DistrictListCreateView(generics.ListCreateAPIView):
|
|
permission_classes = [IsAuthenticated]
|
|
serializer_class = DistrictSerializer
|
|
queryset = DistrictModel.objects.all()
|
|
|
|
def get_queryset(self):
|
|
region = self.request.query_params.get('region')
|
|
name = self.request.query_params.get('name')
|
|
if name:
|
|
return self.queryset.filter(name__icontains=name)
|
|
if region:
|
|
return self.queryset.filter(region=region)
|
|
return super().get_queryset()
|
|
|
|
|
|
class VillageListCreateView(generics.ListCreateAPIView):
|
|
permission_classes = [IsAuthenticated]
|
|
serializer_class = VillageSerializer
|
|
queryset = VillageModel.objects.all()
|
|
|
|
def get_queryset(self):
|
|
district = self.request.query_params.get('district')
|
|
name = self.request.query_params.get('name')
|
|
if district:
|
|
return self.queryset.filter(district=district)
|
|
if name:
|
|
return self.queryset.filter(name__icontains=name)
|
|
return super().get_queryset()
|
|
|
|
|
|
class RegionListCreateView(generics.ListCreateAPIView):
|
|
permission_classes = [IsAuthenticated]
|
|
serializer_class = RegionSerializer
|
|
queryset = RegionModel.objects.all()
|