106 lines
2.8 KiB
Python
106 lines
2.8 KiB
Python
from collections import defaultdict
|
|
|
|
from authlib.common.encoding import json_loads
|
|
from authlib.common.urls import url_decode
|
|
from authlib.common.urls import urlparse
|
|
|
|
from .errors import InsecureTransportError
|
|
|
|
|
|
class OAuth2Request:
|
|
def __init__(self, method: str, uri: str, body=None, headers=None):
|
|
InsecureTransportError.check(uri)
|
|
#: HTTP method
|
|
self.method = method
|
|
self.uri = uri
|
|
self.body = body
|
|
#: HTTP headers
|
|
self.headers = headers or {}
|
|
|
|
self.client = None
|
|
self.auth_method = None
|
|
self.user = None
|
|
self.authorization_code = None
|
|
self.refresh_token = None
|
|
self.credential = None
|
|
|
|
self._parsed_query = None
|
|
|
|
@property
|
|
def args(self):
|
|
if self._parsed_query is None:
|
|
self._parsed_query = url_decode(urlparse.urlparse(self.uri).query)
|
|
return dict(self._parsed_query)
|
|
|
|
@property
|
|
def form(self):
|
|
return self.body or {}
|
|
|
|
@property
|
|
def data(self):
|
|
data = {}
|
|
data.update(self.args)
|
|
data.update(self.form)
|
|
return data
|
|
|
|
@property
|
|
def datalist(self) -> defaultdict[str, list]:
|
|
"""Return all the data in query parameters and the body of the request as a dictionary
|
|
with all the values in lists.
|
|
"""
|
|
if self._parsed_query is None:
|
|
self._parsed_query = url_decode(urlparse.urlparse(self.uri).query)
|
|
values = defaultdict(list)
|
|
for k, v in self._parsed_query:
|
|
values[k].append(v)
|
|
for k, v in self.form.items():
|
|
values[k].append(v)
|
|
return values
|
|
|
|
@property
|
|
def client_id(self) -> str:
|
|
"""The authorization server issues the registered client a client
|
|
identifier -- a unique string representing the registration
|
|
information provided by the client. The value is extracted from
|
|
request.
|
|
|
|
:return: string
|
|
"""
|
|
return self.data.get("client_id")
|
|
|
|
@property
|
|
def response_type(self) -> str:
|
|
rt = self.data.get("response_type")
|
|
if rt and " " in rt:
|
|
# sort multiple response types
|
|
return " ".join(sorted(rt.split()))
|
|
return rt
|
|
|
|
@property
|
|
def grant_type(self) -> str:
|
|
return self.form.get("grant_type")
|
|
|
|
@property
|
|
def redirect_uri(self):
|
|
return self.data.get("redirect_uri")
|
|
|
|
@property
|
|
def scope(self) -> str:
|
|
return self.data.get("scope")
|
|
|
|
@property
|
|
def state(self):
|
|
return self.data.get("state")
|
|
|
|
|
|
class JsonRequest:
|
|
def __init__(self, method, uri, body=None, headers=None):
|
|
self.method = method
|
|
self.uri = uri
|
|
self.body = body
|
|
self.headers = headers or {}
|
|
|
|
@property
|
|
def data(self):
|
|
return json_loads(self.body)
|