diff --git a/moulinette/interfaces/api.py b/moulinette/interfaces/api.py index fdda0334..c73e408e 100644 --- a/moulinette/interfaces/api.py +++ b/moulinette/interfaces/api.py @@ -13,7 +13,7 @@ from gevent.queue import Queue from geventwebsocket import WebSocketError -from bottle import request, response, Bottle, HTTPResponse, FileUpload +from bottle import redirect, request, response, Bottle, HTTPResponse, FileUpload from bottle import abort from moulinette import m18n, Moulinette @@ -381,7 +381,11 @@ def login(self): raise HTTPResponse(e.strerror, 401) else: authenticator.set_session_cookie(auth_infos) - return m18n.g("logged_in") + referer = request.get_header("Referer") + if "referer_redirect" in request.params and referer: + redirect(referer) + else: + return m18n.g("logged_in") # This is called before each time a route is going to be processed def authenticate(self, authenticator): @@ -405,7 +409,11 @@ def logout(self): else: # Delete cookie and clean the session authenticator.delete_session_cookie() - return m18n.g("logged_out") + referer = request.get_header("Referer") + if "referer_redirect" in request.params and referer: + redirect(referer) + else: + return m18n.g("logged_in") def messages(self): """Listen to the messages WebSocket stream