import zipfile
import os
def zip_files(report_dir):
zipf = zipfile.ZipFile('report.zip', 'w', zipfile.ZIP_STORED)
rep = ['1', '2', 'new']
print('RRREP ', rep)
for root, dirs, files in os.walk(report_dir):
for file in files:
title = file.title().split('_')[0]
if title in rep:
zipf.write(os.path.join(root, file))
zipf.close()
zip_files('')
at_least_one_match = False
for root, dirs, files in os.walk(report_dir):
for file in files:
title = file.title().split('_')[0]
if title in rep:
at_least_one_match = True
zipf.write(os.path.join(root, file))
if not at_least_one_match:
raise Exception("No files were added. Archive is empty/corrupt")