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

Update afstest.py

Browse Source
This commit is contained in:
tmpz23 2022-01-18 07:51:08 +01:00 committed by GitHub
parent 279f2e914b
commit 9750bf94c4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 27 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
afstest.py
View File

@ -5,7 +5,7 @@ import os
import shutil import shutil
__version__ = "0.0.1" __version__ = "0.0.2"
__author__ = "rigodron, algoflash, GGLinnk" __author__ = "rigodron, algoflash, GGLinnk"
__license__ = "MIT" __license__ = "MIT"
__status__ = "developpement" __status__ = "developpement"
@ -62,15 +62,16 @@ def verify_AFS_sha256(folder1: Path, folder2: Path):
raise Exception(f"\"{path1}/\" and \"{path2}\" are different.") raise Exception(f"\"{path1}/\" and \"{path2}\" are different.")
print("# Cleaning tests folder") print("###############################################################################")
# Remove tests folders print("# Checking tests folder")
if unpack_path.is_dir(): print("###############################################################################")
shutil.rmtree(unpack_path) # Check if tests folders exist
if unpack2_path.is_dir(): if unpack_path.is_dir() or unpack2_path.is_dir() or repack_path.is_dir():
shutil.rmtree(unpack2_path) raise Exception(f"Error - Please remove:\n-{unpack_path}\n-{unpack2_path}\n{repack_path}")
if repack_path.is_dir():
shutil.rmtree(repack_path) print("###############################################################################")
print("# Comparing unpacked AFS with AFSPacker unpacks.") print("# Comparing unpacked AFS with AFSPacker unpacks.")
print("###############################################################################")
unpack_path.mkdir() unpack_path.mkdir()
if not afspacker_unpack_path.is_dir(): if not afspacker_unpack_path.is_dir():
afspacker_unpack_path.mkdir() afspacker_unpack_path.mkdir()
@ -90,7 +91,9 @@ for folder_path in unpack_path.glob("*"):
if folder_path.is_dir(): if folder_path.is_dir():
verify_AFS_sha256(folder_path/"root", afspacker_unpack_path / folder_path.stem) verify_AFS_sha256(folder_path/"root", afspacker_unpack_path / folder_path.stem)
print("###############################################################################")
print("# Comparing repacked AFS with originals AFS.") print("# Comparing repacked AFS with originals AFS.")
print("###############################################################################")
repack_path.mkdir() repack_path.mkdir()
# repack unpack_path repack_path # repack unpack_path repack_path
for folder_path in unpack_path.glob("*"): for folder_path in unpack_path.glob("*"):
@ -102,6 +105,18 @@ for afs_path in repack_path.glob("*"):
print(f"compare \"{afs_path}\" - \"{afss_path / Path(folder_path.stem).with_suffix('.afs')}\"") print(f"compare \"{afs_path}\" - \"{afss_path / Path(folder_path.stem).with_suffix('.afs')}\"")
compare_sha256(afs_path, afss_path / Path(folder_path.stem).with_suffix(".afs")) compare_sha256(afs_path, afss_path / Path(folder_path.stem).with_suffix(".afs"))
# Patch unpack files whithout changing their len
# Patch unpack files changing len to max
# Patch unpack files with 1 byte in a new block
# Patch unpack files with 1 block less
print("###############################################################################")
print(f"# Cleaning test folders.")
print("###############################################################################")
# Remove tests folders # Remove tests folders
if unpack_path.is_dir(): if unpack_path.is_dir():
shutil.rmtree(unpack_path) shutil.rmtree(unpack_path)
@ -109,4 +124,7 @@ if unpack2_path.is_dir():
shutil.rmtree(unpack2_path) shutil.rmtree(unpack2_path)
if repack_path.is_dir(): if repack_path.is_dir():
shutil.rmtree(repack_path) shutil.rmtree(repack_path)
print("###############################################################################")
print("# All tests are OK") print("# All tests are OK")
print("###############################################################################")