chg: cleanup download of urls in rendered content

fix #148
pull/156/head
Raphaël Vinot 2021-01-11 15:12:44 +01:00
parent 35452cf167
commit f40976eecd
1 changed files with 6 additions and 2 deletions

View File

@ -562,11 +562,15 @@ def urlnode_response_cookies(tree_uuid: str, node_uuid: str):
@app.route('/tree/<string:tree_uuid>/url/<string:node_uuid>/urls_in_rendered_content', methods=['GET']) @app.route('/tree/<string:tree_uuid>/url/<string:node_uuid>/urls_in_rendered_content', methods=['GET'])
def urlnode_urls_in_rendered_content(tree_uuid: str, node_uuid: str): def urlnode_urls_in_rendered_content(tree_uuid: str, node_uuid: str):
urlnode = lookyloo.get_urlnode_from_tree(tree_uuid, node_uuid) ct = lookyloo.get_crawled_tree(tree_uuid)
urlnode = ct.root_hartree.get_url_node_by_uuid(node_uuid)
if not urlnode.rendered_html: if not urlnode.rendered_html:
return return
not_loaded_urls = sorted(set(urlnode.urls_in_rendered_page)
- set(ct.root_hartree.all_url_requests.keys()))
to_return = StringIO() to_return = StringIO()
to_return.writelines([f'{u}\n' for u in urlnode.urls_in_rendered_page]) to_return.writelines([f'{u}\n' for u in not_loaded_urls])
return send_file(BytesIO(to_return.getvalue().encode()), mimetype='text/plain', return send_file(BytesIO(to_return.getvalue().encode()), mimetype='text/plain',
as_attachment=True, attachment_filename='urls_in_rendered_content.txt') as_attachment=True, attachment_filename='urls_in_rendered_content.txt')