From 1f809233f650a7fb6528fdec525fca49df415e3d Mon Sep 17 00:00:00 2001 From: Jan Szejko Date: Tue, 12 Jun 2018 12:37:21 +0200 Subject: [PATCH] allow custom scheme for redirect to app --- src/api/piston_patch.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/api/piston_patch.py b/src/api/piston_patch.py index 4bf9b615d..fb3987bde 100644 --- a/src/api/piston_patch.py +++ b/src/api/piston_patch.py @@ -15,6 +15,10 @@ from piston import oauth from piston.authentication import initialize_server_request, INVALID_PARAMS_RESPONSE, send_oauth_error +class HttpResponseAppRedirect(HttpResponseRedirect): + allowed_schemes = HttpResponseRedirect.allowed_schemes + ['wolnelekturyapp'] + + class OAuthAuthenticationForm(forms.Form): oauth_token = forms.CharField(widget=forms.HiddenInput) oauth_callback = forms.CharField(widget=forms.HiddenInput) # changed from URLField - too strict @@ -52,6 +56,7 @@ class OAuthAuthenticationForm(forms.Form): # The only thing changed in the views below is the form used +# and also the Http Redirect class def oauth_auth_view(request, token, callback, params): @@ -102,7 +107,7 @@ def oauth_user_auth(request): callback = getattr(settings, 'OAUTH_CALLBACK_VIEW') return get_callable(callback)(request, token) - response = HttpResponseRedirect(callback + args) + response = HttpResponseAppRedirect(callback + args) except oauth.OAuthError, err: response = send_oauth_error(err) -- 2.20.1