From 0bab72c9e3c8318269bba2d61d1598971702c5aa Mon Sep 17 00:00:00 2001 From: thec0sm0s Date: Sat, 27 Jun 2020 15:52:46 +0530 Subject: [PATCH] =?UTF-8?q?=E2=9C=94=20Use=20the=20already=20implemented?= =?UTF-8?q?=20bot=5Frequest?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- flask_discord/models/user.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/flask_discord/models/user.py b/flask_discord/models/user.py index 2094689..adbcd77 100644 --- a/flask_discord/models/user.py +++ b/flask_discord/models/user.py @@ -1,10 +1,12 @@ from .. import configs from .guild import Guild +from .. import exceptions from .base import DiscordModelsBase -from flask import current_app, session from .connections import UserConnection +from flask import current_app, session + class User(DiscordModelsBase): """Class representing Discord User. @@ -186,11 +188,11 @@ class User(DiscordModelsBase): Raises :py:class:`flask_discord.Unauthorized` if current user is not authorized. """ - data = {"access_token": session["DISCORD_OAUTH2_TOKEN"]["access_token"]} - headers = {"Authorization": f"Bot {current_app.config['DISCORD_BOT_TOKEN']}"} - return self._request( - f"/guilds/{guild_id}/members/{self.id}", method="PUT", oauth=False, json=data, headers=headers - ) or dict() + try: + data = {"access_token": session["DISCORD_OAUTH2_TOKEN"]["access_token"]} + except KeyError: + raise exceptions.Unauthorized + return self._bot_request(f"/guilds/{guild_id}/members/{self.id}", method="PUT", json=data) or dict() def fetch_guilds(self) -> list: """A method which makes an API call to Discord to get user's guilds. It prepares the internal guilds cache