improve cancel download

Author: pyquick

oclp_r/support/network_handler.py

@@ -529,6 +529,22 @@ def is_active(self) -> bool:
             return True
         return False
 
+    def delete_temp_files(self) -> None:
+        """
+        Delete temporary files created during download
+        """
+        try:
+            # Delete the partially downloaded file
+            if self.filepath.exists():
+                self.filepath.unlink()
+                logging.info(f"Deleted partially downloaded file: {self.filepath}")
+            
+            # Delete the progress file
+            if self.progress_file.exists():
+                self.progress_file.unlink()
+                logging.info(f"Deleted progress file: {self.progress_file}")
+        except Exception as e:
+            logging.warning(f"Failed to delete temporary files: {str(e)}")
 
     def stop(self) -> None:
         """
@@ -539,4 +555,8 @@ def stop(self) -> None:
 
         self.should_stop = True
         if self.active_thread and self.active_thread.is_alive():
-            self.active_thread.join(timeout=10)
+            self.active_thread.join(timeout=10)
+        
+        # Delete temporary files if download was cancelled by user
+        if not self.download_complete:
+            self.delete_temp_files()

oclp_r/wx_gui/gui_kdk_dl.py

@@ -294,7 +294,10 @@ def is_dir_writable(dirpath):
             )
             if download_obj.download_complete is False:
                 import os
-                os.remove(self.constants.user_download_file+"/"+file_name)
+                try:
+                    os.remove(self.constants.user_download_file+"/"+file_name)
+                except:
+                    pass
                 self.on_return_to_main_menu()
                 return
             self.on_return_to_main_menu()

oclp_r/wx_gui/gui_ml_dl.py

@@ -275,7 +275,10 @@ def is_dir_writable(dirpath):
             )
             if download_obj.download_complete is False:
                 import os
-                os.remove(self.constants.user_download_file+"/"+file_name)
+                try:
+                    os.remove(self.constants.user_download_file+"/"+file_name)
+                except:
+                    pass
                 self.on_return_to_main_menu()
                 return