@@ -234,38 +234,36 @@ def authorize(
234
234
)
235
235
webbrowser .open (authorization_url )
236
236
237
- spinner = yaspin (
237
+ with yaspin (
238
238
text = "Please complete authentication in your browser." ,
239
239
spinner = Spinners .dots ,
240
- )
241
- spinner .start ()
242
-
243
- code_queue : queue .Queue [str ] = queue .Queue ()
244
- server = CallbackHTTPServer (
245
- server_address ,
246
- functools .partial (
247
- CallbackHTTPRequestHandler ,
248
- queue = code_queue ,
249
- render_success_page = self .render_success_page ,
250
- render_error_page = self .render_error_page ,
251
- ),
252
- )
253
-
254
- server .serve_forever ()
255
-
256
- try :
257
- code = code_queue .get (block = False )
258
- except queue .Empty as e :
259
- raise FiefAuthAuthorizationCodeMissingError () from e
260
-
261
- spinner .text = "Getting a token..."
262
-
263
- tokens , userinfo = self .client .auth_callback (
264
- code , redirect_uri , code_verifier = code_verifier
265
- )
266
- self ._save_credentials (tokens , userinfo )
267
-
268
- spinner .ok ("Successfully authenticated" )
240
+ ) as spinner :
241
+ code_queue : queue .Queue [str ] = queue .Queue ()
242
+ server = CallbackHTTPServer (
243
+ server_address ,
244
+ functools .partial (
245
+ CallbackHTTPRequestHandler ,
246
+ queue = code_queue ,
247
+ render_success_page = self .render_success_page ,
248
+ render_error_page = self .render_error_page ,
249
+ ),
250
+ )
251
+
252
+ server .serve_forever ()
253
+
254
+ try :
255
+ code = code_queue .get (block = False )
256
+ except queue .Empty as e :
257
+ raise FiefAuthAuthorizationCodeMissingError () from e
258
+
259
+ spinner .text = "Getting a token..."
260
+
261
+ tokens , userinfo = self .client .auth_callback (
262
+ code , redirect_uri , code_verifier = code_verifier
263
+ )
264
+ self ._save_credentials (tokens , userinfo )
265
+
266
+ spinner .ok ("Successfully authenticated" )
269
267
270
268
return tokens , userinfo
271
269
0 commit comments