From 7147ddfda7893b3c68197f81fee61314beaf18ed Mon Sep 17 00:00:00 2001
From: Luke <luke@lmbyrne.dev>
Date: Fri, 25 Oct 2019 18:34:13 +0100
Subject: [PATCH] Add code to oauth2 token get

---
 disco/api/client.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/disco/api/client.py b/disco/api/client.py
index 7febb6b..e38899b 100644
--- a/disco/api/client.py
+++ b/disco/api/client.py
@@ -106,7 +106,7 @@ class APIClient(LoggingClass):
         data = self.http(Routes.GATEWAY_BOT_GET).json()
         return data
 
-    def oauth2_token_get(self, grant_type, scope, refresh_token=None, redirect_url=None):
+    def oauth2_token_get(self, grant_type, scope, code=None, refresh_token=None, redirect_url=None):
         payload = {
             'client_id': self.client.state.me.id,
             'client_secret': self.client.config.secret,
@@ -114,6 +114,7 @@ class APIClient(LoggingClass):
             'scope': scope,
         }
         payload.update(optional(
+            code=code,
             refresh_token=refresh_token,
             redirect_url=redirect_url,
         ))