From 8d40b1a40586dd45d7fe55dddf80b1b36fd5720f Mon Sep 17 00:00:00 2001 From: William 'psyFi' Hatcher Date: Tue, 14 Jul 2020 22:01:50 -0500 Subject: [PATCH] Added Unauthorized redirect to basic usage --- README.md | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 6313b50..ff90e61 100644 --- a/README.md +++ b/README.md @@ -14,7 +14,7 @@ python3 -m pip install Flask-Discord ### Basic Example ```python from flask import Flask, redirect, url_for -from flask_discord import DiscordOAuth2Session, requires_authorization +from flask_discord import DiscordOAuth2Session, requires_authorization, Unauthorized app = Flask(__name__) @@ -37,7 +37,12 @@ def login(): def callback(): discord.callback() return redirect(url_for(".me")) - + + +@app.errorhandler(Unauthorized) +def redirect_unauthorized(e): + return redirect(url_for("login")) + @app.route("/me/") @requires_authorization