diff --git a/store/docker/db/structure.sql b/store/docker/db/structure.sql index 06b6f76..01566a3 100644 --- a/store/docker/db/structure.sql +++ b/store/docker/db/structure.sql @@ -7,5 +7,6 @@ CREATE TABLE processed_agent_data ( z FLOAT, latitude FLOAT, longitude FLOAT, - timestamp TIMESTAMP -); \ No newline at end of file + timestamp TIMESTAMP, + visible BOOLEAN +); diff --git a/store/main.py b/store/main.py index d11896c..6639e41 100644 --- a/store/main.py +++ b/store/main.py @@ -30,8 +30,8 @@ processed_agent_data = Table( Column("z", Float), Column("latitude", Float), Column("longitude", Float), - Column("visible", Boolean), Column("timestamp", DateTime), + Column("visible", Boolean), ) # WebSocket subscriptions @@ -82,8 +82,8 @@ def ProcessedAgentData_to_td(data: List[ProcessedAgentData]): "z": item.agent_data.accelerometer.z, "latitude": item.agent_data.gps.latitude, "longitude": item.agent_data.gps.longitude, - "visible": True, "timestamp": item.agent_data.timestamp, + "visible": True, } for item in data ] @@ -173,6 +173,7 @@ def update_processed_agent_data(processed_agent_data_id: int, data: ProcessedAge z=data.agent_data.accelerometer.z, latitude=data.agent_data.gps.latitude, longitude=data.agent_data.gps.longitude, + visible=data.visible, timestamp=data.agent_data.timestamp, ) ) diff --git a/store/schemas.py b/store/schemas.py index 688d502..a6a5339 100644 --- a/store/schemas.py +++ b/store/schemas.py @@ -12,8 +12,8 @@ class ProcessedAgentDataInDB(BaseModel): z: float latitude: float longitude: float - visible: bool timestamp: datetime + visible: bool # FastAPI models @@ -50,3 +50,4 @@ class AgentData(BaseModel): class ProcessedAgentData(BaseModel): road_state: str agent_data: AgentData + visible: bool