69 lines
1.9 KiB
Python
69 lines
1.9 KiB
Python
|
#!/usr/bin/python
|
||
|
|
# Kevin Rojas 2018
|
||
|
|
|
||
|
|
import os
|
||
|
|
import glob
|
||
|
|
import hashlib
|
||
|
|
import argparse
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
global args
|
||
|
|
global confdir
|
||
|
|
global md5dir
|
||
|
|
global agents
|
||
|
|
|
||
|
|
# Argument parser
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
parser.add_argument('-r', '--run', required=False, dest='run', action='store_true',
|
||
|
|
help='run the tool to recreate md5 files')
|
||
|
|
parser.add_argument('-d', dest='dir', default='/var/spool/pandora/data_in',
|
||
|
|
help='data_in folder path (default /var/spool/pandora/data_in)')
|
||
|
|
parser.add_argument('-v', dest='verb', action='store_true',
|
||
|
|
help='verbose mode: Shows the files being updated')
|
||
|
|
|
||
|
|
# Definitions
|
||
|
|
args = vars(parser.parse_args())
|
||
|
|
datadir = args['dir']
|
||
|
|
confdir = datadir + '/conf/'
|
||
|
|
md5dir = datadir + '/md5/'
|
||
|
|
agents = glob.glob(confdir + '*.conf') # Check folder for .conf files
|
||
|
|
|
||
|
|
# Run script or show help
|
||
|
|
if len(args) > 0 and args['run']:
|
||
|
|
updatemd5()
|
||
|
|
else:
|
||
|
|
parser.print_help()
|
||
|
|
|
||
|
|
|
||
|
|
def md5sum(filename, blocksize=65536):
|
||
|
|
# Open files and calculate MD5 from its content
|
||
|
|
hash = hashlib.md5()
|
||
|
|
with open(filename, 'rb') as f:
|
||
|
|
for block in iter(lambda: f.read(blocksize), b''):
|
||
|
|
hash.update(block)
|
||
|
|
return hash.hexdigest()
|
||
|
|
|
||
|
|
|
||
|
|
def updatemd5():
|
||
|
|
debug = args['verb']
|
||
|
|
if not agents:
|
||
|
|
print(' ERROR: There are no .conf files at ' +
|
||
|
|
confdir + '. Please check the path provided.')
|
||
|
|
else:
|
||
|
|
count = 0
|
||
|
|
for i in agents:
|
||
|
|
agentmd5 = md5dir + os.path.basename(os.path.splitext(i)[0]) + '.md5'
|
||
|
|
with open(agentmd5, 'w') as f:
|
||
|
|
f.write(md5sum(i))
|
||
|
|
f.close()
|
||
|
|
count += 1
|
||
|
|
|
||
|
|
if debug:
|
||
|
|
print(os.path.basename(os.path.splitext(i)[0]) + '--> OK')
|
||
|
|
|
||
|
|
print('Number of configuration files updated: ' + str(count))
|
||
|
|
|
||
|
|
|
||
|
|
main()
|