yawaflua/Flask-Discord
1
0
Fork 0
mirror of https://github.com/yawaflua/Flask-Discord.git synced 2025-12-09 20:09:30 +02:00
Code Issues Packages Projects Releases Wiki Activity

changed a bunch of stuff to async

Browse Source
This commit is contained in:
Philip Dowie
2020-08-16 10:19:14 +12:00
parent 2076f1c1df
commit 511ecc0ce0
3 changed files with 20 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
README.md
View File

@@ -29,24 +29,24 @@ discord = DiscordOAuth2Session(app)
@app.route("/login/")
def login():
async def login():
return discord.create_session()
@app.route("/callback/")
def callback():
discord.callback()
async def callback():
await discord.callback()
return redirect(url_for(".me"))
@app.errorhandler(Unauthorized)
def redirect_unauthorized(e):
async def redirect_unauthorized(e):
return redirect(url_for("login"))
@app.route("/me/")
@requires_authorization
def me():
async def me():
user = discord.fetch_user()
return f"""
<html>

4
quart_discord/client.py
View File

@@ -137,14 +137,14 @@ class DiscordOAuth2Session(_http.DiscordOAuth2HttpClient):
"""
return session.get("DISCORD_OAUTH2_TOKEN")
def callback(self):
async def callback(self):
"""A method which should be always called after completing authorization code grant process
usually in callback view.
It fetches the authorization token and saves it quart
`session <https://pgjones.gitlab.io/quart/reference/source/quart.sessions.html#quart.sessions.Session>`_ object.
"""
error = request.values.get("error")
error = await request.values.get("error")
if error:
if error == "access_denied":
raise exceptions.AccessDenied()

26
tests/test_app.py
View File

@@ -20,7 +20,7 @@ HYPERLINK = '<a href="{}">{}</a>'
@app.route("/")
def index():
async def index():
if not discord.authorized:
return f"""
{HYPERLINK.format(url_for(".login"), "Login")} <br />
@@ -38,34 +38,34 @@ def index():
@app.route("/login/")
def login():
async def login():
return discord.create_session()
@app.route("/login-data/")
def login_with_data():
async def login_with_data():
return discord.create_session(data=dict(redirect="/me/", coupon="15off", number=15, zero=0, status=False))
@app.route("/invite-bot/")
def invite_bot():
async def invite_bot():
return discord.create_session(scope=["bot"], permissions=8, guild_id=464488012328468480, disable_guild_select=True)
@app.route("/invite-oauth/")
def invite_oauth():
async def invite_oauth():
return discord.create_session(scope=["bot", "identify"], permissions=8)
@app.route("/callback/")
def callback():
data = discord.callback()
async def callback():
data = await discord.callback()
redirect_to = data.get("redirect", "/")
return redirect(redirect_to)
@app.route("/me/")
def me():
async def me():
user = discord.fetch_user()
return f"""
<html>
@@ -83,19 +83,19 @@ def me():
@app.route("/me/guilds/")
def user_guilds():
async def user_guilds():
guilds = discord.fetch_guilds()
return "<br />".join([f"[ADMIN] {g.name}" if g.permissions.administrator else g.name for g in guilds])
@app.route("/add_to/<int:guild_id>/")
def add_to_guild(guild_id):
async def add_to_guild(guild_id):
user = discord.fetch_user()
return user.add_to_guild(guild_id)
@app.route("/me/connections/")
def my_connections():
async def my_connections():
user = discord.fetch_user()
connections = discord.fetch_connections()
return f"""
@@ -112,14 +112,14 @@ def my_connections():
@app.route("/logout/")
def logout():
async def logout():
discord.revoke()
return redirect(url_for(".index"))
@app.route("/secret/")
@requires_authorization
def secret():
async def secret():
return os.urandom(16)