Merge branch 'main' into doc/#50-Added_section_Add_how_to_release_to_Developer_Guide

Author: ckunki

doc/changes/changelog.md

@@ -1,6 +1,7 @@
 # Changes
 
 * [unreleased](unreleased.md)
+* [0.6.0](changes_0.6.0.md)
 * [0.5.0](changes_0.5.0.md)
 * [0.4.0](changes_0.4.0.md)
 * [0.3.0](changes_0.3.0.md)
@@ -13,6 +14,7 @@
 hidden:
 ---
 unreleased
+changes_0.6.0
 changes_0.5.0
 changes_0.4.0
 changes_0.3.0

doc/changes/changes_0.5.0.md

@@ -1,5 +1,5 @@
 # 0.5.0 - 2024-05-16
 
-# Bugfixes
+## Bugfixes
 
 * Fixed typehint for optional argument to enable usage with python 3.9

doc/changes/changes_0.6.0.md

@@ -0,0 +1,17 @@
+# 0.6.0 - 2024-05-22
+
+## Feature
+
+* #45 Added a helper function to assemble DB connection parameters.
+
+## Documentation
+
+* #50: Described release process in Developer Guide
+
+## Bugfixes
+
+* #44 Fixed the return value of the operational_saas_database_id fixture.
+
+## Refactoring
+
+* #19: Removed slack notifications for events other than `schedule`

doc/changes/unreleased.md

@@ -1,11 +1 @@
 # Unreleased
-
-This release removes slack notifications for events other than `schedule`.
-
-## Refactoring
-
-* #19: Removed slack notifications for events other than `schedule`
-
-## Doc
-
-* #50: Described release process in Developer Guide

exasol/saas/client/api_access.py

@@ -3,7 +3,7 @@
 import logging
 import time
 
-from typing import Iterable, List, Optional
+from typing import Iterable, List, Optional, Any
 from contextlib import contextmanager
 import datetime as dt
 from datetime import datetime, timedelta
@@ -32,6 +32,7 @@
     add_allowed_ip,
     delete_allowed_ip,
 )
+from exasol.saas.client.openapi.types import UNSET
 
 
 LOG = logging.getLogger(__name__)
@@ -76,6 +77,75 @@ def create_saas_client(
     )
 
 
+def _get_database_id(
+        account_id: str,
+        client: openapi.AuthenticatedClient,
+        database_name: str,
+) -> str:
+    """
+    Finds the database id, given an optional database name. If the name is not
+    provided returns an id of any non-deleted database. The latter option may be
+    useful for testing.
+    """
+    dbs = list_databases.sync(account_id, client=client)
+    dbs = list(filter(lambda db: (db.name == database_name) and         # type: ignore
+                                 (db.deleted_at is UNSET) and           # type: ignore
+                                 (db.deleted_by is UNSET), dbs))        # type: ignore
+    if not dbs:
+        raise RuntimeError(f'SaaS database {database_name} was not found.')
+    return dbs[0].id
+
+
+def get_connection_params(
+        host: str,
+        account_id: str,
+        pat: str,
+        database_id: str | None = None,
+        database_name: str | None = None,
+) -> dict[str, Any]:
+    """
+    Gets the database connection parameters, such as those required by pyexasol:
+    - dns
+    - user
+    - password.
+    Returns the parameters in a dictionary that can be used as kwargs when
+    creating a connection, like in the code below:
+
+    connection_params = get_connection_params(...)
+    connection = pyexasol.connect(**connection_params)
+
+    Args:
+        host:           SaaS service URL.
+        account_id:     User account ID
+        pat:            Personal Access Token.
+        database_id:    Database ID, id known.
+        database_name:  Database name, in case the id is unknown.
+    """
+
+    with create_saas_client(host, pat) as client:
+        if not database_id:
+            if not database_name:
+                raise ValueError(('To get SaaS connection parameters, '
+                                  'either database name or database id must be provided.'))
+            database_id = _get_database_id(account_id, client, database_name=database_name)
+        clusters = list_clusters.sync(account_id,
+                                      database_id,
+                                      client=client)
+        cluster_id = next(filter(lambda cl: cl.main_cluster, clusters)).id  # type: ignore
+        connections = get_cluster_connection.sync(account_id,
+                                                  database_id,
+                                                  cluster_id,
+                                                  client=client)
+        if connections is None:
+            raise RuntimeError('Failed to get the SaaS connection data.')
+        connection_params = {
+            'dsn': f'{connections.dns}:{connections.port}',
+            'user': connections.db_username,
+            'password': pat
+        }
+        return connection_params
+
+
 class OpenApiAccess:
     """
     This class is meant to be used only in the context of the API
@@ -190,7 +260,6 @@ def poll_status():
         if poll_status() not in success:
             raise DatabaseStartupFailure()
 
-
     def clusters(
             self,
             database_id: str,