From e213cb5ae86d70cbc226d01960e3833765cd3493 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Rapha=C3=ABl=20Vinot?= Date: Fri, 3 Sep 2021 17:05:26 +0200 Subject: [PATCH] fix: make mypy happy. --- website/web/__init__.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/website/web/__init__.py b/website/web/__init__.py index 223e1ad0..aa38e3e0 100644 --- a/website/web/__init__.py +++ b/website/web/__init__.py @@ -754,7 +754,7 @@ def search(): return render_template('search.html') -def _prepare_capture_template(user_ua: str, predefined_url: Optional[str]=None): +def _prepare_capture_template(user_ua: Optional[str], predefined_url: Optional[str]=None): user_agents: Dict[str, Any] = {} if use_own_ua: user_agents = get_user_agents('own_user_agents') @@ -777,7 +777,10 @@ def _prepare_capture_template(user_ua: str, predefined_url: Optional[str]=None): @app.route('/recapture/', methods=['GET']) def recapture(tree_uuid: str): cache = lookyloo.capture_cache(tree_uuid) - return _prepare_capture_template(user_ua=request.headers.get('User-Agent'), predefined_url=cache.url) + if cache: + return _prepare_capture_template(user_ua=request.headers.get('User-Agent'), predefined_url=cache.url) + flash(f'Unable to find the capture {tree_uuid} in the cache.', 'error') + return _prepare_capture_template(user_ua=request.headers.get('User-Agent')) @app.route('/capture', methods=['GET', 'POST'])