1.6.12 fix next package for my_routes

RoutesApp/admin.py

@@ -3,6 +3,7 @@ from .models import *
 from django.contrib import admin
 
 class Admin_Route(Admin_Trans_BaseModel):
+    readonly_fields = ['highlight_end_DT', 'rising_DT']
     list_display = [
         'id', 'owner_type', 'receive_msg_by_email', 'type_transport', 'cargo_type',
         'departure_DT', 'from_city', 'from_place',

RoutesApp/funcs.py

@@ -118,10 +118,10 @@ 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)
+        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)

RoutesApp/js_views.py

@@ -10,7 +10,7 @@ from django.template import loader, RequestContext
 from django.contrib.auth.decorators import login_required
 from BaseModels.mailSender import techSendMail
 from django.utils.translation import gettext as _
-from datetime import datetime
+from datetime import datetime, timedelta
 from django.template.loader import render_to_string
 from django.urls import reverse
 from .forms import *
@@ -41,14 +41,20 @@ def highlight_route_ajax(request):
         msg = _('Нет доступа к выделению')
         return JsonResponse({'errors': msg})
 
-    route.highlight_color = '#FF0000'
-    route.save(update_fields=['highlight_color'])
 
     from SubscribesApp.funcs import get_cur_user_subscribe
     user_subscribe = get_cur_user_subscribe(request.user)
     user_subscribe.used_route_highlight_count += 1
     user_subscribe.save(update_fields=['used_route_highlight_count'])
 
+
+    route.highlight_color = '#FF0000'
+    highlight_hours = user_subscribe.get_highlight_hours()
+    route.highlight_end_DT = datetime.now() + timedelta(hours=highlight_hours)
+    route.save(update_fields=['highlight_color', 'highlight_end_DT'])
+
+
+
     Dict = {
         'route': route,
     }

RoutesApp/migrations/0008_route_highlight_end_dt.py

@@ -0,0 +1,18 @@
+# Generated by Django 4.2.2 on 2024-08-13 13:28
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('RoutesApp', '0007_rename_select_color_route_highlight_color'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='route',
+            name='highlight_end_DT',
+            field=models.DateTimeField(blank=True, null=True, verbose_name='Дата и время окончания выделения'),
+        ),
+    ]

RoutesApp/models.py

@@ -71,6 +71,10 @@ class Route(BaseModel):
         verbose_name=_('Цвет выделения'),
         blank=True, null=True
     )
+    highlight_end_DT = models.DateTimeField(
+        verbose_name=_('Дата и время окончания выделения'),
+        blank=True, null=True
+    )
 
     def __str__(self):
         if self.name: