121 lines
3.8 KiB
Python
121 lines
3.8 KiB
Python
from .models import *
|
|
|
|
|
|
def get_city_by_type_transport_and_address_point(type_transport, address_point):
|
|
from ReferenceDataApp.models import Airport, City
|
|
|
|
try:
|
|
if type_transport == 'avia':
|
|
return Airport.objects.get(id=address_point).city
|
|
else:
|
|
return City.objects.get(id=address_point)
|
|
except Exception as e:
|
|
msg = f'get_city_by_type_transport_and_address_point Error = {str(e)}, type_transport = {type_transport}, address_point = {address_point}'
|
|
print(msg)
|
|
return None
|
|
|
|
def get_routes_Dict(user=None, data=None):
|
|
from ReferenceDataApp.models import Airport, Country, City
|
|
|
|
# if not user and user.is_authenticated:
|
|
# errors_Dict = {
|
|
# 'errors': {
|
|
# 'all__': f'ошибка идентификации пользователя'
|
|
# }
|
|
# }
|
|
# return errors_Dict
|
|
kwargs = {}
|
|
if user:
|
|
kwargs.update({
|
|
'owner': user
|
|
})
|
|
|
|
from_el = None
|
|
to_el = None
|
|
|
|
if data:
|
|
for key, val in data.items():
|
|
if val:
|
|
if key == 'weight':
|
|
weight_list = val.split(';')
|
|
if weight_list[0]:
|
|
kwargs.update({f'{key}__gte': int(weight_list[0])})
|
|
if weight_list[1]:
|
|
kwargs.update({f'{key}__lte': int(weight_list[1])})
|
|
|
|
if key == 'type_transport':
|
|
items_list = val.split(',')
|
|
kwargs.update({f'{key}__in': items_list})
|
|
|
|
if key not in (
|
|
'from_address_point_txt', 'to_address_point_txt', 'csrfmiddlewaretoken', 'sort', 'weight',
|
|
'from_el', 'to_el', 'from_address_point', 'to_address_point', 'type_transport'
|
|
):
|
|
kwargs.update({key: val})
|
|
|
|
if key == 'from_address_point':
|
|
kwargs.update({f'from_city__id': val})
|
|
|
|
if key == 'to_address_point':
|
|
kwargs.update({f'to_city__id': val})
|
|
|
|
if key == 'from_el':
|
|
from_el = int(val)
|
|
if key == 'to_el':
|
|
to_el = int(val)
|
|
|
|
routes = Route.objects.filter(**kwargs).order_by('-modifiedDT')
|
|
routes_count = routes.count()
|
|
|
|
if from_el and to_el:
|
|
routes = routes[from_el:to_el]
|
|
elif from_el:
|
|
routes = routes[from_el:]
|
|
elif to_el:
|
|
routes = routes[:to_el]
|
|
else:
|
|
routes = routes[:25]
|
|
|
|
last_block_routes = False
|
|
if to_el and to_el >= routes_count:
|
|
last_block_routes = True
|
|
|
|
res_Dict = {}
|
|
|
|
try:
|
|
|
|
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_routes_for_user get route Error = {str(e)}'
|
|
print(msg)
|
|
|
|
res_Dict = {
|
|
'routes': routes,
|
|
'last_block_routes': last_block_routes
|
|
}
|
|
return res_Dict
|
|
|
|
|
|
except Exception as e:
|
|
msg = f'get_routes_for_user = {str(e)}'
|
|
print(msg)
|
|
errors_Dict = {
|
|
'errors': {
|
|
'all__': msg
|
|
}
|
|
}
|
|
return errors_Dict |