debug mqtt

Utils.py

@@ -4,4 +4,5 @@ from pydantic import BaseModel
 class CoordinatesPayload(BaseModel):
     coords: Union[str, List[List[float]]]
     state: str
-    time: int
+    duration: int
+    delay : int

app.py

@@ -41,17 +41,17 @@ def main():
 def receive_coordinates(payload: CoordinatesPayload,background_tasks: BackgroundTasks):
     coords= payload.coords
     state_= payload.state
-    time_= payload.time
+    duration= payload.duration
+    delay= payload.delay
 
     if isinstance(coords, str):
         coords = ast.literal_eval(coords)
 
     traffic_lights = get_traffic_in_container(coords)
     tl_ids = check_signals(traffic_lights)
-    print("ok")
     if tl_ids:
         for tl_id in tl_ids:
-            background_tasks.add_task(open_signal, tl_id, state_, time_) #"EMR" # "ACC"
+            background_tasks.add_task(open_signal, tl_id, state_, duration,delay) #"EMR" # "ACC"
 
         return {"found_signals": tl_ids}
     else:

mqtt_manager.py

@@ -26,15 +26,22 @@ def topic_rep(tl_id):
 def get_State(tl_id):
     return state.singlas_state.get(tl_id, "FREE")
 
-
+def checked_ql(ql):
+    return True
 # ------------------ Listener ------------------ #
 def mqtt_listener():
     def on_message(client, userdata, message):
         msg = message.payload.decode()
         topic = message.topic
+        tl_id = topic_rep.split('/')[1]
         print(f"📥 Reply Received on {topic}: {msg}")
-        #if (msg == 'CHK'):
-        client.publish(topic_msg('6082411793'), 'DONE'+str(msg))
+        if (msg == 'AVBL'):
+            client.publish(topic_msg(tl_id), 'DONE'+str(msg))
+
+        if (msg.startswith("QL")):
+            ql = msg.split()[1]
+            if checked_ql(ql):
+              client.publish(topic_msg(tl_id), 'ACC'+str(msg))  
 
     client = mqtt.Client()
     client.connect(mqtt_broker, mqtt_port)
@@ -58,12 +65,17 @@ def mqtt_publisher_loop(tl_id):
     last_state = None
     while True:
         current_state = get_State(tl_id)
+        '''
         if current_state != last_state:
             topic = topic_msg(tl_id)
             pub_client.publish(topic, current_state)
             print(f"📡 Published {current_state} to {topic}")
             last_state = current_state
-        time.sleep(1)
+
+        '''
+        if current_state != 'FREE':
+            topic = topic_msg(tl_id)
+        time.sleep(.01)
 
 
 # ------------------ Startup ------------------ #

state.py

@@ -5,4 +5,21 @@ status_chk = "CHK"
 
 singlas_state = {
     "1698478721" :status_free,
-    "6082411793":status_free } 
+    "6082411793":status_free } 
+
+request = {
+    "1698478721":{
+        'State' : status_free,
+        'time' : None,
+        'Duration': None,
+        'accepted' : False
+    },
+    "6082411793":{
+        'State' : status_free,
+        'time' : None,
+        'Duration': None,
+        'accepted' : False
+
+    }
+
+}

traffic_utils.py

@@ -30,11 +30,20 @@ def get_rectangle_container(coordinates):
     min_lat, max_lat = min(lats), max(lats)
     min_lon, max_lon = min(lons), max(lons)
     return min_lat, max_lat, min_lon, max_lon
+import asyncio
 
-def open_signal(tl_id,state_,time_):
-    state.singlas_state[tl_id] = state_ 
-    time.sleep(time_)
-    state.singlas_state[tl_id] = state.status_free 
+async def open_signal(tl_id,state_,duration,delay):
+    check_time = 10
+    asyncio.sleep(delay)
+    state.request[tl_id]['State'] = state.status_chk
+    asyncio.sleep(check_time)
+    if  (state.request[tl_id]['accepted']== True):
+        state.request[tl_id]['State'] = state_
+        state.request[tl_id]['time'] = time.time()
+        state.request[tl_id]['duration'] = duration
+        return "Request Accepted"
+    else:
+        return "Request Refused , No Internet Connection or Very High Queue Length"