diff --git a/website/web/__init__.py b/website/web/__init__.py index bebcb2c2..a5b7f175 100644 --- a/website/web/__init__.py +++ b/website/web/__init__.py @@ -377,7 +377,11 @@ def redirects(tree_uuid: str): @app.route('/tree//image', methods=['GET']) def image(tree_uuid: str): - to_return = lookyloo.get_screenshot(tree_uuid) + max_width = request.args.get('width') + if max_width: + to_return = lookyloo.get_screenshot_thumbnail(tree_uuid, width=int(max_width)) + else: + to_return = lookyloo.get_screenshot(tree_uuid) return send_file(to_return, mimetype='image/png', as_attachment=True, attachment_filename='image.png') diff --git a/website/web/templates/tree.html b/website/web/templates/tree.html index 6bedb9a6..60c3d5c4 100644 --- a/website/web/templates/tree.html +++ b/website/web/templates/tree.html @@ -314,7 +314,7 @@