Xefir/Divent
1
1
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases 23 Wiki Activity

Add redirection if not logged
All checks were successful
continuous-integration/drone/push Build is passing
Details
continuous-integration/drone/tag Build is passing
Details

Browse Source
This commit is contained in:
Michel Roux 2022-09-18 21:15:03 +00:00
parent bb842f7fd0
commit 1498587437
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
divent/bot.py
View File

@ -140,6 +140,8 @@ def login_required(fn):
async def wrapper(*args, **kwargs): async def wrapper(*args, **kwargs):
if session.get("oauth2_token"): if session.get("oauth2_token"):
return await fn(*args, **kwargs) return await fn(*args, **kwargs)
session["redirect_url"] = request.path
return redirect(url_for(".login")) return redirect(url_for(".login"))
return wrapper return wrapper
@ -171,7 +173,7 @@ async def callback():
authorization_response=request.url, authorization_response=request.url,
) )
token_updater(token) token_updater(token)
return redirect(url_for(".guilds")) return redirect(session.pop("redirect_url", url_for(".guilds")))
@app.route("/guilds") @app.route("/guilds")