From: Jan Szejko Date: Tue, 12 Jun 2018 10:37:21 +0000 (+0200) Subject: allow custom scheme for redirect to app X-Git-Url: https://git.mdrn.pl/wolnelektury.git/commitdiff_plain/1f809233f650a7fb6528fdec525fca49df415e3d allow custom scheme for redirect to app --- 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)