aid_manager: handles empty task queue

Author: AlyaGomaa

slips_files/core/aid_manager.py

@@ -1,4 +1,5 @@
 from multiprocessing import Process, Queue
+from queue import Empty
 from threading import Event
 
 from slips_files.common.slips_utils import utils
@@ -38,7 +39,6 @@ def _worker_loop(self, aid_queue, db: DBManager):
         - Initialize DBManager once.
         - Loop forever processing tasks.
         """
-
         while not self.stop_profiler_workers_event.is_set():
             try:
                 task = aid_queue.get(timeout=1)
@@ -55,6 +55,8 @@ def _worker_loop(self, aid_queue, db: DBManager):
                 self.db.add_flow(flow, profileid, twid, label=label)
             except KeyboardInterrupt:
                 continue
+            except Empty:
+                continue
 
     def submit_aid_task(self, flow, profileid: str, twid: str, label: str):
         """