RenPy-Android-Unpack: Some improvements

RenPy-Android-Unpack/unpack.py

@@ -1,10 +1,24 @@
 #!python3
+import colorama
 import zipfile
-import os
 import shutil
+import os
+import sys
+
+
+def printer(msg, level):
+    match level:
+        case "info":
+            print(f"\033[100m[INFO] {msg}\033[49m")
+        case "warn":
+            print(f"\033[93m[WARN]\033[0m {msg}\033[49m")
+        case "err":
+            print(f"\033[31m[ERROR]\033[0m {msg} Exiting...\033[49m")
+            exit()
 
 
 def extract_assets(file):
+    try:
         with zipfile.ZipFile(file, 'r') as zip_ref:
             for content in zip_ref.namelist():
                 if content.split('/')[0] == 'assets':
@@ -16,8 +30,13 @@ def extract_assets(file):
                     os.rename('assets/res/mipmap-xxxhdpi-v4/icon_background.png', 'assets/android-icon_background.png')
                     os.rename('assets/res/mipmap-xxxhdpi-v4/icon_foreground.png', 'assets/android-icon_foreground.png')
                 except KeyError:
+                    try:
                         zip_ref.extract('res/drawable/icon.png', 'assets')
                         os.rename('assets/res/drawable/icon.png', 'assets/icon.png')
+                    except KeyError:
+                        printer("Icon not found. Maybe it is not supported apk?", "warn")
+    except zipfile.BadZipFile:
+        return printer("Cant extract .apk file!", "err")
 
 
 def rename_files(directory):
@@ -41,27 +60,29 @@ def rename_dirs(directory):
         os.rename(dir__, f'{folder}/{newname}')
 
 
+def remove_unneeded(names, ignore):
+    for name in names:
+        try:
+            shutil.rmtree(name)
+        except FileNotFoundError:
+            if not ignore:
+                printer(f"Path {name} not found!", "warn")
+
+
 if __name__ == '__main__':
+    if sys.platform == "win32":
+        colorama.init()
     for filename in os.listdir(os.getcwd()):
-        renpy_warn = 0
         if os.path.splitext(filename)[1] == '.apk' or os.path.splitext(filename)[1] == '.obb':
-            print(f'[INFO] Extracting assets from {filename}... ', end='')
+            remove_unneeded(['assets'], True)
+            printer(f'Extracting assets from {filename}... ', "info")
             extract_assets(filename)
-            print('Done')
-            print('[INFO] Renaming game assets... ', end='')
+            printer('Renaming game assets... ', "info")
             rename_files('assets')
             rename_dirs('assets')
-            print('Done')
-            print('[INFO] Removing unneeded files... ', end='')
-            try:
-                shutil.rmtree('assets/renpy')
-            except FileNotFoundError:
-                renpy_warn = 1
+            printer('Removing unneeded files... ', "info")
             if os.path.splitext(filename)[1] == '.apk':
-                shutil.rmtree('assets/res')
-            print('Done')
-            if renpy_warn:
-                print("[WARN] File does not contain renpy folder!")
-            print('[INFO] Renaming directory... ', end='')
-            os.rename('assets', f'{os.path.splitext(filename)[0]}')
-            print('Done')
+                remove_unneeded(['assets/renpy', 'assets/res', 'assets/dexopt'], False)
+            printer('Renaming directory... ', "info")
+            remove_unneeded([os.path.splitext(filename)[0]], True)
+            os.rename('assets', os.path.splitext(filename)[0])