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 diff --git a/docs/introduction.rst b/docs/introduction.rst index 28205aa..c10201f 100644 --- a/docs/introduction.rst +++ b/docs/introduction.rst @@ -43,7 +43,7 @@ in exchange for fetching user's details and display them on web page. .. code-block:: python3 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__) @@ -69,6 +69,11 @@ in exchange for fetching user's details and display them on web page. return redirect(url_for(".me")) + @app.errorhandler(Unauthorized) + def redirect_unauthorized(e): + return redirect(url_for("login")) + + @app.route("/me/") @requires_authorization def me(): @@ -85,4 +90,4 @@ in exchange for fetching user's details and display them on web page. if __name__ == "__main__": - app.run() \ No newline at end of file + app.run()