Ali Asadi
eb5eb38665
django-filter was already installed and set as DEFAULT_FILTER_BACKENDS but unused everywhere. Replace hand-rolled get_queryset filtering with FilterSet classes (stores/catalog/locations/reviews) and filterset_fields (orders status). drf-spectacular auto-documents these for swagger, so the manual OpenApiParameter declarations for the migrated fields are removed (stores keeps lat/lng manual since geo-distance isn't a plain filter). Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
38 lines
1.5 KiB
Python
38 lines
1.5 KiB
Python
from rest_framework import mixins, viewsets
|
|
from rest_framework.permissions import AllowAny, IsAuthenticated
|
|
|
|
from apps.gooyal_oauth2.rest_framework import IsAuthenticatedOrTokenMatchesOASRequirements
|
|
|
|
from .filters import CityFilter, NeighborhoodFilter
|
|
from .models import Address, City, Neighborhood
|
|
from .serializers import AddressSerializer, CitySerializer, NeighborhoodSerializer
|
|
|
|
|
|
class CityViewSet(mixins.ListModelMixin, mixins.RetrieveModelMixin, viewsets.GenericViewSet):
|
|
schema_tags = ['Locations']
|
|
permission_classes = [AllowAny]
|
|
serializer_class = CitySerializer
|
|
queryset = City.objects.filter(is_active=True)
|
|
filterset_class = CityFilter
|
|
|
|
|
|
class NeighborhoodViewSet(mixins.ListModelMixin, mixins.RetrieveModelMixin, viewsets.GenericViewSet):
|
|
schema_tags = ['Locations']
|
|
permission_classes = [AllowAny]
|
|
serializer_class = NeighborhoodSerializer
|
|
queryset = Neighborhood.objects.filter(is_active=True).select_related('city')
|
|
filterset_class = NeighborhoodFilter
|
|
|
|
|
|
class AddressViewSet(viewsets.ModelViewSet):
|
|
schema_tags = ['Addresses']
|
|
serializer_class = AddressSerializer
|
|
permission_classes = [IsAuthenticatedOrTokenMatchesOASRequirements]
|
|
required_alternate_scopes = {
|
|
"POST": [[]],
|
|
}
|
|
|
|
def get_queryset(self):
|
|
if getattr(self, 'swagger_fake_view', False):
|
|
return Address.objects.none()
|
|
return Address.objects.filter(user=self.request.user).select_related('city', 'neighborhood')
|