ADX-1021 edited return url

fjelltopp · Jul 20, 2023 · e4b7675 · e4b7675
1 parent 1af74e4
commit e4b7675
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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", ""):