0.0.49

fix routes

RoutesApp/js_views.py

@@ -12,6 +12,7 @@ from datetime import datetime
 from django.template.loader import render_to_string
 from django.urls import reverse
 from .forms import *
+from .funcs import get_routes_for_user
 
 
 def new_route_view_ajax(request):
@@ -27,39 +28,18 @@ def new_route_view_ajax(request):
 
 
 def get_routes_ajax(request):
-    from ReferenceDataApp.models import Airport, Country, City
+
 
     if request.method != 'POST':
         raise Http404
 
     try:
-        routes = []
-        if request.user and request.user.is_authenticated:
-            routes = Route.objects.filter(owner=request.user)
+        routes_Dict = get_routes_for_user(request.user)
+        if 'errors' in routes_Dict:
+            return JsonResponse(routes_Dict, status=400)
 
-            for route in routes:
 
-                try:
-
-                    if route.type_transport == 'avia':
-                        route.from_airport = Airport.objects.get(id=route.from_address_point)
-                        route.to_airport = Airport.objects.get(id=route.to_address_point)
-                        route.from_city = route.from_airport.city
-                        route.to_city = route.to_airport.city
-                    else:
-                        route.from_city = City.objects.get(id=route.from_address_point)
-                        route.to_city = City.objects.get(id=route.to_address_point)
-
-                    route.from_country = route.from_city.country
-                    route.to_country = route.to_city.country
-                except Exception as e:
-                    msg = f'get route points = {str(e)}'
-                    print(msg)
-
-        Dict = {
-            'routes': routes
-        }
-        html = render_to_string('blocks/profile/b_my_routes.html', Dict, request=request)
+        html = render_to_string('blocks/profile/b_my_routes.html', routes_Dict, request=request)
 
         res_Dict = {
             'html': html
@@ -97,11 +77,15 @@ def create_route_ajax(request):
         obj.owner = request.user
         obj.save()
 
-        routes = Route.objects.filter(owner=request.user)
-        Dict = {
-            'routes': routes
-        }
-        html = render_to_string('blocks/profile/b_my_routes.html', Dict, request=request)
+        routes_Dict = get_routes_for_user(request.user)
+
+        if 'errors' in routes_Dict:
+            form.errors.update(routes_Dict['errors'])
+            Dict = {'form': form}
+            html = render_to_string('blocks/profile/b_new_route.html', Dict, request=request)
+            return JsonResponse({'html': html}, status=400)
+
+        html = render_to_string('blocks/profile/b_my_routes.html', routes_Dict, request=request)
 
         res_Dict = {
             'html': html