pass args/kwargs to storage in TokenHandler

s4w3d0ff · Feb 3, 2025 · 09ab6b4 · 09ab6b4
1 parent 2482ffa
commit 09ab6b4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/spotifio/oauth.py b/spotifio/oauth.py
@@ -55,11 +55,11 @@ async def stop(self):
 
 
 class TokenHandler:
-    def __init__(self, client_id, client_secret, redirect_uri=None, scope=[], storage=None):
+    def __init__(self, client_id, client_secret, redirect_uri=None, scope=[], storage=None, *args, **kwargs):
         self.client_id = client_id
         self.redirect_uri = redirect_uri or "http://localhost:8888/callback"
         self.scope = scope
-        self.storage = storage or JSONStorage()
+        self.storage = storage or JSONStorage(*args, **kwargs)
         self._state = os.urandom(14).hex()
         self._auth_code = None
         self._auth_future = None