FreakU
/
takahe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Support query params in redirect URIs

This commit is contained in:
Andrew Godwin 2022-12-19 22:06:40 +00:00
parent c3a60bbf13
commit 5ab5684798
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
api/views/oauth.py
View File

@ -1,5 +1,5 @@
import secrets import secrets
from urllib.parse import urlparse from urllib.parse import urlparse, urlunparse
from django.contrib.auth.mixins import LoginRequiredMixin from django.contrib.auth.mixins import LoginRequiredMixin
from django.http import HttpResponseRedirect, JsonResponse from django.http import HttpResponseRedirect, JsonResponse
@ -13,8 +13,15 @@ from api.parser import FormOrJsonParser
class OauthRedirect(HttpResponseRedirect): class OauthRedirect(HttpResponseRedirect):
def __init__(self, redirect_uri, key, value): def __init__(self, redirect_uri, key, value):
self.allowed_schemes = [urlparse(redirect_uri).scheme] url_parts = urlparse(redirect_uri)
super().__init__(redirect_uri + f"?{key}={value}") self.allowed_schemes = [url_parts.scheme]
# Either add or join the query section
url_parts = list(url_parts)
if url_parts[4]:
url_parts[4] = url_parts[4] + f"&{key}={value}"
else:
url_parts[4] = f"{key}={value}"
super().__init__(urlunparse(url_parts))
class AuthorizationView(LoginRequiredMixin, TemplateView): class AuthorizationView(LoginRequiredMixin, TemplateView):