display from_city and to_city on ui

backend/api/search/views.py

@@ -8,35 +8,8 @@ from rest_framework.exceptions import ValidationError
 from routes.constants.routeChoices import owner_type_choices
 from django.utils import timezone
 from django.db.models import Q
-import cyrtranslit
 from urllib.parse import unquote
-
-def get_city_variations(city_name: str) -> list[str]:
-    """
-    Получает варианты написания города, включая транслитерацию
-    """
-    variations = set()
-    
-    # добавляем оригинальное название и его варианты с разным регистром
-    variations.add(city_name)
-    variations.add(city_name.lower())
-    variations.add(city_name.title())
-    
-    # пробуем добавить транслитерации
-    try:
-        # пробуем транслитерировать в обе стороны
-        lat = cyrtranslit.to_latin(city_name, 'ru')
-        cyr = cyrtranslit.to_cyrillic(city_name, 'ru')
-        
-        # добавляем варианты транслитерации с разным регистром
-        for variant in [lat, cyr]:
-            variations.add(variant)
-            variations.add(variant.lower())
-            variations.add(variant.title())
-    except:
-        pass
-    
-    return list(variations)
+from api.utils.cityVariator import get_city_variations
 
 class SearchRouteListView(generics.ListAPIView):
     serializer_class = SearchRouteSerializer
@@ -48,15 +21,12 @@ class SearchRouteListView(generics.ListAPIView):
         
         # получаем маршрут из URL и разбиваем его на города
         route = self.kwargs.get('route', '')
-        print(f"Raw route from URL: {route}")
         
         if route:
             # декодируем URL дважды, так как он может быть дважды закодирован
             route = unquote(unquote(route))
-            print(f"Decoded route: {route}")
             try:
                 from_city, to_city = route.split('-', 1)
-                print(f"Split cities - from: {from_city}, to: {to_city}")
             except ValueError:
                 from_city = to_city = None
         else:
@@ -82,8 +52,7 @@ class SearchRouteListView(generics.ListAPIView):
                         to_city = unquote(to_city)
                 except Exception as e:
                     print(f"Error decoding to_city: {e}")
-            
-            print(f"Query params - from: {from_city}, to: {to_city}")
+
         
         valid_types = [choice[0] for choice in owner_type_choices]
         current_time = timezone.now()
@@ -99,10 +68,8 @@ class SearchRouteListView(generics.ListAPIView):
         
         # фильтруем по городам если они указаны
         if from_city:
-            print(f"Searching for from_city: {from_city}")
             # Получаем варианты написания для поиска
             from_city_variations = get_city_variations(from_city)
-            print(f"From city variations: {from_city_variations}")
             # Используем Q objects для поиска по обоим полям
             from_city_filter = (
                 Q(from_city__name__in=from_city_variations) |
@@ -111,10 +78,8 @@ class SearchRouteListView(generics.ListAPIView):
             queryset = queryset.filter(from_city_filter)
 
         if to_city:
-            print(f"Searching for to_city: {to_city}")
             # получаем варианты написания для поиска
             to_city_variations = get_city_variations(to_city)
-            print(f"To city variations: {to_city_variations}")
             # используем Q objects для поиска по обоим полям
             to_city_filter = (
                 Q(to_city__name__in=to_city_variations) |
@@ -128,4 +93,5 @@ class SearchRouteListView(generics.ListAPIView):
         else:  # customer
             queryset = queryset.filter(arrival_DT__gt=current_time)
             
+        # print(queryset)
         return queryset.order_by('-arrival_DT')