import os
import sysdef decrypt_rpgmvp(input_file, output_file):# PNG文件的文件头png_header = b'\x89\x50\x4E\x47\x0D\x0A\x1A\x0A\x00\x00\x00\x0D\x49\x48\x44\x52'# 读取输入文件的内容with open(input_file, 'rb') as f_in:data = f_in.read()# 写入输出文件,包括PNG文件头和原始数据(跳过前32个字节)with open(output_file, 'wb') as f_out:f_out.write(png_header)f_out.write(data[32:])def decrypt_directory(directory):# 获取输入目录的基本名称base_dir = os.path.basename(directory)# 在当前工作目录下创建同名的输出目录output_dir = os.path.join(os.getcwd(), base_dir)# 遍历输入目录下的所有文件和子目录for root, dirs, files in os.walk(directory):for file in files:# 如果文件扩展名为.rpgmvp,则进行解密if file.endswith('.rpgmvp'):# 构建输入文件的完整路径input_file = os.path.join(root, file)# 获取输入文件相对于输入目录的相对路径relative_path = os.path.relpath(input_file, directory)# 获取输出文件的子目录路径output_subdir = os.path.dirname(relative_path)# 构建输出文件的文件名(将扩展名改为.png)output_file = os.path.splitext(relative_path)[0] + '.png'# 构建输出文件的完整目录路径output_path = os.path.join(output_dir, output_subdir)# 创建输出文件的目录(如果不存在)os.makedirs(output_path, exist_ok=True)# 构建输出文件的完整路径output_file_path = os.path.join(output_path, os.path.basename(output_file))# 调用decrypt_rpgmvp函数解密单个文件
decrypt_rpgmvp(input_file, output_file_path)if __name__ == '__main__':# 如果命令行参数为3个,则解密单个文件if len(sys.argv) == 3:input_file = sys.argv[1]output_file = sys.argv[2]decrypt_rpgmvp(input_file, output_file)# 如果命令行参数为2个,则解密整个目录elif len(sys.argv) == 2:path = sys.argv[1]if os.path.exists(path):if os.path.isfile(path):print(f"Error: '{path}' is a file, not a directory.")else:decrypt_directory(path)else:print(f"Error: Path '{path}' does not exist.")# 如果命令行参数不正确,则打印使用说明else:print("Usage:")print(" For single file: python decrypt_rpgmvp.py input.rpgmvp output.png")print(" For entire directory: python decrypt_rpgmvp.py directory_path")
