Update extract.py

updated to allow you to choose your own directory for input and output

extract.py

@@ -1,53 +1,62 @@
 import os
 import patoolib
 import shutil
+from alive_progress import alive_bar
+import time
+from tkinter import Tk
+from tkinter.filedialog import askdirectory
 
 # Get the user's home directory
 home_dir = os.path.expanduser("~")
 
 # Get the path to the XBLA folder
-xbla_dir = os.path.join(home_dir, "XBLA")
+print("Please choose your input folder: ")
+xbla_dir = askdirectory()
 
 # Get the path to the XBLA_Unpacked folder
-xbla_unpacked_dir = os.path.join(home_dir, "XBLA_Unpacked")
+print("Please choose your output folder: ")
+xbla_unpacked_dir = askdirectory()
 
 # Create the XBLA_Unpacked folder if it doesn't exist
 if not os.path.exists(xbla_unpacked_dir):
     os.mkdir(xbla_unpacked_dir)
 
 # Iterate over all of the RAR archives in the XBLA folder
-for archive in os.listdir(xbla_dir):
-    if archive.endswith(".rar"):
-        if not os.path.exists(os.path.join(xbla_unpacked_dir, archive[:-4].replace(" ", "_"))):
-            os.mkdir(os.path.join(xbla_unpacked_dir, archive[:-4].replace(" ", "_")))
-
-        # Unpack the archive to the XBLA_Unpacked folder
-        patoolib.extract_archive(os.path.join(xbla_dir, archive),
-                                 outdir = (os.path.join(xbla_unpacked_dir, archive[:-4].replace(" ", "_"))))
-
-        # Get the path to the innermost subdirectory
-        for subdir, dirs, files in os.walk(xbla_unpacked_dir):
-            for file in files:
-    # Get the path to the innermost file in each subdirectory
-                innermost_file = os.path.join(subdir, file)
+with alive_bar(len(os.listdir(xbla_dir))) as bar:
+    for archive in os.listdir(xbla_dir):
+        if archive.endswith(".rar"):
+            if not os.path.exists(os.path.join(xbla_unpacked_dir, archive[:-4].replace(" ", "_"))):
+                os.mkdir(os.path.join(xbla_unpacked_dir, archive[:-4].replace(" ", "_")))
+
+            # Unpack the archive to the XBLA_Unpacked folder
+            patoolib.extract_archive(os.path.join(xbla_dir, archive),
+                                        outdir = os.path.join(xbla_unpacked_dir, archive[:-4].replace(" ", "_")), verbosity=-1)
+
+            # Get the path to the innermost subdirectory
+            for subdir, dirs, files in os.walk(xbla_unpacked_dir):
+                for file in files:
+        # Get the path to the innermost file in each subdirectory
+                    innermost_file = os.path.join(subdir, file)
 
 
-        # Get the name of the extensionless file in the innermost subdirectory
-        extensionless_file = innermost_file
+            # Get the name of the extensionless file in the innermost subdirectory
+            extensionless_file = innermost_file
 
-        # Rename the extensionless file to match the name of the archive it came from
-        os.rename(os.path.join(extensionless_file), os.path.join(xbla_unpacked_dir, archive.replace(".rar", ".pirs")))
+            # Rename the extensionless file to match the name of the archive it came from
+            os.rename(os.path.join(extensionless_file), os.path.join(xbla_unpacked_dir, archive.replace(".rar", ".pirs")))
 
-        # Delete the innermost directory
-        shutil.rmtree(os.path.join(xbla_unpacked_dir, archive[:-4].replace(" ", "_")))
+            # Delete the innermost directory
+            shutil.rmtree(os.path.join(xbla_unpacked_dir, archive[:-4].replace(" ", "_")))
 
-        #Remove the .pirs extension
-        for subdir, dirs, files in os.walk(xbla_unpacked_dir):
-            for file in files:
-                src = file
-                dst = file.replace(".pirs", "")
-                os.rename(os.path.join(xbla_unpacked_dir, src), (os.path.join(xbla_unpacked_dir, dst)))
-
+            #Remove the .pirs extension
+            for subdir, dirs, files in os.walk(xbla_unpacked_dir):
+                for file in files:
+                    src = file
+                    dst = file.replace(".pirs", "")
+                    os.rename(os.path.join(xbla_unpacked_dir, src), (os.path.join(xbla_unpacked_dir, dst)))
+
+        print(file)
+        bar()
 
 # Define a function to create a text file named after each file in the XBLA_Unpacked folder, with the extension .xbox360, and inside that .xbox360 text file list the original file name without the extension.
 def create_xbox360_text_file(file_name):
@@ -64,10 +73,15 @@ def create_xbox360_text_file(file_name):
         # Create a text file for the file
         create_xbox360_text_file(file)
 
-
-
-
-
-
 # Print a message to let the user know that the operation was successful
-print("Unpacking complete!")
+print("Unpacking complete!")
+
+# Ask if the user wants to delete the .rar archives
+deleteRar = ""
+while deleteRar != "Y" and deleteRar != "y" and deleteRar != "N" and deleteRar != "n":
+    deleteRar = input("Would you like to delete the XBLA folder and the .rar archives inside it? Enter Y or N: ")
+    if deleteRar == "Y" or deleteRar == "y":
+        shutil.rmtree(xbla_dir)
+        print("XBLA folder has been deleted.")
+    else:
+        quit