Added Unauthorized redirect to basic usage

This commit is contained in:
William 'psyFi' Hatcher
2020-07-14 22:01:50 -05:00
committed by GitHub
parent a0682fc192
commit 8d40b1a405

View File

@@ -14,7 +14,7 @@ python3 -m pip install Flask-Discord
### Basic Example ### Basic Example
```python ```python
from flask import Flask, redirect, url_for 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__) app = Flask(__name__)
@@ -37,7 +37,12 @@ def login():
def callback(): def callback():
discord.callback() discord.callback()
return redirect(url_for(".me")) return redirect(url_for(".me"))
@app.errorhandler(Unauthorized)
def redirect_unauthorized(e):
return redirect(url_for("login"))
@app.route("/me/") @app.route("/me/")
@requires_authorization @requires_authorization