file-looping.py

import os, re, time

p_dir = os.getcwd()

regex = re.compile('[ \t]*(mass ?= ?[0-9]+);')
results = {}


start = time.clock()

file_list = os.listdir(p_dir)

for filename in file_list:
    if filename != os.path.basename(__file__):
        file = open(p_dir + '/' + filename)
        results[filename] = []
        for line in file:
            match = regex.match(line)
            if match:
                results[filename].append(match.group(1))
        if not results[filename]:
            del results[filename]
        file.close()

end = time.clock()


print('Result list: {0}'.format(results))
print('Time taken: {0}'.format(end - start))
input('Press enter to exit.')