sender page

backend/api/account/client/views.py

@@ -7,12 +7,13 @@ from rest_framework.response import Response
 from django.shortcuts import get_object_or_404
 from django.core.validators import validate_email
 from django.core.exceptions import ValidationError
+from django.db import models
 
 from api.auth.serializers import UserResponseSerializer
 from api.models import UserProfile
 from api.utils.decorators import handle_exceptions
-from routes.models import Route
-from .serializers import RouteSerializer, CreateRouteSerializer
+from routes.models import Route, City, Country
+from .serializers import RouteSerializer, CreateRouteSerializer, CitySerializer, CountrySerializer
 
 class UserDataView(ViewSet):
     permission_classes = [IsAuthenticated]
@@ -95,8 +96,53 @@ class AccountActionsView(ViewSet):
     
     @action(detail=False, methods=['post'])
     @handle_exceptions
-    def create_sender_route(self, request):
+    def create_route(self, request):
         serializer = CreateRouteSerializer(data=request.data)
         serializer.is_valid(raise_exception=True)
         serializer.save()
         return Response(serializer.data, status=status.HTTP_201_CREATED)
+
+class CityView(ViewSet):
+    
+    @action(detail=False, methods=['get'])
+    @handle_exceptions
+    def get_cities(self, request):
+        # получаем параметр country_id из query params
+        country_id = request.query_params.get('country_id')
+        
+        # базовый QuerySet
+        cities = City.objects.all()
+        
+        # фильтруем города по стране, если указан country_id
+        if country_id:
+            cities = cities.filter(country_id=country_id)
+            
+        # поиск по названию города
+        search = request.query_params.get('search')
+        if search:
+            cities = cities.filter(name__icontains=search)
+            
+        # ограничиваем количество результатов и сортируем по имени
+        cities = cities.order_by('name')[:100]
+        
+        return Response(CitySerializer(cities, many=True).data, status=status.HTTP_200_OK)
+
+class CountryView(ViewSet):
+    @action(detail=False, methods=['get'])
+    @handle_exceptions
+    def get_countries(self, request):
+        # базовый QuerySet
+        countries = Country.objects.all()
+        
+        # поиск по названию страны
+        search = request.query_params.get('search')
+        if search:
+            countries = countries.filter(
+                models.Q(international_name__icontains=search) |
+                models.Q(official_name__icontains=search)
+            )
+        
+        # сортируем по международному названию
+        countries = countries.order_by('international_name')
+        
+        return Response(CountrySerializer(countries, many=True).data, status=status.HTTP_200_OK)