From 260ca14b7007ff047093ed51d9a0d84c8ffd5e0a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Rapha=C3=ABl=20Vinot?= Date: Thu, 16 Mar 2023 16:47:37 +0100 Subject: [PATCH] fix: Avoid stacktrace on partially broken capture. --- website/web/__init__.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/website/web/__init__.py b/website/web/__init__.py index 18688eb..3971cf1 100644 --- a/website/web/__init__.py +++ b/website/web/__init__.py @@ -561,8 +561,12 @@ def export(tree_uuid: str): @app.route('/tree//urls_rendered_page', methods=['GET']) def urls_rendered_page(tree_uuid: str): - urls = lookyloo.get_urls_rendered_page(tree_uuid) - return render_template('urls_rendered.html', base_tree_uuid=tree_uuid, urls=urls) + try: + urls = lookyloo.get_urls_rendered_page(tree_uuid) + return render_template('urls_rendered.html', base_tree_uuid=tree_uuid, urls=urls) + except Exception: + flash('Unable to find the rendered node in this capture, cannot get the URLs.', 'error') + return redirect(url_for('tree', tree_uuid=tree_uuid)) @app.route('/tree//hashlookup', methods=['GET'])