52 lines
1.8 KiB
Python
Executable File
52 lines
1.8 KiB
Python
Executable File
|
|
from rest_framework import viewsets
|
|
# from rest_framework.response import Response
|
|
from rest_framework.decorators import action
|
|
from django_filters.rest_framework import DjangoFilterBackend
|
|
from django.db import connection
|
|
from .. import models, serializers
|
|
from django.http import JsonResponse
|
|
|
|
class LineaViewSet(viewsets.ModelViewSet):
|
|
queryset = models.Linea.objects.all()
|
|
serializer_class = serializers.LineaSerializer
|
|
filter_backends = [DjangoFilterBackend]
|
|
filterset_fields = ['id_operador', 'route_short_name', 'route_long_name', 'vigente']
|
|
|
|
def paraderos_x_linea(request, pk):
|
|
paraderos = models.Paradero.objects \
|
|
.filter(vigente=True, lineaparadero__id_linea=pk) \
|
|
.values('id_paradero','stop_lat','stop_lon','stop_name') \
|
|
.all()
|
|
return JsonResponse(list(paraderos), safe=False)
|
|
|
|
def buses_x_linea(request, pk):
|
|
detalle_buses = []
|
|
paraderos = models.Paradero.objects \
|
|
.filter(vigente=True, lineaparadero__id_linea=pk) \
|
|
.values('id_paradero') \
|
|
.all()
|
|
|
|
query = "SELECT json_data \
|
|
from gtfs_posiciones_json \
|
|
where stop_id = %s"
|
|
|
|
for p in paraderos:
|
|
params = [ p['id_paradero'] ]
|
|
with connection.cursor() as cursor:
|
|
cursor.execute(query, params)
|
|
row = cursor.fetchone()
|
|
|
|
if row != None:
|
|
buses = list(filter(lambda linea: linea['linea'] == pk, row[0]))
|
|
for bus in buses:
|
|
for llegada in bus['Llegadas']:
|
|
data_bus = {
|
|
'patente': llegada['patente'],
|
|
'estimada_gps': llegada['EstimadaGPS'],
|
|
'distancia_gps': llegada['DistanciaGPS'],
|
|
}
|
|
detalle_buses.append(data_bus)
|
|
|
|
return JsonResponse(detalle_buses, safe=False)
|