add example

Author: vgvoleg

examples/static-credentials/__main__.py

@@ -0,0 +1,23 @@
+# -*- coding: utf-8 -*-
+import argparse
+from example import run
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+        description="""\033[92mStatic Credentials example.\x1b[0m\n""",
+    )
+    parser.add_argument("-e", "--endpoint", help="Endpoint url to use", default="grpc://localhost:2136")
+    parser.add_argument("-d", "--database", help="Name of the database to use", default="/local")
+    parser.add_argument("-u", "--user", help="User to auth with", default="root")
+    parser.add_argument("-p", "--password", help="Password from user to auth with", default="1234")
+
+    args = parser.parse_args()
+
+    run(
+        endpoint=args.endpoint,
+        database=args.database,
+        user=args.user,
+        password=args.password,
+    )

examples/static-credentials/example.py

@@ -0,0 +1,66 @@
+import ydb
+
+
+def test_driver_works(driver: ydb.Driver):
+    driver.wait(5)
+    pool = ydb.QuerySessionPool(driver)
+    pool.execute_with_retries("SELECT 1")
+    print("everything is fine")
+
+
+def auth_with_static_credentials_old(endpoint: str, database: str, user: str, password: str):
+    driver_config_temp = ydb.DriverConfig(
+        endpoint=endpoint,
+        database=database,
+    )
+    creds = ydb.StaticCredentials(
+        driver_config=driver_config_temp,
+        user=user,
+        password=password,
+    )
+
+    driver_config = ydb.DriverConfig(
+        endpoint=endpoint,
+        database=database,
+        credentials=creds,
+    )
+
+    with ydb.Driver(driver_config=driver_config) as driver:
+        test_driver_works(driver)
+
+
+def auth_with_static_credentials_new(endpoint: str, database: str, user: str, password: str):
+    driver_config = ydb.DriverConfig(
+        endpoint,
+        database,
+    )
+    creds = ydb.StaticCredentials(
+        driver_config,
+        user,
+        password,
+    )
+
+    with ydb.Driver(driver_config=driver_config, credentials=creds) as driver:
+        test_driver_works(driver)
+
+
+def auth_with_user_password_credentials(endpoint: str, database: str, user: str, password: str):
+    driver_config = ydb.DriverConfig(
+        endpoint=endpoint,
+        database=database,
+        credentials=ydb.UserPasswordCredentials(
+            user=user,
+            password=password,
+            endpoint=endpoint,
+            database=database,
+        ),
+    )
+
+    with ydb.Driver(driver_config=driver_config) as driver:
+        test_driver_works(driver)
+
+
+def run(endpoint: str, database: str, user: str, password: str):
+    auth_with_static_credentials_old(endpoint, database, user, password)
+    auth_with_static_credentials_new(endpoint, database, user, password)
+    auth_with_user_password_credentials(endpoint, database, user, password)

ydb/credentials.py

@@ -215,7 +215,7 @@ def _make_token_request(self):
 
 class UserPasswordCredentials(AbstractExpiringTokenCredentials):
     def __init__(self, user, password, endpoint, database, root_certificates=None, tracer=None):
-        super(UserPasswordCredentials).__init__(tracer)
+        super(UserPasswordCredentials, self).__init__(tracer)
 
         from .driver import DriverConfig  # to prevent circular dependencies
 
@@ -224,6 +224,7 @@ def __init__(self, user, password, endpoint, database, root_certificates=None, t
             database=database,
             root_certificates=root_certificates,
         )
+
         self.user = user
         self.password = password
         self.request_timeout = 10

ydb/driver.py

@@ -1,12 +1,16 @@
 # -*- coding: utf-8 -*-
+import grpc
+import logging
+import os
+from typing import Any  # noqa
+
 from . import credentials as credentials_impl, table, scheme, pool
 from . import tracing
-import os
-import grpc
 from . import iam
 from . import _utilities
 
-from typing import Any  # noqa
+
+logger = logging.getLogger(__name__)
 
 
 class RPCCompression:
@@ -190,9 +194,13 @@ def set_grpc_keep_alive_timeout(self, timeout):
         self.grpc_keep_alive_timeout = timeout
         return self
 
-    def _update_attrs_by_kwargs(self, kwargs: dict):
+    def _update_attrs_by_kwargs(self, **kwargs):
         for key, value in kwargs.items():
-            if getattr(self, key) is None:
+            if value is not None:
+                if getattr(self, key) is not None:
+                    logger.warning(
+                        f"Arg {key} was used in both DriverConfig and Driver. Value from Driver will be used."
+                    )
                 setattr(self, key, value)
 
 
@@ -225,7 +233,7 @@ def get_config(
     kwargs["root_certificates"] = root_certificates
     kwargs["credentials"] = credentials
 
-    driver_config._update_attrs_by_kwargs(kwargs)
+    driver_config._update_attrs_by_kwargs(**kwargs)
 
     return driver_config