Add root_certificates option for ydb.DriverConfig

Author: Pseudolukian

examples/static-credentials/example.py

@@ -1,18 +1,58 @@
 import ydb
 
+def load_ydb_ca_cert(path:str) -> str:
+    """Load CA certification.
+
+    Args:
+        path (str): path to CA certification.
+    
+    The function load_ydb_ca_cert accepts a path to a CA certificate 
+    and returns its content as a byte string for further passing to ydb.DriverConfig. 
+    If the specified path is incorrect or the certificate does not exist, 
+    it raises an exception with the message "CA not found".
+
+    """
+    if path is not None and os.path.exists(path):
+        with open(path, "rb") as file:
+            return file.read()
+    else:
+        raise FileNotFoundError("CA not found")
 
 def test_driver_works(driver: ydb.Driver):
+    """Tests the functionality of the YDB driver.
+
+    Waits for the driver to become ready and executes a simple SQL query to verify that the driver works as expected.
+
+    Args:
+        driver (ydb.Driver): The YDB driver instance to test.
+
+    Raises:
+        AssertionError: If the SQL query does not return the expected result.
+    """
     driver.wait(5)
     pool = ydb.QuerySessionPool(driver)
     result = pool.execute_with_retries("SELECT 1 as cnt")
     assert result[0].rows[0].cnt == 1
 
-
 def auth_with_static_credentials(endpoint: str, database: str, user: str, password: str):
+    """Authenticate using static credentials.
+
+    Args:
+        endpoint (str): Accepts a string in the format `grpcs://<node-fqdn>:2136` or `grpcs://<node-ip>:2136`.
+        database (str): Accepts a string, the database name in the format `/Root/<database-name>`.
+        user (str): Username.
+        password (str): User password.
+
+    Notes:
+        The argument `root_certificates` of the function `ydb.DriverConfig` takes the content of the cluster's root certificate for connecting to cluster nodes via TLS.
+        Note that the VM from which you are connecting must be in the cluster's domain for which the CA certificate is issued.
+    """
+
     driver_config = ydb.DriverConfig(
-        endpoint=endpoint,
-        database=database,
-        credentials=ydb.StaticCredentials.from_user_password(user, password),
+        endpoint = endpoint,
+        database = database,
+        credentials = ydb.StaticCredentials.from_user_password(user, password),
+        root_certificates = load_ydb_ca_cert(path = <path_to_ca-cert.crt>)
     )
 
     with ydb.Driver(driver_config=driver_config) as driver: