Siguientes funcionalidades

This commit is contained in:
hack_23030943_f11325
2026-05-22 21:24:43 -06:00
parent 369060a997
commit 5eae8782bf
7 changed files with 1593 additions and 173 deletions

Binary file not shown.

View File

@@ -17,10 +17,11 @@
from fastapi import FastAPI, HTTPException, Depends
from fastapi.middleware.cors import CORSMiddleware
from sqlalchemy import create_engine, Column, Integer, String, ForeignKey
from sqlalchemy import create_engine, Column, Integer, String, ForeignKey, inspect, text
from sqlalchemy.orm import declarative_base, sessionmaker, Session, relationship
from pydantic import BaseModel
from typing import Optional
from typing import Optional, List, Dict
from datetime import datetime, timezone, timedelta
import logging
# ---------------------------------------------------------------
@@ -62,11 +63,12 @@ class Usuario(Base):
id = Column(Integer, primary_key=True, index=True)
nombre = Column(String, nullable=False)
email = Column(String, nullable=False, unique=True, index=True)
# Token FCM que Flutter registrará al iniciar la app
fcm_token = Column(String, nullable=True)
# Relación 1-a-1 con Domicilio (un usuario, un domicilio registrado)
domicilio = relationship("Domicilio", back_populates="usuario", uselist=False)
# Relación 1-a-muchos con Domicilio (un usuario puede tener varias direcciones)
direcciones = relationship("Domicilio", back_populates="usuario", cascade="all, delete-orphan")
class Domicilio(Base):
@@ -83,15 +85,30 @@ class Domicilio(Base):
__tablename__ = "domicilios"
id = Column(Integer, primary_key=True, index=True)
usuario_id = Column(Integer, ForeignKey("usuarios.id"), unique=True)
usuario_id = Column(Integer, ForeignKey("usuarios.id"))
colonia = Column(String, nullable=False)
direccion = Column(String, nullable=False)
# route_id es dato interno - no lo devolvemos al cliente
route_id = Column(String, nullable=False)
usuario = relationship("Usuario", back_populates="domicilio")
usuario = relationship("Usuario", back_populates="direcciones")
# Crea las tablas en hackathon.db si no existen
# Asegura que la DB local tenga las columnas necesarias cuando se actualiza el esquema.
inspector = inspect(engine)
if inspector.has_table("usuarios"):
columnas_usuario = [col["name"] for col in inspector.get_columns("usuarios")]
if "email" not in columnas_usuario:
with engine.connect() as conn:
conn.execute(text("ALTER TABLE usuarios ADD COLUMN email TEXT"))
if inspector.has_table("domicilios"):
columnas_domicilio = [col["name"] for col in inspector.get_columns("domicilios")]
if "direccion" not in columnas_domicilio:
with engine.connect() as conn:
conn.execute(text("ALTER TABLE domicilios ADD COLUMN direccion TEXT NOT NULL DEFAULT ''"))
# Crea las tablas nuevas en hackathon.db si no existen.
Base.metadata.create_all(bind=engine)
@@ -103,27 +120,102 @@ Base.metadata.create_all(bind=engine)
# desarrollo. En producción, estos datos vendrían de la DB.
# ===============================================================
# Mapeo: Nombre de Colonia -> ID de Ruta interna
# El Flutter usa estas colonias para el Dropdown del login
COLONIAS_A_RUTAS: dict[str, str] = {
"Zona Centro": "RUTA-01",
"Col. Hidalgo": "RUTA-01",
"Col. Independencia":"RUTA-02",
"Col. Obrera": "RUTA-02",
"Col. San Juan": "RUTA-03",
"Fracc. Los Pinos": "RUTA-03",
"Col. Reforma": "RUTA-04",
}
# Rutas disponibles y sus posiciones GPS para monitoreo de avance.
ROUTE_DATA: List[Dict[str, object]] = [
{
"route_id": "RUTA-01",
"name": "Zona Centro - Las Arboledas",
"truck_id": 101,
"status": "EN_RUTA",
"positions": [
{"positionId": 1, "lat": 20.5111, "lng": -100.9037, "speed": 0, "timestamp": "2026-05-22T06:00:00Z"},
{"positionId": 2, "lat": 20.5185, "lng": -100.8450, "speed": 45, "timestamp": "2026-05-22T06:12:00Z"},
{"positionId": 3, "lat": 20.5215, "lng": -100.8142, "speed": 22, "timestamp": "2026-05-22T06:25:00Z"},
{"positionId": 4, "lat": 20.5212, "lng": -100.8175, "speed": 15, "timestamp": "2026-05-22T06:38:00Z"},
{"positionId": 5, "lat": 20.5210, "lng": -100.8210, "speed": 0, "timestamp": "2026-05-22T06:50:00Z"},
{"positionId": 6, "lat": 20.5235, "lng": -100.8212, "speed": 18, "timestamp": "2026-05-22T07:05:00Z"},
{"positionId": 7, "lat": 20.5260, "lng": -100.8215, "speed": 20, "timestamp": "2026-05-22T07:18:00Z"},
{"positionId": 8, "lat": 20.5111, "lng": -100.9037, "speed": 40, "timestamp": "2026-05-22T07:40:00Z"},
],
},
{
"route_id": "RUTA-02",
"name": "Sector Norte - Av. Tecnológico",
"truck_id": 102,
"status": "EN_RUTA",
"positions": [
{"positionId": 1, "lat": 20.5111, "lng": -100.9037, "speed": 0, "timestamp": "2026-05-22T06:05:00Z"},
{"positionId": 2, "lat": 20.5280, "lng": -100.8135, "speed": 38, "timestamp": "2026-05-22T06:18:00Z"},
{"positionId": 3, "lat": 20.5410, "lng": -100.8130, "speed": 25, "timestamp": "2026-05-22T06:30:00Z"},
{"positionId": 4, "lat": 20.5445, "lng": -100.8132, "speed": 12, "timestamp": "2026-05-22T06:45:00Z"},
{"positionId": 5, "lat": 20.5480, "lng": -100.8135, "speed": 0, "timestamp": "2026-05-22T06:58:00Z"},
{"positionId": 6, "lat": 20.5515, "lng": -100.8138, "speed": 15, "timestamp": "2026-05-22T07:10:00Z"},
{"positionId": 7, "lat": 20.5540, "lng": -100.8110, "speed": 22, "timestamp": "2026-05-22T07:25:00Z"},
{"positionId": 8, "lat": 20.5111, "lng": -100.9037, "speed": 45, "timestamp": "2026-05-22T07:50:00Z"},
],
},
]
# Horarios estimados por ruta (ETA en texto amigable para el usuario)
# Formato: { route_id: { "eta_texto": str, "eta_minutos": int } }
HORARIOS_POR_RUTA: dict[str, dict] = {
ROUTAS_POR_ID: Dict[str, Dict[str, object]] = {route["route_id"]: route for route in ROUTE_DATA}
HORARIOS_POR_RUTA: Dict[str, Dict[str, object]] = {
"RUTA-01": {"eta_texto": "Llega en aproximadamente 15 minutos", "eta_minutos": 15},
"RUTA-02": {"eta_texto": "Llega en aproximadamente 30 minutos", "eta_minutos": 30},
"RUTA-03": {"eta_texto": "Llega en aproximadamente 45 minutos", "eta_minutos": 45},
"RUTA-03": {"eta_texto": "Llega en aproximadamente 40 minutos", "eta_minutos": 40},
"RUTA-04": {"eta_texto": "Llega en aproximadamente 60 minutos", "eta_minutos": 60},
"RUTA-05": {"eta_texto": "Llega en aproximadamente 75 minutos", "eta_minutos": 75},
"RUTA-12": {"eta_texto": "Llega en aproximadamente 35 minutos", "eta_minutos": 35},
"RUTA-13": {"eta_texto": "Llega en aproximadamente 40 minutos", "eta_minutos": 40},
}
HORARIOS_POR_COLONIA: List[Dict[str, str]] = [
{"colonia": "Zona Centro", "routeId": "RUTA-01", "horarioEstimado": "Matutino (06:30 - 07:15)"},
{"colonia": "Las Arboledas", "routeId": "RUTA-01", "horarioEstimado": "Matutino (07:00 - 07:30)"},
{"colonia": "Trojes", "routeId": "RUTA-13", "horarioEstimado": "Matutino (06:40 - 07:10)"},
{"colonia": "San Juanico", "routeId": "RUTA-03", "horarioEstimado": "Matutino (06:45 - 07:15)"},
{"colonia": "Los Olivos", "routeId": "RUTA-04", "horarioEstimado": "Matutino (07:00 - 07:40)"},
{"colonia": "Rancho Seco", "routeId": "RUTA-05", "horarioEstimado": "Vespertino (14:15 - 15:00)"},
{"colonia": "Las Insurgentes", "routeId": "RUTA-12", "horarioEstimado": "Matutino (06:35 - 07:10)"},
]
COLONIAS_A_RUTAS: Dict[str, str] = {
item["colonia"]: item["routeId"] for item in HORARIOS_POR_COLONIA
}
TRIGGER_NOTIFICATIONS = {
"ROUTE_START": {
"position_id": 2,
"title": "¡Ruta Iniciada!",
"body": "El camión recolector ha salido del Relleno Sanitario rumbo a tu sector. Asegúrate de tener listos tus residuos."
},
"TRUCK_PROXIMITY": {
"position_id": 4,
"title": "Camión Cercano",
"body": "El camión está a menos de 15 minutos de tu domicilio. Es momento de sacar tus bolsas a la acera."
},
"ROUTE_COMPLETED": {
"position_id": 8,
"title": "Servicio Finalizado",
"body": "El camión de tu sector ha concluido su jornada de recolección diaria."
},
"GPS_OUTAGE": {
"title": "Alerta GPS",
"body": "El GPS del camión dejó de reportar su ubicación. Estamos investigando la ruta."
},
}
# Estado de avance de cada ruta en memoria.
ROUTE_STATE: Dict[str, Dict[str, object]] = {}
for route in ROUTE_DATA:
posiciones = route.get("positions", [])
ROUTE_STATE[route["route_id"]] = {
"last_position_id": posiciones[0]["positionId"] if posiciones else 0,
"last_timestamp": datetime.now(timezone.utc),
"gps_ok": True,
"gps_alert_sent": False,
"triggers_sent": {trigger_key: False for trigger_key in TRIGGER_NOTIFICATIONS if trigger_key != "GPS_OUTAGE"},
}
# Tipos de evento válidos para el simulador
TIPOS_EVENTO_VALIDOS = ["en_camino", "llegando", "completado", "retrasado"]
@@ -194,11 +286,68 @@ class ETAResponse(BaseModel):
"""Respuesta del endpoint de ETA. Sin route_id por privacidad."""
usuario_id: int
colonia: str
ruta_nombre: str
ruta_status: str
gps_ok: bool
eta_texto: str
eta_minutos: int
mensaje_preventivo: str # Ej: "No saques tu basura aún"
class UsuarioRegisterRequest(BaseModel):
nombre: str
email: str
colonia: str
direccion: str
class DireccionRequest(BaseModel):
colonia: str
direccion: str
class UsuarioLoginRequest(BaseModel):
email: str
class DomicilioResponse(BaseModel):
colonia: str
direccion: str
class UsuarioResponse(BaseModel):
usuario_id: int
nombre: str
email: str
direcciones: List[DomicilioResponse]
class RoutePositionUpdateRequest(BaseModel):
position_id: int
lat: float
lng: float
timestamp: str
class RouteStatusResponse(BaseModel):
route_id: str
name: str
status: str
last_position_id: int
last_timestamp: str
gps_ok: bool
class RegisterResponse(BaseModel):
usuario_id: int
mensaje: str
class LoginResponse(BaseModel):
usuario_id: int
mensaje: str
class SimularEventoRequest(BaseModel):
"""
Payload para simular un evento de ruta.
@@ -268,6 +417,123 @@ def enviar_notificacion_firebase(fcm_token: str, titulo: str, cuerpo: str) -> bo
return False
def _obtener_ruta_por_colonia(colonia: str) -> Optional[Dict[str, object]]:
route_id = COLONIAS_A_RUTAS.get(colonia)
if not route_id:
return None
return ROUTAS_POR_ID.get(route_id)
def _calcular_eta_por_ruta(route_id: str) -> Dict[str, object]:
"""Calcula un ETA con base en la posición actual de la ruta."""
estado = ROUTE_STATE.get(route_id)
ruta = ROUTAS_POR_ID.get(route_id, {})
horario = HORARIOS_POR_RUTA.get(route_id)
if not horario:
return {
"eta_texto": "No hay horario disponible.",
"eta_minutos": 60,
}
if estado and ruta.get("positions"):
posiciones = ruta["positions"]
ultimo_id = estado.get("last_position_id", 1)
if ultimo_id >= len(posiciones):
return {
"eta_texto": "El servicio ya pasó por tu zona.",
"eta_minutos": 0,
}
pasos_restantes = max(0, len(posiciones) - ultimo_id)
eta = pasos_restantes * 10
return {
"eta_texto": f"Llega en aproximadamente {eta} minutos",
"eta_minutos": eta,
}
return {
"eta_texto": horario["eta_texto"],
"eta_minutos": horario["eta_minutos"],
}
def _verificar_gps_outage(route_id: str, db: Session) -> None:
"""Verifica si una ruta dejó de reportar GPS y notifica a los usuarios una sola vez."""
estado = ROUTE_STATE.get(route_id)
if not estado:
return
ultimo_timestamp = estado.get("last_timestamp", datetime.now(timezone.utc))
gps_ok = (datetime.now(timezone.utc) - ultimo_timestamp) < timedelta(minutes=10)
if not gps_ok and not estado.get("gps_alert_sent", False):
logger.warning(f"Alerta GPS outage para {route_id}. Enviando notificaciones.")
_notificar_ruta(db, route_id, "GPS_OUTAGE")
estado["gps_alert_sent"] = True
def _obtener_estado_ruta(route_id: str, db: Optional[Session] = None) -> Dict[str, object]:
ruta = ROUTAS_POR_ID.get(route_id)
estado = ROUTE_STATE.get(route_id, {})
if not ruta:
raise ValueError("Ruta no encontrada")
ultimo_timestamp = estado.get("last_timestamp", datetime.now(timezone.utc))
gps_ok = (datetime.now(timezone.utc) - ultimo_timestamp) < timedelta(minutes=10)
if db is not None and not gps_ok:
_verificar_gps_outage(route_id, db)
return {
"route_id": route_id,
"name": ruta.get("name", "Ruta desconocida"),
"status": ruta.get("status", "DESCONOCIDA"),
"last_position_id": estado.get("last_position_id", 0),
"last_timestamp": ultimo_timestamp.isoformat(),
"gps_ok": gps_ok,
}
def _procesar_trigger_posicion(route_id: str, position_id: int, db: Session) -> list[str]:
"""Envía notificaciones basadas en el position_id y evita duplicados."""
estado = ROUTE_STATE.get(route_id)
if not estado:
return []
mensajes = []
sent_map = estado.setdefault("triggers_sent", {})
for trigger_key, trigger in TRIGGER_NOTIFICATIONS.items():
if trigger_key == "GPS_OUTAGE":
continue
if trigger.get("position_id") == position_id and not sent_map.get(trigger_key, False):
mensajes.extend(_notificar_ruta(db, route_id, trigger_key))
sent_map[trigger_key] = True
return mensajes
def _notificar_ruta(db: Session, route_id: str, trigger_key: str) -> list[str]:
trigger = TRIGGER_NOTIFICATIONS.get(trigger_key)
if not trigger:
return [f"Trigger desconocido: {trigger_key}"]
domicilios = db.query(Domicilio).filter(Domicilio.route_id == route_id).all()
mensajes = []
for domicilio in domicilios:
usuario = domicilio.usuario
if not usuario or not usuario.fcm_token:
mensajes.append(f"Usuario no tiene token o no existe.")
continue
enviado = enviar_notificacion_firebase(usuario.fcm_token, trigger["title"], trigger["body"])
if enviado:
mensajes.append(f"Notificación enviada a {usuario.nombre} (ID {usuario.id}).")
else:
mensajes.append(f"Fallo al enviar a {usuario.nombre} (ID {usuario.id}).")
return mensajes
# ===============================================================
# ENDPOINT: SEED DE DATOS DE PRUEBA
#
@@ -286,21 +552,26 @@ def seed_datos(db: Session = Depends(get_db)):
return {"mensaje": "Ya hay datos en la DB. No se hizo nada."}
usuarios_seed = [
{"nombre": "Ana García", "colonia": "Zona Centro", "fcm_token": "token-ana-fake-001"},
{"nombre": "Carlos López", "colonia": "Col. Hidalgo", "fcm_token": "token-carlos-fake-002"},
{"nombre": "María Torres", "colonia": "Col. Independencia", "fcm_token": "token-maria-fake-003"},
{"nombre": "Pedro Ruiz", "colonia": "Col. San Juan", "fcm_token": "token-pedro-fake-004"},
{"nombre": "Ana García", "email": "ana@example.com", "colonia": "Zona Centro", "direccion": "Calle Principal 123", "fcm_token": "token-ana-fake-001"},
{"nombre": "Carlos López", "email": "carlos@example.com", "colonia": "Col. Hidalgo", "direccion": "Av. Hidalgo 45", "fcm_token": "token-carlos-fake-002"},
{"nombre": "María Torres", "email": "maria@example.com", "colonia": "Col. Independencia", "direccion": "Calle Luna 12", "fcm_token": "token-maria-fake-003"},
{"nombre": "Pedro Ruiz", "email": "pedro@example.com", "colonia": "Col. San Juan", "direccion": "Calle Sol 78", "fcm_token": "token-pedro-fake-004"},
]
for u in usuarios_seed:
colonia = u["colonia"]
route_id = COLONIAS_A_RUTAS.get(colonia, "RUTA-01")
usuario = Usuario(nombre=u["nombre"], fcm_token=u["fcm_token"])
usuario = Usuario(nombre=u["nombre"], email=u["email"], fcm_token=u["fcm_token"])
db.add(usuario)
db.flush() # flush para obtener el id antes del commit
domicilio = Domicilio(usuario_id=usuario.id, colonia=colonia, route_id=route_id)
domicilio = Domicilio(
usuario_id=usuario.id,
colonia=colonia,
direccion=u["direccion"],
route_id=route_id,
)
db.add(domicilio)
db.commit()
@@ -308,6 +579,162 @@ def seed_datos(db: Session = Depends(get_db)):
return {"mensaje": "Seed exitoso. Usuarios IDs: 1, 2, 3, 4"}
@app.post("/api/usuarios/register", response_model=RegisterResponse, tags=["Usuarios"])
def registrar_usuario(payload: UsuarioRegisterRequest, db: Session = Depends(get_db)):
existing = db.query(Usuario).filter(Usuario.email == payload.email).first()
if existing:
raise HTTPException(status_code=400, detail="El correo ya está registrado.")
route_id = COLONIAS_A_RUTAS.get(payload.colonia)
if not route_id:
raise HTTPException(status_code=400, detail="Colonia no válida.")
usuario = Usuario(nombre=payload.nombre, email=payload.email.lower().strip(), fcm_token=None)
db.add(usuario)
db.flush()
direccion = Domicilio(
usuario_id=usuario.id,
colonia=payload.colonia,
direccion=payload.direccion,
route_id=route_id,
)
db.add(direccion)
db.commit()
return RegisterResponse(usuario_id=usuario.id, mensaje="Usuario registrado correctamente.")
@app.post("/api/usuarios/login", response_model=LoginResponse, tags=["Usuarios"])
def login_usuario(payload: UsuarioLoginRequest, db: Session = Depends(get_db)):
usuario = db.query(Usuario).filter(Usuario.email == payload.email.lower().strip()).first()
if not usuario:
raise HTTPException(status_code=404, detail="Usuario no encontrado. Regístrate primero.")
return LoginResponse(usuario_id=usuario.id, mensaje="Login exitoso.")
@app.get("/api/usuarios/{usuario_id}", response_model=UsuarioResponse, tags=["Usuarios"])
def obtener_usuario(usuario_id: int, db: Session = Depends(get_db)):
usuario = db.query(Usuario).filter(Usuario.id == usuario_id).first()
if not usuario:
raise HTTPException(status_code=404, detail="Usuario no encontrado.")
direcciones = [
DomicilioResponse(colonia=d.colonia, direccion=d.direccion)
for d in usuario.direcciones
]
return UsuarioResponse(
usuario_id=usuario.id,
nombre=usuario.nombre,
email=usuario.email,
direcciones=direcciones,
)
@app.post("/api/usuarios/{usuario_id}/direcciones", tags=["Usuarios"])
def agregar_direccion(usuario_id: int, payload: DireccionRequest, db: Session = Depends(get_db)):
usuario = db.query(Usuario).filter(Usuario.id == usuario_id).first()
if not usuario:
raise HTTPException(status_code=404, detail="Usuario no encontrado.")
route_id = COLONIAS_A_RUTAS.get(payload.colonia)
if not route_id:
raise HTTPException(status_code=400, detail="Colonia no válida.")
direccion = Domicilio(
usuario_id=usuario.id,
colonia=payload.colonia,
direccion=payload.direccion,
route_id=route_id,
)
db.add(direccion)
db.commit()
return {"mensaje": "Dirección agregada correctamente."}
@app.get("/api/rutas", tags=["Rutas"])
def listar_rutas(db: Session = Depends(get_db)):
return {"rutas": [_obtener_estado_ruta(route["route_id"], db) for route in ROUTE_DATA]}
@app.post("/api/rutas/{route_id}/avanzar", response_model=RouteStatusResponse, tags=["Rutas"])
def avanzar_ruta(route_id: str, db: Session = Depends(get_db)):
if route_id not in ROUTAS_POR_ID:
raise HTTPException(status_code=404, detail="Ruta no encontrada.")
ruta = ROUTAS_POR_ID[route_id]
estado = ROUTE_STATE.get(route_id)
posiciones = ruta.get("positions", [])
if not estado or not posiciones:
raise HTTPException(status_code=500, detail="Estado interno de la ruta no disponible.")
actual = estado.get("last_position_id", 0)
if actual < len(posiciones):
siguiente = actual + 1
estado["last_position_id"] = siguiente
estado["last_timestamp"] = datetime.now(timezone.utc)
estado["gps_ok"] = True
estado["gps_alert_sent"] = False
mensaje_log = _procesar_trigger_posicion(route_id, siguiente, db)
if siguiente >= len(posiciones):
ruta["status"] = "COMPLETADO"
else:
ruta["status"] = "EN_RUTA"
else:
mensaje_log = []
if mensaje_log:
logger.info(f"Notificaciones disparadas en {route_id}: {mensaje_log}")
return RouteStatusResponse(**_obtener_estado_ruta(route_id, db))
@app.get("/api/rutas/{route_id}/estado", response_model=RouteStatusResponse, tags=["Rutas"])
def estado_ruta(route_id: str, db: Session = Depends(get_db)):
try:
estado = _obtener_estado_ruta(route_id, db)
except ValueError as e:
raise HTTPException(status_code=404, detail=str(e))
return RouteStatusResponse(**estado)
@app.post("/api/rutas/{route_id}/posicion", tags=["Rutas"])
def actualizar_posicion_ruta(route_id: str, payload: RoutePositionUpdateRequest, db: Session = Depends(get_db)):
if route_id not in ROUTAS_POR_ID:
raise HTTPException(status_code=404, detail="Ruta no encontrada.")
estado = ROUTE_STATE.get(route_id)
if not estado:
raise HTTPException(status_code=500, detail="Estado de ruta no inicializado.")
try:
timestamp = datetime.fromisoformat(payload.timestamp.replace("Z", "+00:00"))
except ValueError:
raise HTTPException(status_code=400, detail="Timestamp inválido. Usa formato ISO 8601 UTC.")
mensaje_log = []
if payload.position_id > estado["last_position_id"]:
mensaje_log.extend(_procesar_trigger_posicion(route_id, payload.position_id, db))
estado["last_position_id"] = payload.position_id
estado["last_timestamp"] = timestamp
estado["gps_ok"] = True
estado["gps_alert_sent"] = False
if payload.position_id == 8:
ROUTAS_POR_ID[route_id]["status"] = "COMPLETADO"
return {
"route_id": route_id,
"position_id": payload.position_id,
"timestamp": payload.timestamp,
"gps_ok": True,
"mensajes": mensaje_log,
}
# ===============================================================
# ENDPOINT 1: GET /api/eta/{usuario_id}
#
@@ -337,34 +764,30 @@ def obtener_eta(usuario_id: int, db: Session = Depends(get_db)):
detail=f"Usuario {usuario_id} no encontrado. ¿Corriste /api/seed?"
)
# Paso 2: Verificar que tiene domicilio registrado
if not usuario.domicilio:
# Paso 2: Verificar que tiene al menos una dirección registrada
if not usuario.direcciones:
raise HTTPException(
status_code=404,
detail=f"El usuario {usuario_id} no tiene domicilio registrado."
detail=f"El usuario {usuario_id} no tiene direcciones registradas."
)
colonia = usuario.domicilio.colonia
route_id = usuario.domicilio.route_id # Uso INTERNO, no se devuelve
direccion = db.query(Domicilio).filter(Domicilio.usuario_id == usuario_id).order_by(Domicilio.id.desc()).first()
colonia = direccion.colonia
route_id = direccion.route_id # Uso INTERNO, no se devuelve
ruta = ROUTAS_POR_ID.get(route_id, {})
estado_ruta = _obtener_estado_ruta(route_id, db)
# Paso 3: Buscar el horario de la ruta en los datos mockeados
horario = HORARIOS_POR_RUTA.get(route_id)
if not horario:
# Fallback gracioso: si la ruta no tiene horario, decimos que no hay info
raise HTTPException(
status_code=503,
detail="No hay información de horario disponible para esta zona."
)
# Paso 3: Calcular ETA usando el estado actual de la ruta
calculo = _calcular_eta_por_ruta(route_id)
# Paso 4: Construir respuesta con mensaje preventivo
eta_minutos = horario["eta_minutos"]
return ETAResponse(
usuario_id=usuario_id,
colonia=colonia,
eta_texto=horario["eta_texto"],
eta_minutos=eta_minutos,
mensaje_preventivo=generar_mensaje_preventivo(eta_minutos),
# NOTA: route_id NO está en ETAResponse -> privacidad garantizada
ruta_nombre=ruta.get("name", "Ruta desconocida"),
ruta_status=estado_ruta["status"],
eta_texto=calculo["eta_texto"],
eta_minutos=calculo["eta_minutos"],
mensaje_preventivo=generar_mensaje_preventivo(calculo["eta_minutos"]),
)
@@ -469,7 +892,7 @@ def listar_colonias():
Lista todas las colonias disponibles.
Flutter las usa para poblar el Dropdown del login screen.
"""
return {"colonias": list(COLONIAS_A_RUTAS.keys())}
return {"colonias": [item["colonia"] for item in HORARIOS_POR_COLONIA]}
# ===============================================================