Prefixed session data with discord to allow users to have their own other OAuth sessions

This commit is contained in:
thecosmos
2019-05-19 17:45:04 +05:30
parent bf49390a4c
commit f91bdca7d9
2 changed files with 6 additions and 6 deletions

View File

@@ -32,7 +32,7 @@ class DiscordOAuth2HttpClient(abc.ABC):
@staticmethod
def _token_updater(token):
session["oauth2_token"] = token
session["discord_oauth2_token"] = token
def _make_session(self, token: str = None, state: str = None, scope: list = None) -> OAuth2Session:
"""A low level method used for creating OAuth2 session.
@@ -55,7 +55,7 @@ class DiscordOAuth2HttpClient(abc.ABC):
"""
return OAuth2Session(
client_id=self.client_id,
token=token or session.get("oauth2_token"),
token=token or session.get("discord_oauth2_token"),
state=state,
scope=scope,
redirect_uri=self.redirect_uri,
@@ -86,7 +86,7 @@ class DiscordOAuth2HttpClient(abc.ABC):
return self._make_session().get(configs.API_BASE_URL + route).json()
def get_json(self):
discord_session = self._make_session(token=session.get("oauth2_token"))
discord_session = self._make_session(token=session.get("discord_oauth2_token"))
user = discord_session.get(configs.API_BASE_URL + '/users/@me').json()
guilds = discord_session.get(configs.API_BASE_URL + '/users/@me/guilds').json()
connections = discord_session.get(configs.API_BASE_URL + '/users/@me/connections').json()