50 lines
2.3 KiB
Python
50 lines
2.3 KiB
Python
from datetime import date, datetime, time, timedelta
|
|
|
|
from django.db.models import Q
|
|
from django.views.generic import TemplateView
|
|
|
|
from sncfgtfs.models import Stop, StopTime, CalendarDate, Calendar
|
|
|
|
|
|
class GareView(TemplateView):
|
|
template_name = "sncfgtfs/gare.html"
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
stop = Stop.objects.filter(name__iexact=kwargs["gare"], location_type=1).first()
|
|
context['stop'] = stop
|
|
next_departures = []
|
|
|
|
now = datetime.now()
|
|
if self.request.GET.get('date', None):
|
|
query_date = date.fromisoformat(self.request.GET['date'])
|
|
else:
|
|
query_date = now.date()
|
|
if self.request.GET.get('time', None):
|
|
query_time = time.fromisoformat(self.request.GET['time'])
|
|
else:
|
|
query_time = now.time()
|
|
context['query_date'] = query_date
|
|
context['query_time'] = query_time
|
|
|
|
query_time = query_time.isoformat(timespec='seconds')
|
|
query_time = timedelta(seconds=int(query_time[:2]) * 3600 + int(query_time[3:5]) * 60 + int(query_time[6:]))
|
|
|
|
qs = StopTime.objects.none()
|
|
for child in Stop.objects.filter(name__iexact=kwargs["gare"], location_type=0).all():
|
|
qs_child = StopTime.objects.filter(stop=child, pickup_type=0, departure_time__gte=query_time)
|
|
qs_child = qs_child.filter(Q(trip__service_id__in=CalendarDate.objects.filter(
|
|
date=query_date, exception_type=1).values_list('service_id'))
|
|
| Q(trip__service_id__in=Calendar.objects.filter(
|
|
start_date__lte=query_date,
|
|
end_date__gte=query_date,
|
|
**{f"{query_date:%A}".lower(): True})
|
|
.filter(~Q(id__in=CalendarDate.objects.filter(
|
|
date=query_date, exception_type=2).values_list('service_id')))
|
|
.values_list('id')))
|
|
qs = qs.union(qs_child)
|
|
next_departures += qs.order_by("departure_time")[:20].all()
|
|
print(len(next_departures))
|
|
context['next_departures'] = next_departures
|
|
return context
|