file-looping.py (improved)

import os, re, time, json, pathlib, zipfile

path = pathlib.PurePath(os.getcwd())

if path.name == 'Blockland' and 'Add-Ons' in os.listdir():
    addon_path = path / 'Add-Ons'
else:
    addon_path = path

regex = re.compile(b'\s*function serverCmd([\w\d]+)\s*\(([%\w\d,\s]+)*\)', re.I)
results = {}


start = time.clock()

addon_list = [f for f in os.listdir(str(addon_path)) if f[-3:] == 'zip']

for zip_file_name in addon_list:
    zip = zipfile.ZipFile(str(addon_path / zip_file_name))
    file_list = zip.namelist()
    results[zip_file_name[:-4]] = {}
    for file_name in file_list:
        if file_name[-2:] == 'cs':
            file = zip.open(file_name)
            results[zip_file_name[:-4]][file_name[:-3]] = {}
            for line in file:
                match = regex.match(line)
                if match:
                    results[zip_file_name[:-4]][file_name[:-3]][match.group(1).decode()] = [m.strip() for m in match.group(2).decode().split(',')]
            if not results[zip_file_name[:-4]][file_name[:-3]]:
                del results[zip_file_name[:-4]][file_name[:-3]]
            file.close()
    if not results[zip_file_name[:-4]]:
        del results[zip_file_name[:-4]]
    zip.close()

end = time.clock()

serv_cmd_json = open('server_commands.json', 'w+')
json.dump(results, serv_cmd_json, sort_keys=True, indent=4)
serv_cmd_json.close()

print('Results can be found in server_commands.json.')
print('Time taken: {0}'.format(end - start))
input('Press enter to exit.')