REST API: support all authentication methods (#354)

support all authentication methods, e.g. HTTP basic or subclasses of AuthBase
dlt-hub · Feb 15, 2024 · 6ec138c · 6ec138c
1 parent 4428038
commit 6ec138c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sources/rest_api/__init__.py b/sources/rest_api/__init__.py
@@ -19,7 +19,7 @@
     HeaderLinkPaginator,
     UnspecifiedPaginator,
 )
-from .auth import BearerTokenAuth
+from .auth import BearerTokenAuth, AuthBase
 
 
 PAGINATOR_MAP = {
@@ -83,7 +83,7 @@ def create_paginator(paginator_config):
 
 
 def create_auth(auth_config):
-    if isinstance(auth_config, BearerTokenAuth):
+    if isinstance(auth_config, AuthBase):
         return auth_config
     return BearerTokenAuth(auth_config.get("token")) if auth_config else None