IoT-Systems/edge/app/usecases/data_processing.py

import time
from app.entities.agent_data import AgentData
from app.entities.processed_agent_data import ProcessedAgentData

_last_processed_times = {}

def process_agent_data(
    agent_data: AgentData,
) -> ProcessedAgentData:
    """
    Process agent data and classify the state of the road surface.
    Parameters:
        agent_data (AgentData): Agent data that containing accelerometer, GPS, and timestamp.
    Returns:
        processed_data_batch (ProcessedAgentData): Processed data containing the classified state of the road surface and agent data.
    """
    user_id = agent_data.user_id
    road_state = "normal"

    curr_time = time.time()
    last_processed_time = _last_processed_times.get(user_id, 0)
    
    if curr_time - last_processed_time > 1.0: 
        if agent_data.accelerometer.z < -1.0:
            road_state = "pothole"
        elif agent_data.accelerometer.z > 1.0:
            road_state = "bump"

    _last_processed_times[user_id] = curr_time

    return ProcessedAgentData(
        road_state=road_state,
        agent_data=agent_data
    )