def main():
print('\n\tFiles containing {}'.format(args.text))
print('\t-------------------\n')
files_searched = 0
file_matches = 0
# traverse the file tree and find files with .dotm extension
for path, folders, files in os.walk(args.folder):
for file in files:
files_searched += 1
# find files with .dotm file extensions
if '.dotm' in file:
file_path = os.path.join(path, file)
document = ZipFile(file_path, 'r').read('word/document.xml')
byte_string = args.text.encode('utf-8')
# find files with arg text in them
if byte_string in document:
text_index = document.index(byte_string)
sample_text = document[text_index - 40:text_index + 41]
# print file information
print('\tFile Path:\t{}'.format(file_path))
print('\tSample:\t...{}...\n'.format(sample_text))
print('\t*****************\n')
file_matches += 1
# print scraping report
print('\tFiles Found:\t{}'.format(str(file_matches)))
print('\tFiles Searched:\t{}\n'.format(files_searched))
