Yadciel/Titan
1
0
Fork 0
mirror of https://github.com/TitanEmbeds/Titan.git synced 2024-12-24 14:07:03 +01:00
Code Issues Packages Projects Releases Wiki Activity

Redirect to administrate guild properly after authenticating

Browse Source
This commit is contained in:
Jeremy Zhang 2017-04-07 17:50:09 -07:00
parent c75948ca1d
commit f7196f96c7
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
titanembeds/blueprints/user/user.py
View File

@ -60,6 +60,10 @@ def dashboard():
if not guilds: if not guilds:
session["redirect"] = url_for("user.dashboard") session["redirect"] = url_for("user.dashboard")
return redirect(url_for("user.logout")) return redirect(url_for("user.logout"))
if session["redirect"]:
redir = session['redirect']
session['redirect'] = None
return redirect(redir)
return render_template("dashboard.html.j2", servers=guilds, icon_generate=generate_guild_icon_url) return render_template("dashboard.html.j2", servers=guilds, icon_generate=generate_guild_icon_url)
@user.route("/administrate_guild/<guild_id>", methods=["GET"]) @user.route("/administrate_guild/<guild_id>", methods=["GET"])
@ -69,6 +73,7 @@ def administrate_guild(guild_id):
return redirect(url_for("user.dashboard")) return redirect(url_for("user.dashboard"))
guild = discord_api.get_guild(guild_id) guild = discord_api.get_guild(guild_id)
if guild['code'] != 200: if guild['code'] != 200:
session["redirect"] = url_for("user.administrate_guild", guild_id=guild_id, _external=True)
return redirect(generate_bot_invite_url(guild_id)) return redirect(generate_bot_invite_url(guild_id))
db_guild = db.session.query(Guilds).filter(Guilds.guild_id == guild_id).first() db_guild = db.session.query(Guilds).filter(Guilds.guild_id == guild_id).first()
if not db_guild: if not db_guild: