RenPy-Android-Unpack: Some improvements

This commit is contained in:
OleSTEEP 2024-02-03 22:45:47 +03:00
parent 248f08c7d9
commit 20f86f7c0e

View file

@ -1,23 +1,42 @@
#!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):
with zipfile.ZipFile(file, 'r') as zip_ref:
for content in zip_ref.namelist():
if content.split('/')[0] == 'assets':
zip_ref.extract(content)
if os.path.splitext(file)[1] == '.apk':
try:
zip_ref.extract('res/mipmap-xxxhdpi-v4/icon_background.png', 'assets')
zip_ref.extract('res/mipmap-xxxhdpi-v4/icon_foreground.png', 'assets')
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:
zip_ref.extract('res/drawable/icon.png', 'assets')
os.rename('assets/res/drawable/icon.png', 'assets/icon.png')
try:
with zipfile.ZipFile(file, 'r') as zip_ref:
for content in zip_ref.namelist():
if content.split('/')[0] == 'assets':
zip_ref.extract(content)
if os.path.splitext(file)[1] == '.apk':
try:
zip_ref.extract('res/mipmap-xxxhdpi-v4/icon_background.png', 'assets')
zip_ref.extract('res/mipmap-xxxhdpi-v4/icon_foreground.png', 'assets')
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])