mirror of
https://github.com/yawaflua/Flask-Discord.git
synced 2025-12-09 20:09:30 +02:00
94 lines
3.2 KiB
Python
94 lines
3.2 KiB
Python
import os
|
|
import abc
|
|
|
|
from . import configs
|
|
|
|
from flask import session, jsonify
|
|
from requests_oauthlib import OAuth2Session
|
|
|
|
|
|
class DiscordOAuth2HttpClient(abc.ABC):
|
|
"""An OAuth2 http abstract base class providing some factory methods.
|
|
This class is meant to be overridden by flask_discord.DiscordOAuth2Session class.
|
|
|
|
Attributes
|
|
----------
|
|
client_id : int
|
|
The client ID of discord application provided.
|
|
client_secret : str
|
|
The client secret of discord application provided.
|
|
redirect_uri : str
|
|
The default URL to use to redirect user to after authorization.
|
|
|
|
"""
|
|
|
|
def __init__(self, client_id, client_secret, redirect_uri):
|
|
self.client_id = client_id
|
|
self.client_secret = client_secret
|
|
self.redirect_uri = redirect_uri
|
|
if "http://" in self.redirect_uri:
|
|
os.environ["OAUTHLIB_INSECURE_TRANSPORT"] = "true"
|
|
|
|
@staticmethod
|
|
def _token_updater(token):
|
|
session["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.
|
|
|
|
Parameters
|
|
----------
|
|
token : str, optional
|
|
The authorization token to use which was previously received from authorization code grant.
|
|
state : str, optional
|
|
The state to use for OAuth2 session.
|
|
scope : list, optional
|
|
List of valid `Discord OAuth2 Scopes
|
|
<https://discordapp.com/developers/docs/topics/oauth2#shared-resources-oauth2-scopes>`_.
|
|
|
|
Returns
|
|
-------
|
|
OAuth2Session
|
|
An instance of OAuth2Session class.
|
|
|
|
"""
|
|
return OAuth2Session(
|
|
client_id=self.client_id,
|
|
token=token or session.get("oauth2_token"),
|
|
state=state,
|
|
scope=scope,
|
|
redirect_uri=self.redirect_uri,
|
|
auto_refresh_kwargs={
|
|
'client_id': self.client_id,
|
|
'client_secret': self.client_secret,
|
|
},
|
|
auto_refresh_url=configs.TOKEN_URL,
|
|
token_updater=self._token_updater)
|
|
|
|
def get(self, route: str) -> dict:
|
|
"""Sends HTTP GET request to provided route or discord endpoint.
|
|
|
|
Note
|
|
----
|
|
It automatically prefixes the API Base URL so you will just have to pass routes or URL endpoints.
|
|
|
|
Parameters
|
|
----------
|
|
route : str
|
|
Route or endpoint URL to send HTTP GET request to.
|
|
Example: ``/users/@me``
|
|
|
|
Returns
|
|
-------
|
|
dict
|
|
Dictionary containing received from sent HTTP GET request.
|
|
"""
|
|
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"))
|
|
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()
|
|
return jsonify(user=user, guilds=guilds, connections=connections)
|