Virtual-World-RE/NeoGF
4
0
Fork 0
mirror of https://github.com/Virtual-World-RE/NeoGF.git synced 2024-11-15 15:05:34 +01:00
Code Issues Releases Wiki Activity

Update pzztool.py

Browse Source 
Ajout du logging -> debug avec l'option -v
This commit is contained in:
tmpz23 2021-11-19 15:47:34 +01:00 committed by GitHub
parent 2fc1f210b7
commit 4dc0e35f32
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 35 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

68
pzztool.py
View File

@ -1,8 +1,10 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
from math import ceil from math import ceil
from struct import unpack
from pathlib import Path from pathlib import Path
from os import listdir, path from struct import unpack
from os import listdir
import logging
__version__ = "1.3.9" __version__ = "1.3.9"
__author__ = "rigodron, algoflash, GGLinnk" __author__ = "rigodron, algoflash, GGLinnk"
__OriginalAutor__ = "infval" __OriginalAutor__ = "infval"
@ -37,7 +39,7 @@ def pzz_decompress(compressed_bytes: bytes):
compress_flag = cb & (1 << cb_bit) compress_flag = cb & (1 << cb_bit)
cb_bit -= 1 cb_bit -= 1
# print(compress_flag) # logging.debug(compress_flag)
if compress_flag: if compress_flag:
c = compressed_bytes[i + 1] c = compressed_bytes[i + 1]
c |= compressed_bytes[i + 0] << 8 c |= compressed_bytes[i + 0] << 8
@ -144,12 +146,14 @@ def pzz_compress(b):
def pzz_unpack(pzz_path, dest_folder): def pzz_unpack(pzz_path, dest_folder):
if pzz_path.suffix != ".pzz": if pzz_path.suffix != ".pzz":
print(f"WARNING - Unpack : Invalid file format '{pzz_path.suffix}'; it should be .pzz file format") logging.warning(f"Invalid file format '{pzz_path.suffix}'; it should be .pzz file format")
if dest_folder != Path('.'): if dest_folder != Path('.'):
unpacked_pzz_path = dest_folder unpacked_pzz_path = dest_folder
else: else:
unpacked_pzz_path = pzz_path.parent / pzz_path.stem unpacked_pzz_path = pzz_path.parent / pzz_path.stem
logging.info(f" unpacking {pzz_path} in folder {unpacked_pzz_path}")
unpacked_pzz_path.mkdir(exist_ok=True) unpacked_pzz_path.mkdir(exist_ok=True)
with open(pzz_path, "rb") as pzz_file: with open(pzz_path, "rb") as pzz_file:
@ -160,7 +164,7 @@ def pzz_unpack(pzz_path, dest_folder):
# files_descriptors reçoit un tuple avec l'ensemble des descripteurs de fichiers (groupes d'uint32 big-endian) # files_descriptors reçoit un tuple avec l'ensemble des descripteurs de fichiers (groupes d'uint32 big-endian)
files_descriptors = unpack(f">{file_count}I", pzz_file.read(file_count * 4)) files_descriptors = unpack(f">{file_count}I", pzz_file.read(file_count * 4))
print(f"File count:{file_count}") logging.debug(f"File count : {file_count}")
offset = CHUNK_SIZE offset = CHUNK_SIZE
# on parcours le tuple de descripteurs de fichiers # on parcours le tuple de descripteurs de fichiers
@ -185,7 +189,7 @@ def pzz_unpack(pzz_path, dest_folder):
filename = f"{index:03}{compression_status}_{pzz_path.stem}" filename = f"{index:03}{compression_status}_{pzz_path.stem}"
file_path = (unpacked_pzz_path / filename).with_suffix(".dat") file_path = (unpacked_pzz_path / filename).with_suffix(".dat")
print(f"Offset: {offset:010} - {file_path.stem}") logging.debug(f"Offset: {offset:010} - {file_path.stem}")
# Si la taille est nulle, on créé un fichier vide et on passe au descripteur de fichier suivant # Si la taille est nulle, on créé un fichier vide et on passe au descripteur de fichier suivant
if file_len == 0: if file_len == 0:
@ -210,7 +214,7 @@ def pzz_pack(src_path, dest_file):
src_files = listdir(src_path) src_files = listdir(src_path)
# On récupère le nombre total de fichiers # On récupère le nombre total de fichiers
file_count = int(src_files[-1][0:3]) + 1 file_count = len(src_files)
if dest_file != Path('.'): if dest_file != Path('.'):
if dest_file.suffix != ".pzz": if dest_file.suffix != ".pzz":
@ -218,8 +222,8 @@ def pzz_pack(src_path, dest_file):
pzz_path = dest_file pzz_path = dest_file
else: else:
pzz_path = src_path.with_suffix(".pzz") pzz_path = src_path.with_suffix(".pzz")
logging.info(f" packing {src_path} in pzz {pzz_path}")
print(f"{file_count} files to pack in {pzz_path}") logging.debug(f" -> {file_count} files to pack")
with pzz_path.open("wb") as pzz_file: with pzz_path.open("wb") as pzz_file:
# On écrit file_count au début de header # On écrit file_count au début de header
@ -287,63 +291,61 @@ def get_argparser():
if __name__ == '__main__': if __name__ == '__main__':
logging.basicConfig(format='%(levelname)s:%(message)s', level=logging.INFO)
args = get_argparser().parse_args() args = get_argparser().parse_args()
p_input = Path(args.input_path) p_input = Path(args.input_path)
p_output = Path(args.output_path) p_output = Path(args.output_path)
if args.verbose:
logging.basicConfig(level=logging.debug)
if args.compress: if args.compress:
print("### Compress") logging.info("### Compress")
p_output.write_bytes(pzz_compress(p_input.read_bytes())) p_output.write_bytes(pzz_compress(p_input.read_bytes()))
elif args.decompress: elif args.decompress:
print("### Decompress") logging.info("### Decompress")
p_output.write_bytes(pzz_decompress(p_input.read_bytes())) p_output.write_bytes(pzz_decompress(p_input.read_bytes()))
elif args.batch_compress: elif args.batch_compress:
print("### Batch Compress") logging.info("### Batch Compress")
p_output.mkdir(exist_ok=True) p_output.mkdir(exist_ok=True)
for filename in listdir(p_input): for filename in listdir(p_input):
if (not args.disable_ignore) and not ("_compressed" in filename): if (not args.disable_ignore) and not ("_compressed" in filename):
if args.verbose: logging.debug(f"Compressing {filename}")
print(f"Compressing {filename}") recomp_filename = f"{Path(filename).stem}_compressed{Path(filename).suffix}"
recomp_filename = Path(filename).stem + "_compressed" + Path(filename).suffix
uncompressed = open(path.join(p_input, filename), 'rb') with open(p_input / filename, 'rb') as uncompressed, open(p_output / filename, 'wb') as recompressed:
recompressed = open(path.join(p_output, filename), 'wb') recompressed.write(pzz_compress(uncompressed.read()))
recompressed.write(pzz_compress(uncompressed.read()))
recompressed.close()
uncompressed.close()
else: else:
print(f"Ignored: {filename}") logging.info(f"Ignored: {filename}")
elif args.batch_decompress: elif args.batch_decompress:
print("### Batch Decompress") logging.info("### Batch Decompress")
p_output.mkdir(exist_ok=True) p_output.mkdir(exist_ok=True)
for filename in listdir(p_input): for filename in listdir(p_input):
if (not args.disable_ignore) and ("_compressed" in filename): if (not args.disable_ignore) and ("_compressed" in filename):
print(f"Decompressing {filename}") logging.info(f"Decompressing {filename}")
uncomp_filename = filename.replace("_compressed", "") uncomp_filename = filename.replace("_compressed", "")
compressed = open(path.join(p_input, filename), 'rb') with open(p_output / uncomp_filename, 'wb') as uncompressed, open(p_input / filename, 'rb') as compressed:
uncompressed = open(path.join(p_output, uncomp_filename), 'wb') uncompressed.write(pzz_decompress(compressed.read()))
uncompressed.write(pzz_decompress(compressed.read()))
uncompressed.close()
compressed.close()
else: else:
print(f"Ignored: {filename}") logging.info(f"Ignored: {filename}")
elif args.pack: elif args.pack:
print("### Pack") logging.info("### Pack")
pzz_pack(p_input, p_output) pzz_pack(p_input, p_output)
elif args.unpack: elif args.unpack:
print("### Unpack") logging.info("### Unpack")
pzz_unpack(p_input, p_output) pzz_unpack(p_input, p_output)
elif args.batch_pack: elif args.batch_pack:
print("### Batch Pack") logging.info("### Batch Pack")
p_output.mkdir(exist_ok=True) p_output.mkdir(exist_ok=True)
for folder in listdir(p_input): for folder in listdir(p_input):
pzz_pack(p_input / folder, p_output / Path(folder).with_suffix(".pzz")) pzz_pack(p_input / folder, p_output / Path(folder).with_suffix(".pzz"))
elif args.batch_unpack: elif args.batch_unpack:
print("### Batch Unpack") logging.info("### Batch Unpack")
p_output.mkdir(exist_ok=True) p_output.mkdir(exist_ok=True)
for filename in listdir(p_input): for filename in listdir(p_input):