diff --git a/GeneralApp/urls.py b/GeneralApp/urls.py index 93dc521..0066e4e 100644 --- a/GeneralApp/urls.py +++ b/GeneralApp/urls.py @@ -8,4 +8,5 @@ urlpatterns = [ path('', MainPage, name='main'), path('page//', StaticPageView, name='static_page'), path('test_code', test_code, name='test_code'), + path('generate_routes//', generate_routes, name='generate_routes'), ] \ No newline at end of file diff --git a/GeneralApp/views.py b/GeneralApp/views.py index 0fd41a0..590e6e3 100644 --- a/GeneralApp/views.py +++ b/GeneralApp/views.py @@ -16,6 +16,65 @@ from webpush import send_user_notification import json from datetime import datetime, timedelta +def generate_routes(request, routes_count): + if (not request.user + or not request.user.is_active + or not request.user.is_authenticated + or not request.user.is_staff + ): + raise Http404 + + from RoutesApp.funcs import get_city_by_type_transport_and_address_point + from RoutesApp.models import Route + from ReferenceDataApp.models import Airport, City + + res = None + + from_air = Airport.objects.get(iata_code='MSQ') + to_air = Airport.objects.get(iata_code='SVO') + + routes = [ + Route( + type_transport='road', + departure_DT=datetime.now() + timedelta(days=7), + arrival_DT=datetime.now() + timedelta(days=8), + from_address_point=to_air.city.id, + to_address_point=from_air.city.id, + from_city=to_air.city, + to_city=from_air.city, + weight=item, + phone='0987654321', + owner=request.user + ) for item in range(routes_count) + ] + + # routes = [ + # Route( + # type_transport='avia', + # departure_DT=datetime(year=2024, month=9, day=1), + # arrival_DT=datetime(year=2024, month=9, day=3), + # from_address_point = from_air.id, + # to_address_point = to_air.id, + # from_city = from_air.city, + # to_city = to_air.city, + # weight = item, + # phone = '1234567890', + # owner = request.user + # ) for item in range(1000) + # ] + + Route.objects.bulk_create(routes) + + if res: + if type(res) == str: + return HttpResponse(res) + else: + return res + + return HttpResponse('finished') + + + def test_code(request): if (not request.user @@ -25,6 +84,8 @@ def test_code(request): ): raise Http404 + res = '' + from RoutesApp.search_matches import search_matches from RoutesApp.models import Route search_matches(Route.objects.filter(from_city__id=57062))