Update server.py
Browse files
server.py
CHANGED
@@ -211,17 +211,24 @@ async def login(request: Request):
|
|
211 |
|
212 |
@app.route("/auth")
|
213 |
async def auth(request: Request):
|
214 |
-
|
215 |
-
|
216 |
-
|
217 |
-
|
218 |
-
|
219 |
-
|
220 |
-
|
221 |
-
|
222 |
-
|
223 |
-
|
224 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
225 |
|
226 |
@app.route("/logout")
|
227 |
async def logout(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 |
+
print("token type:", type(token))
|
218 |
+
id_token = token.get("id_token")
|
219 |
+
print("ID_TOKEN:", id_token)
|
220 |
+
if id_token:
|
221 |
+
user_info = await oauth.google.parse_id_token(request, token)
|
222 |
+
else:
|
223 |
+
user_info = token.get("userinfo")
|
224 |
+
if not user_info:
|
225 |
+
user_info = await oauth.google.userinfo(request, token=token)
|
226 |
+
request.session["user"] = dict(user_info)
|
227 |
+
return RedirectResponse(url="/")
|
228 |
+
except Exception as e:
|
229 |
+
import traceback
|
230 |
+
traceback.print_exc()
|
231 |
+
return PlainTextResponse(f"OAuth error: {e}")
|
232 |
|
233 |
@app.route("/logout")
|
234 |
async def logout(request: Request):
|