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 internal flask_discord request method

Browse Source
This commit is contained in:
thec0sm0s
2020-05-10 17:20:43 +05:30
parent fbf341a3ec
commit 1e97db58f5
1 changed files with 7 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
flask_discord/models/user.py
View File

@@ -1,9 +1,8 @@
from flask import current_app, session from .. import configs
import requests
from json import JSONDecodeError
from .base import DiscordModelsBase from .base import DiscordModelsBase
from .. import configs, exceptions from flask import current_app, session
class User(DiscordModelsBase): class User(DiscordModelsBase):
@@ -92,19 +91,14 @@ class User(DiscordModelsBase):
Raises :py:class:`flask_discord.Unauthorized` if current user is not authorized. Raises :py:class:`flask_discord.Unauthorized` if current user is not authorized.
""" """
route = configs.DISCORD_API_BASE_URL + f"/guilds/{guild_id}/members/{self.id}"
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']}"}
response = requests.put(route, json=data, headers=headers) try:
return current_app.discord.request(
if response.status_code == 401: f"/guilds/{guild_id}/members/{self.id}", method="PUT", oauth=False, json=data, headers=headers)
raise exceptions.Unauthorized except JSONDecodeError:
if response.status_code == 204:
return dict() return dict()
return response.json()
class Bot(User): class Bot(User):
"""Class representing the client user itself.""" """Class representing the client user itself."""