pgurazada1 commited on
Commit
d356ddf
·
verified ·
1 Parent(s): 8fd81bf

Update server.py

Browse files
Files changed (1) hide show
  1. server.py +10 -12
server.py CHANGED
@@ -211,19 +211,17 @@ async def login(request: Request):
211
 
212
  @app.route("/auth")
213
  async def auth(request: Request):
214
- try:
215
- token = await oauth.google.authorize_access_token(request)
216
- print("TOKEN:", token)
217
- id_token = token.get("id_token")
218
- if id_token is not None:
219
- user_info = await oauth.google.parse_id_token(request, token)
220
- else:
 
221
  user_info = await oauth.google.userinfo(request, token=token)
222
- request.session["user"] = dict(user_info)
223
- return RedirectResponse(url="/")
224
- except Exception as e:
225
- print("OAuth error:", e)
226
- return PlainTextResponse(f"OAuth error: {e}")
227
 
228
  @app.route("/logout")
229
  async def logout(request: Request):
 
211
 
212
  @app.route("/auth")
213
  async def auth(request: Request):
214
+ token = await oauth.google.authorize_access_token(request)
215
+ print("TOKEN:", token)
216
+ id_token = token.get("id_token")
217
+ if id_token:
218
+ user_info = await oauth.google.parse_id_token(request, token)
219
+ else:
220
+ user_info = token.get("userinfo")
221
+ if not user_info:
222
  user_info = await oauth.google.userinfo(request, token=token)
223
+ request.session["user"] = dict(user_info)
224
+ return RedirectResponse(url="/")
 
 
 
225
 
226
  @app.route("/logout")
227
  async def logout(request: Request):