fix routes models
This commit is contained in:
SDE
2023-08-29 18:34:11 +03:00
parent 1f3bb2dce9
commit 01863bcf38
6 changed files with 89 additions and 4 deletions

View File

@@ -14,7 +14,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_Dict
from .funcs import *
def edit_route_ajax(request):
@@ -217,6 +217,7 @@ def get_routes_ajax(request):
def create_or_change_route_ajax(request, route_id=None):
from ReferenceDataApp.models import Airport, City
if request.method != 'POST':
raise Http404
@@ -249,6 +250,13 @@ def create_or_change_route_ajax(request, route_id=None):
obj = form.save(commit=False)
if 'owner_type' in data:
obj.owner_type = data['owner_type']
if obj.from_address_point:
obj.from_city = get_city_by_type_transport_and_address_point(obj.type_transport, obj.from_address_point)
if obj.to_address_point:
obj.to_address_point = get_city_by_type_transport_and_address_point(obj.type_transport, obj.to_address_point)
obj.owner = request.user
obj.save()