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

tests

Browse Source
This commit is contained in:
thecosmos
2019-05-06 21:16:28 +05:30
parent bf61177a3e
commit 1654c55f77
3 changed files with 43 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
flask_discord/client.py
View File

@@ -1,7 +1,9 @@
import os
from . import configs
from requests_oauthlib import OAuth2Session
from flask import request, session, redirect
from flask import request, session, redirect, jsonify
class DiscordOAuth2Session(object):
@@ -10,6 +12,8 @@ class DiscordOAuth2Session(object):
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):
@@ -46,3 +50,10 @@ class DiscordOAuth2Session(object):
authorization_response=request.url
)
session["oauth2_token"] = token
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)