make _update_driver_config common method to all credentials

ydb-platform · Oct 4, 2024 · dff7303 · dff7303
1 parent e3a2f98
commit dff7303
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/ydb/credentials.py b/ydb/credentials.py
@@ -45,6 +45,9 @@ def get_auth_token(self) -> str:
                 return token
         return ""
 
+    def _update_driver_config(self, driver_config):
+        pass
+
 
 class OneToManyValue(object):
     def __init__(self):

diff --git a/ydb/driver.py b/ydb/driver.py
@@ -234,8 +234,7 @@ def get_config(
 
         driver_config._update_attrs_by_kwargs(**kwargs)
 
-    if isinstance(driver_config.credentials, credentials_impl.StaticCredentials):
-        driver_config.credentials._update_driver_config(driver_config)
+    driver_config.credentials._update_driver_config(driver_config)
 
     return driver_config