From: Jan Szejko <janek37@gmail.com>
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)