CIRCL
/
lookyloo
mirror of https://github.com/CIRCL/lookyloo
2
0
Fork 0
Code Issues Releases Wiki Activity

Cleanup HAR_DIR if a scrap failed

pull/1/head
Raphaël Vinot 2017-08-12 19:19:24 +02:00
parent e9903f421c
commit aa659234bb
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
lookyloo/__init__.py
View File

@ -57,9 +57,17 @@ def scrap():
return render_template('scrap.html') return render_template('scrap.html')
def get_report_dirs():
# Cleanup HAR_DIR of failed runs.
for report_dir in os.listdir(HAR_DIR):
if not os.listdir(os.path.join(HAR_DIR, report_dir)):
os.rmdir(os.path.join(HAR_DIR, report_dir))
return sorted(os.listdir(HAR_DIR))
@app.route('/tree/<int:tree_id>', methods=['GET']) @app.route('/tree/<int:tree_id>', methods=['GET'])
def tree(tree_id): def tree(tree_id):
report_dir = sorted(os.listdir(HAR_DIR))[tree_id] report_dir = get_report_dirs()[tree_id]
tree = load_tree(report_dir) tree = load_tree(report_dir)
nodes, faces, base64 = tree.redraw() nodes, faces, base64 = tree.redraw()
return render_template('tree.html', nodes=nodes, faces=faces, base64_img=base64) return render_template('tree.html', nodes=nodes, faces=faces, base64_img=base64)
@ -71,7 +79,7 @@ def index():
titles = [] titles = []
if not os.path.exists(HAR_DIR): if not os.path.exists(HAR_DIR):
os.makedirs(HAR_DIR) os.makedirs(HAR_DIR)
for report_dir in sorted(os.listdir(HAR_DIR)): for report_dir in get_report_dirs():
har_files = sorted(glob(os.path.join(HAR_DIR, report_dir, '*.har'))) har_files = sorted(glob(os.path.join(HAR_DIR, report_dir, '*.har')))
if not har_files: if not har_files:
continue continue