ChecksumCheck.py

import os
import hashlib
import sys
import json

version = "0.0.1"
author = "@AdSanz_IT"

#dictionaries that will hold the file with his checksum
dict_hashes = {}
dict_hashes1 = {}
#file where the hashes will be saved
filename = 'hash.txt'
#source admits a list of dicrectories, remember if the path has "\" it must be specified like C:\\System\\... Because in python backslash is an escape character ej -> "source = ['E:\\ASIR2\\ACTUALIZADO\\ASOE104','E:\\ASIR2\\ACTUALIZADO\\ASOE103']"
source = []
#source = ''


def help_s():
 print("This script is intended to check the integrity of files located in the \"source\" recursively. It saves the checksums on {}, and later on you can compare them overtime, it also checks whether a new file was created.\n\nThe tool itself is to check whether a file has been modified by any means.\n\n Usage = \n ChecksumChecker.py 1/2 \n\n Options: \n--> 1 = Get checksums on source and save it on {} \n--> 2 = Compare the checksums on source with checksums saved on {} \n\nRemember: whenever you change the source variable, you should use (1) to save the checksum of those files before checking!!!".format(filename,filename,filename))

def get_checksums_list():
 # Use this def if "source" is a list of dicrectories
 for items in source:
 for subdirs,dirs,files in os.walk(items):
 for name in files:
 filename1 = os.path.join(subdirs,name)
 with open(filename1, 'rb') as f:
 hasher = hashlib.md5()
 while True:
 data = f.read(65536)
 if not data:
 break
 hasher.update(data)
 hex_val = hasher.hexdigest()
 dict_hashes.update({filename1:hex_val})

def get_checksums():
 # Use this def is "source" is only a directory and not a list
 for subdirs,dirs,files in os.walk(source):
 for name in files:
 filename1 = os.path.join(subdirs,name)
 with open(filename1, 'rb') as f:
 hasher = hashlib.md5()
 while True:
 data = f.read(65536)
 if not data:
 break
 hasher.update(data)
 hex_val = hasher.hexdigest()
 dict_hashes.update({filename1:hex_val})

def get_checksums1():
 # This def get the checksums of the file and saves it on a dictionary
 with open(filename, 'r+') as output:
 dict_hashes1.update(json.load(output))


def create():
 # Save checksums (from get_checksums / get_checksums_list) on the file
 with open(filename, 'w+') as checksums:
 checksums.write(json.dumps(dict_hashes))

def compare_checksums():

 for key,value in zip(dict_hashes.keys(),dict_hashes.values()):
 if key in dict_hashes1.keys():
 if value in dict_hashes1.values():
 print("Checked: {}".format(key))
 print("Integrity OK")
 elif value not in dict_hashes1.values():
 print("Checked: {}".format(key))
 print("New checksum -> \"{}\" | checksum on {} -> \"{}\"".format(value,filename,dict_hashes1[key]))
 print("Integrity not OK")
 elif key not in dict_hashes1.keys():
 print("The file {} is not on {}, not checking".format(key,filename))
 pass

if len(sys.argv) < 2:
 help_s()
elif len(sys.argv) == 2 and len(sys.argv) < 3:
 print(" Script made by {}: https://github.com/adsanz/Python-scripts \n Version: {}\n\n".format(author,version))
 if sys.argv[1] == "1":
 try:
 get_checksums()
 except:
 get_checksums_list()
 create()
 if os.path.isfile(filename):
 print("Saved checksums on {}".format(filename))
 else:
 print("Created file {} and saved checksums".format(filename))
 elif sys.argv[1] == "2":
 try:
 get_checksums()
 except:
 get_checksums_list()
 get_checksums1()
 compare_checksums()
 print("Done...")
 else:
 print("Choose a value between 1 and 2\n")
 help_s()
else:
 print("This script only takes 1 argument between 1 and 2\n")
 help_s()