diff --git a/pzztest.py b/pzztest.py new file mode 100644 index 0000000..d496d85 --- /dev/null +++ b/pzztest.py @@ -0,0 +1,41 @@ +import argparse +import pzztool +import hashlib +import os + + +def argparser(): + parser = argparse.ArgumentParser(description='TEST TOOL') + parser.add_argument('input_path', metavar='INPUT', help='') + return parser.parse_args() + + +if __name__ == '__main__': + listofinvalid = [] + args = argparser() + + for filename in os.listdir(args.input_path): + file = open(os.path.join(args.input_path, filename), 'rb') + original_bytes = file.read() + decomp_bytes = pzztool.pzz_decompress(original_bytes) + recomp_bytes = pzztool.pzz_compress(decomp_bytes) + + original_digest = hashlib.sha256(original_bytes).hexdigest() + recomp_digest = hashlib.sha256(recomp_bytes).hexdigest() + + if original_digest != recomp_digest: + listofinvalid.append(f"{filename} : ({original_digest}) ({recomp_digest})") + file.close() + + for invalid in listofinvalid: + print(invalid) + + """ + Code pour le developement --> pzztool.py -a a + compare le sha256 de chaque PZZ du dossier pzz et pzz2 puis affiche le nom de fichier en cas de différence + import hashlib + for pzz_file in listdir("pzz"): + with open("pzz/"+pzz_file, "rb") as f1, open("pzz2/"+pzz_file, "rb") as f2: + if hashlib.sha256( f1.read() ).hexdigest() != hashlib.sha256( f2.read() ).hexdigest() : + print(pzz_file) + """