diff --git a/store/main.py b/store/main.py index 3edd1af..81d51a1 100644 --- a/store/main.py +++ b/store/main.py @@ -126,12 +126,13 @@ async def send_data_to_subscribers(user_id: int, data): @app.post("/processed_agent_data/") -async def create_processed_agent_data(data: List[ProcessedAgentData]): +async def create_processed_agent_data(data: List[ProcessedAgentData], user_id: int = Body(..., embed=True)): session = SessionLocal() try: created_data = [ { "road_state": item.road_state, + "user_id": user_id, "x": item.agent_data.accelerometer.x, "y": item.agent_data.accelerometer.y, "z": item.agent_data.accelerometer.z, @@ -147,7 +148,7 @@ async def create_processed_agent_data(data: List[ProcessedAgentData]): session.commit() for record in created_records: - await send_data_to_subscribers(jsonable_encoder(record)) + await send_data_to_subscribers(user_id, jsonable_encoder(record)) return created_records except Exception as err: session.rollback()