Created
March 2, 2022 03:56
-
-
Save johnpili/7da5a32fb413717650333a04231093a1 to your computer and use it in GitHub Desktop.
Python batch file MD5 checksum generator and checker
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import hashlib | |
| import os | |
| import sys | |
| def generate_md5(file_path): | |
| with open(file_path.strip(), "rb") as f: | |
| v = f.read() | |
| x = hashlib.new("md5", v) | |
| return x.hexdigest() | |
| def validate(md5_digest, file_path): | |
| return generate_md5(file_path) == md5_digest | |
| def run(argv): | |
| if len(argv) > 0: | |
| if argv[0].lower() == "generate": | |
| with open("checksum.md5", "w") as checksum_file: | |
| for root, dirs, files, root_fd in os.fwalk( | |
| "."): # start walking in current path or directory, you can parameterize this if you want | |
| for file in files: | |
| md5_digest = generate_md5(f"{root}/{file}") | |
| print(f"{md5_digest}\t{root}/{file}") | |
| checksum_file.write(f"{md5_digest}\t{root}/{file}\n") | |
| checksum_file.flush() | |
| elif argv[0].lower() == "check": | |
| stats = {"valid": 0, "not_valid": 0} | |
| with open("checksum.md5", "r") as checksum_file: | |
| s = checksum_file.readline() | |
| while s: | |
| v = s.split("\t") | |
| validation_result = validate(v[0], v[1]) | |
| print(f"{'Ok' if validation_result else 'Not Okay'} = {v[1]}", end="") | |
| if validation_result: | |
| stats["valid"] = stats["valid"] + 1 | |
| else: | |
| stats["not_valid"] = stats["not_valid"] + 1 | |
| s = checksum_file.readline() | |
| print(stats) | |
| if __name__ == '__main__': | |
| run(sys.argv[1:]) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment