Skip to content

Commit

Permalink
✔ Use the already implemented bot_request
Browse files Browse the repository at this point in the history
  • Loading branch information
weibeu committed Jun 27, 2020
1 parent 5d84158 commit 0bab72c
Showing 1 changed file with 8 additions and 6 deletions.
14 changes: 8 additions & 6 deletions flask_discord/models/user.py
Original file line number Diff line number Diff line change
@@ -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.
Expand Down Expand Up @@ -186,11 +188,11 @@ def add_to_guild(self, guild_id) -> dict:
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
Expand Down

0 comments on commit 0bab72c

Please sign in to comment.