Basic

flask_discord/client.py

@@ -0,0 +1,49 @@
+from . import configs
+from requests_oauthlib import OAuth2Session
+
+from flask import request, session, redirect
+
+
+
+class DiscordOAuth2Session(object):
+
+    def __init__(self, client_id, client_secret, redirect_uri):
+        self.client_id = client_id
+        self.client_secret = client_secret
+        self.redirect_uri = redirect_uri
+
+    @staticmethod
+    def __token_updater(token):
+        session["oauth2_token"] = token
+
+    def __make_session(self, token=None, state=None, scope=None)
+        return OAuth2Session(
+            client_id=self.client_id,
+            token=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 make_session(self):
+        scope = request.args.get("scope", configs.DEFAULT_SCOPES).split()
+        discord_session = self.__make_session(scope=scope)
+        authorization_url, state = discord_session.authorization_url(configs.AUTHORIZATION_BASE_URL)
+        session["oauth2_state"] = state
+        return redirect(authorization_url)
+
+    def callback(self):
+        if request.values.get("error"):
+            return request.values["error"]
+        discord = self.__make_session(state=session.get("oauth2_state"))
+        token = discord.fetch_token(
+            configs.TOKEN_URL,
+            client_secret=self.client_secret,
+            authorization_response=request.url
+        )
+        session["oauth2_token"] = token