Fix problem stopping PBar

slips_files/core/helpers/progress_bar.py

@@ -159,8 +159,8 @@ def pbar_supported(self) -> bool:
         return True
 
     def run(self):
+        """keeps receiving events until pbar reaches 100%"""
         try:
-            """keeps receiving events until pbar reaches 100%"""
             while self.pbar_supported():
                 try:
                     msg: dict = self.pipe.recv()
@@ -182,10 +182,12 @@ def run(self):
 
                 if event == "terminate":
                     self.terminate()
+                    return
 
                 if event == "print":
                     # let tqdm do th eprinting to avoid conflicts with the pbar
                     self.print(msg)
+
         except Exception as e:
             tqdm.write(f"PBar Error: {e}")
 

slips_files/core/output.py

@@ -289,6 +289,8 @@ def shutdown_gracefully(self):
         self.manager.shutdown()
         self.send_pipe.close()
         self.recv_pipe.close()
+        if hasattr(self, 'pbar'):
+            self.pbar.join(3)
 
 
     def tell_pbar(self, msg: dict):
@@ -309,7 +311,8 @@ def update(self, msg: dict):
             wanna log the text to all logfiles or the cli only?
             txt: text to log to the logfiles and/or the cli
             bar_info: {
-                input_type: only given when we send bar:'init', specifies the type of the input file given to slips
+                input_type: only given when we send bar:'init',
+                            specifies the type of the input file given to slips
                     eg zeek, argus, etc
                 total_flows: int,
         }