From 3d48d7761572092dc19adaff1b8606863c472e0b Mon Sep 17 00:00:00 2001
From: IdmFoundInHim <idmfoundinhim@gmail.com>
Date: Mon, 24 Aug 2020 14:57:05 -0400
Subject: [PATCH] Add code parameter to get_access_token of SpotifyPKCE (#556)

* Add code parameter to oauth2.SpotifyPKCE.get_access_token

* Fix the PKCE integration test cache
---
 spotipy/oauth2.py                        | 15 ++++++++++-----
 tests/integration/test_user_endpoints.py |  2 +-
 2 files changed, 11 insertions(+), 6 deletions(-)

diff --git a/spotipy/oauth2.py b/spotipy/oauth2.py
index dcb595d..03f6b99 100644
--- a/spotipy/oauth2.py
+++ b/spotipy/oauth2.py
@@ -803,12 +803,17 @@ class SpotifyPKCE(SpotifyAuthBase):
         self.code_verifier = self._get_code_verifier()
         self.code_challenge = self._get_code_challenge()
 
-    def get_access_token(self, check_cache=True):
-        """ Gets the access token for the app given the code
+    def get_access_token(self, code=None, check_cache=True):
+        """ Gets the access token for the app
+
+            If the code is not given and no cached token is used, an
+            authentication window will be shown to the user to get a new
+            code.
 
             Parameters:
-                - check_cache - check for locally stored token before request
-                                a new token if True
+                - code - the response code from authentication
+                - check_cache - if true, checks for locally stored token
+                                before requesting a new token if True
         """
 
         if check_cache:
@@ -826,7 +831,7 @@ class SpotifyPKCE(SpotifyAuthBase):
         payload = {
             "client_id": self.client_id,
             "grant_type": "authorization_code",
-            "code": self.get_authorization_code(),
+            "code": code or self.get_authorization_code(),
             "redirect_uri": self.redirect_uri,
             "code_verifier": self.code_verifier
         }
diff --git a/tests/integration/test_user_endpoints.py b/tests/integration/test_user_endpoints.py
index 4438b65..ade2061 100644
--- a/tests/integration/test_user_endpoints.py
+++ b/tests/integration/test_user_endpoints.py
@@ -435,7 +435,7 @@ class SpotifyPKCETests(unittest.TestCase):
             'user-follow-read '
             'user-follow-modify '
         )
-        auth_manager = SpotifyPKCE(scope=scope, cache_path=".cache-implicittest")
+        auth_manager = SpotifyPKCE(scope=scope, cache_path=".cache-pkcetest")
         cls.spotify = Spotify(auth_manager=auth_manager)
 
     def test_user_follows_and_unfollows_artist(self):