diving into adding connection to database ,simple test

app.py

@@ -1,9 +1,11 @@
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI, HTTPException,BackgroundTasks
 from typing import List, Union
+from pydantic import BaseModel
 import requests
 import ast
 import pymysql
-import asyncio
+import time
+
 
 # -------------------------
 # App Initialization
@@ -53,16 +55,13 @@ def get_traffic_in_container(coordinates):
     data = response.json()
     return [(el['lat'], el['lon']) for el in data['elements']]
 
+
 singlas_state = {}
-import time
 
 def open_signal(tl_id,state,time_):
     singlas_state[tl_id] = state 
-    start=time.time()
     time.sleep(time_)
-    end=time.time()
     singlas_state.pop(tl_id)
-    print(f"action for {tl_id} --> {state} is Done ,time {end-start}")
 
 
 # -------------------------
@@ -79,7 +78,28 @@ def get_State(tl_id):
         return singlas_state[tl_id]
     else:
         return "FREE"
+    
+from fastapi import FastAPI, WebSocket, WebSocketDisconnect
 
+singlas_state = {
+    "123": "EMR",
+    "456": "ACC",
+    # ...
+}
+
+@app.websocket("/ws/state")
+async def websocket_get_state(websocket: WebSocket):
+    await websocket.accept()
+    try:
+        while True:
+            tl_id = await websocket.receive_text()
+            state = singlas_state.get(tl_id, "FREE")
+            await websocket.send_text(state)
+    except WebSocketDisconnect:
+        print("Client disconnected")
+
+
+        
 @app.get("/test-db")
 def test_db():
     try:
@@ -92,14 +112,12 @@ def test_db():
     except Exception as e:
         return {"status": "error", "detail": str(e)}
 
-from pydantic import BaseModel
 
 class CoordinatesPayload(BaseModel):
     coords: Union[str, List[List[float]]]
     state: str
     time: int
 
-from fastapi import BackgroundTasks
 
 @app.post("/send-coordinates")
 def receive_coordinates(payload: CoordinatesPayload,background_tasks: BackgroundTasks):
@@ -115,8 +133,7 @@ def receive_coordinates(payload: CoordinatesPayload,background_tasks: Background
     
     if tl_ids:
         for tl_id in tl_ids:
-            background_tasks.add_task(open_signal, tl_id, state, time)
-            #open_signal(tl_id,state,time) #"EMR" # "ACC"
+            background_tasks.add_task(open_signal, tl_id, state, time) #"EMR" # "ACC"
 
         return {"found_signals": tl_ids}
     else: