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

Use _request shorthand method

Browse Source
This commit is contained in:
thec0sm0s
2020-05-11 21:28:14 +05:30
parent 94d0c536c5
commit bc22196695
2 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
flask_discord/models/base.py
View File

@@ -1,8 +1,17 @@
from flask import current_app
from abc import ABC from abc import ABC
class DiscordModelsBase(ABC): class DiscordModelsBase(ABC):
@staticmethod
def _request(*args, **kwargs):
"""A shorthand to :py:func:flask_discord.request`. It uses Flask current_app local proxy to get the
Flask-Discord client.
"""
return current_app.discord.request(*args, **kwargs)
def to_json(self): def to_json(self):
"""A utility method which returns raw payload object as it was received from discord. """A utility method which returns raw payload object as it was received from discord.

2
flask_discord/models/user.py
View File

@@ -92,7 +92,7 @@ class User(DiscordModelsBase):
""" """
data = {"access_token": session["DISCORD_OAUTH2_TOKEN"]["access_token"]} data = {"access_token": session["DISCORD_OAUTH2_TOKEN"]["access_token"]}
headers = {"Authorization": f"Bot {current_app.config['DISCORD_BOT_TOKEN']}"} headers = {"Authorization": f"Bot {current_app.config['DISCORD_BOT_TOKEN']}"}
return current_app.discord.request( return self._request(
f"/guilds/{guild_id}/members/{self.id}", method="PUT", oauth=False, json=data, headers=headers f"/guilds/{guild_id}/members/{self.id}", method="PUT", oauth=False, json=data, headers=headers
) or dict() ) or dict()