Update proton_manager.py

Author: michal92299

source-code/proton_manager.py

@@ -145,8 +145,7 @@ def install_proton(self, version, proton_type, progress_callback=None):
                 print(f"No tar.gz asset found for {version}")
                 return False, f"No tar.gz asset found for {version}"
             dl_url = tar_asset['browser_download_url']
-            if progress_callback:
-                progress_callback("Downloading", 0, 100)
+            progress_callback("Downloading", 0, 100)
             with tempfile.NamedTemporaryFile(suffix='.tar.gz', delete=False) as temp_tar:
                 response = requests.get(dl_url, stream=True, timeout=30)
                 response.raise_for_status()
@@ -161,8 +160,7 @@ def install_proton(self, version, proton_type, progress_callback=None):
                             if progress_callback and total_size:
                                 progress_callback("Downloading", downloaded, total_size)
                 temp_tar_path = temp_tar.name
-            if progress_callback:
-                progress_callback("Extracting", 0, 100)
+            progress_callback("Extracting", 0, 100)
             extract_dir = os.path.join(self.protons_dir, version)
             os.makedirs(extract_dir, exist_ok=True)
             with tarfile.open(temp_tar_path) as tar:
@@ -193,18 +191,13 @@ def install_proton(self, version, proton_type, progress_callback=None):
         except Exception as e:
             logging.error(f"Error installing {proton_type} proton {version}: {e}")
             print(f"Error installing {proton_type} proton {version}: {e}")
-            if 'extract_dir' in locals():
-                shutil.rmtree(extract_dir, ignore_errors=True)
-            if 'temp_tar_path' in locals() and os.path.exists(temp_tar_path):
-                os.remove(temp_tar_path)
             return False, str(e)
 
     def install_custom_tar(self, tar_path, version, progress_callback=None):
         try:
             extract_dir = os.path.join(self.protons_dir, version)
             os.makedirs(extract_dir, exist_ok=True)
-            if progress_callback:
-                progress_callback("Extracting", 0, 100)
+            progress_callback("Extracting", 0, 100)
             with tarfile.open(tar_path) as tar:
                 for member in tar.getmembers():
                     member_path = os.path.join(extract_dir, member.name)
@@ -232,8 +225,6 @@ def install_custom_tar(self, tar_path, version, progress_callback=None):
         except Exception as e:
             logging.error(f"Error installing custom tar: {e}")
             print(f"Error installing custom tar: {e}")
-            if 'extract_dir' in locals():
-                shutil.rmtree(extract_dir, ignore_errors=True)
             return False, str(e)
 
     def install_custom_folder(self, src_folder, version):
@@ -248,8 +239,6 @@ def install_custom_folder(self, src_folder, version):
         except Exception as e:
             logging.error(f"Error installing custom folder: {e}")
             print(f"Error installing custom folder: {e}")
-            if 'dest' in locals():
-                shutil.rmtree(dest, ignore_errors=True)
             return False, str(e)
 
     def remove_proton(self, version):