diff --git a/ape/routes.py b/ape/routes.py index 359b6ab..2e2649f 100644 --- a/ape/routes.py +++ b/ape/routes.py @@ -1,6 +1,6 @@ import logging -from urllib.parse import quote_plus, urlencode +from urllib.parse import quote_plus, urlencode, urlparse from authlib.integrations.flask_client import OAuth from flask import redirect, render_template, session, url_for, \ @@ -20,7 +20,9 @@ def home(): return_url = request.args.get("return_url", None) lang = request.args.get("lang", None) if return_url: - session["return_url"] = return_url + parsed_url = urlparse(return_url) + session['return_url'] = parsed_url.scheme + "://" + \ + parsed_url.netloc + "/ape_data_receiver" if lang and lang in current_app.config['LANGUAGES']: session["lang"] = lang if session.get("user_id", ""):