44from urllib .parse import urlencode
55
66from plain .auth import login as auth_login
7- from plain .http import HttpRequest , Response , ResponseRedirect
7+ from plain .http import Request , Response , ResponseRedirect
88from plain .runtime import settings
99from plain .urls import reverse
1010from plain .utils .cache import add_never_cache_headers
@@ -65,7 +65,7 @@ def __init__(
6565 self .client_secret = client_secret
6666 self .scope = scope
6767
68- def get_authorization_url_params (self , * , request : HttpRequest ) -> dict :
68+ def get_authorization_url_params (self , * , request : Request ) -> dict :
6969 return {
7070 "redirect_uri" : self .get_callback_url (request = request ),
7171 "client_id" : self .get_client_id (),
@@ -77,13 +77,13 @@ def get_authorization_url_params(self, *, request: HttpRequest) -> dict:
7777 def refresh_oauth_token (self , * , oauth_token : OAuthToken ) -> OAuthToken :
7878 raise NotImplementedError ()
7979
80- def get_oauth_token (self , * , code : str , request : HttpRequest ) -> OAuthToken :
80+ def get_oauth_token (self , * , code : str , request : Request ) -> OAuthToken :
8181 raise NotImplementedError ()
8282
8383 def get_oauth_user (self , * , oauth_token : OAuthToken ) -> OAuthUser :
8484 raise NotImplementedError ()
8585
86- def get_authorization_url (self , * , request : HttpRequest ) -> str :
86+ def get_authorization_url (self , * , request : Request ) -> str :
8787 return self .authorization_url
8888
8989 def get_client_id (self ) -> str :
@@ -95,14 +95,14 @@ def get_client_secret(self) -> str:
9595 def get_scope (self ) -> str :
9696 return self .scope
9797
98- def get_callback_url (self , * , request : HttpRequest ) -> str :
98+ def get_callback_url (self , * , request : Request ) -> str :
9999 url = reverse ("oauth:callback" , provider = self .provider_key )
100100 return request .build_absolute_uri (url )
101101
102102 def generate_state (self ) -> str :
103103 return get_random_string (length = 32 )
104104
105- def check_request_state (self , * , request : HttpRequest ) -> None :
105+ def check_request_state (self , * , request : Request ) -> None :
106106 if error := request .query_params .get ("error" ):
107107 raise OAuthError (error )
108108
@@ -117,7 +117,7 @@ def check_request_state(self, *, request: HttpRequest) -> None:
117117 raise OAuthStateMismatchError ()
118118
119119 def handle_login_request (
120- self , * , request : HttpRequest , redirect_to : str = ""
120+ self , * , request : Request , redirect_to : str = ""
121121 ) -> Response :
122122 authorization_url = self .get_authorization_url (request = request )
123123 authorization_params = self .get_authorization_url_params (request = request )
@@ -138,11 +138,11 @@ def handle_login_request(
138138 return self .get_redirect_response (redirect_url )
139139
140140 def handle_connect_request (
141- self , * , request : HttpRequest , redirect_to : str = ""
141+ self , * , request : Request , redirect_to : str = ""
142142 ) -> Response :
143143 return self .handle_login_request (request = request , redirect_to = redirect_to )
144144
145- def handle_disconnect_request (self , * , request : HttpRequest ) -> Response :
145+ def handle_disconnect_request (self , * , request : Request ) -> Response :
146146 provider_user_id = request .data ["provider_user_id" ]
147147 connection = OAuthConnection .query .get (
148148 provider_key = self .provider_key , provider_user_id = provider_user_id
@@ -151,7 +151,7 @@ def handle_disconnect_request(self, *, request: HttpRequest) -> Response:
151151 redirect_url = self .get_disconnect_redirect_url (request = request )
152152 return self .get_redirect_response (redirect_url )
153153
154- def handle_callback_request (self , * , request : HttpRequest ) -> Response :
154+ def handle_callback_request (self , * , request : Request ) -> Response :
155155 self .check_request_state (request = request )
156156
157157 oauth_token = self .get_oauth_token (
@@ -181,13 +181,13 @@ def handle_callback_request(self, *, request: HttpRequest) -> Response:
181181 redirect_url = self .get_login_redirect_url (request = request )
182182 return self .get_redirect_response (redirect_url )
183183
184- def login (self , * , request : HttpRequest , user : Any ) -> None :
184+ def login (self , * , request : Request , user : Any ) -> None :
185185 auth_login (request = request , user = user )
186186
187- def get_login_redirect_url (self , * , request : HttpRequest ) -> str :
187+ def get_login_redirect_url (self , * , request : Request ) -> str :
188188 return request .session .pop (SESSION_NEXT_KEY , "/" )
189189
190- def get_disconnect_redirect_url (self , * , request : HttpRequest ) -> str :
190+ def get_disconnect_redirect_url (self , * , request : Request ) -> str :
191191 return request .data .get ("next" , "/" )
192192
193193 def get_redirect_response (self , redirect_url : str ) -> Response :
0 commit comments