diff --git a/project/api/views/dispositivo.py b/project/api/views/dispositivo.py index f7ca1dd..de96b0d 100755 --- a/project/api/views/dispositivo.py +++ b/project/api/views/dispositivo.py @@ -75,27 +75,31 @@ class DispositivoViewSet(viewsets.ModelViewSet): pk_linea = f'{t["route_id"]}-{t["direction_id"]}' if pk_linea not in lineas: - lineas[pk_linea] = models.Linea.objects.get(id_linea=pk_linea) + lineas[pk_linea] = models.Linea.objects.filter(id_linea=pk_linea).first() linea = lineas[pk_linea] - if pk_linea not in lineas_agrupadas: - lineas_agrupadas[pk_linea] = { - 'Linea': pk_linea, - 'Descripcion': linea.route_long_name, - 'TipoLocomocion': linea.route_type.descripcion, - 'colorFondo': linea.route_color, - 'colorTexto': linea.route_text_color, - 'Llegadas': [] - } + if linea == None: + logging.error(pk_linea) - lineas_agrupadas[pk_linea]['Llegadas'].append({ - 'patente': t['vehicle_license_plate'], - 'Planificada': None, - 'EstimadaGPS': t['hora_llegada'], - 'DistanciaGPS': None, - 'Mensajelinea': None, - }) + if linea != None: + if pk_linea not in lineas_agrupadas: + lineas_agrupadas[pk_linea] = { + 'Linea': linea.route_long_name, + 'Descripcion': linea.route_short_name, + 'TipoLocomocion': linea.route_type.descripcion, + 'colorFondo': linea.route_color, + 'colorTexto': linea.route_text_color, + 'Llegadas': [] + } + + lineas_agrupadas[pk_linea]['Llegadas'].append({ + 'patente': t['vehicle_license_plate'], + 'Planificada': None, + 'EstimadaGPS': t['hora_llegada'], + 'DistanciaGPS': None, + 'Mensajelinea': None, + }) detalle_lineas = []