peak-sdk 1.8.0__py3-none-any.whl → 1.9.0__py3-none-any.whl

peak/_version.py

@@ -18,4 +18,4 @@
 # # You should have received a copy of the APACHE LICENSE, VERSION 2.0
 # # along with this program. If not, see <https://apache.org/licenses/LICENSE-2.0>
 #
-__version__: str = "1.8.0"
+__version__: str = "1.9.0"

peak/cli/resources/tenants.py

@@ -23,7 +23,7 @@ from typing import Optional
 
 import typer
 from peak.cli.args import OUTPUT_TYPES, PAGING
-from peak.constants import OutputTypes
+from peak.constants import OutputTypes, OutputTypesNoTable
 from peak.output import Writer
 from peak.resources.tenants import Tenant
 
@@ -37,6 +37,11 @@ _ENTITY_TYPE = typer.Option(
     help="Entity type to be used in this operation (e.g. - `workflow`, `webapp`, `api-deployment`).",
 )
 
+_DATA_STORE_TYPE = typer.Option(
+    None,
+    help="Data store type. The only allowed values is data-warehouse.",
+)
+
 
 @app.command(
     "list-instance-options",
@@ -58,7 +63,7 @@ def list_instance_options(
 
     \b
     🆗 ***Response:***
-    ```
+    ```json
     {
       "data": [
         {
@@ -66,7 +71,7 @@ def list_instance_options(
           "gpu": null,
           "gpuMemory": null,
           "id": 20,
-          "instanceClass": "Genaeral Purpose",
+          "instanceClass": "General Purpose",
           "memory": 125,
           "name": "Pico (0.125CPU, 0.125GB RAM)",
           "provider": "k8s",
@@ -84,3 +89,48 @@ def list_instance_options(
     with writer.pager():
         response = tenants_client.list_instance_options(entity_type=entity_type)
         writer.write(response)
+
+
+@app.command(
+    short_help="Get credentails for a data store.",
+)
+def get_credentials(
+    ctx: typer.Context,
+    data_store_type: Optional[str] = _DATA_STORE_TYPE,
+    output_type: Optional[OutputTypesNoTable] = OUTPUT_TYPES,  # noqa: ARG001
+) -> None:
+    """***Get*** credentials for the given data store type.
+
+    \b
+    📝 ***Example Usage:***<br/>
+    ```bash
+    peak tenants get-credentials --data-store-type data-warehouse
+    ```
+
+    \b
+    🆗 ***Response:***
+    ```json
+    {
+      "application": "application",
+      "connectionString": "snowflake://host/database?authenticator=OAUTH&token=generated-access-token",
+      "integration": "integration_name",
+      "port": 443,
+      "role": "role_name",
+      "schema": "schema",
+      "warehouse": "warehouse",
+      "accessToken": "generated-access-token",
+      "authType": "oauth",
+      "database": "database",
+      "host": "host",
+      "dataWarehouseType": "snowflake"
+    }
+    ```
+
+    🔗 [**API Documentation**](https://service.peak.ai/connections/api-docs/index.htm#/connections/get_api_v1_connections_credentials)
+    """
+    tenants_client: Tenant = ctx.obj["client"]
+    writer = Writer()
+
+    with writer.pager():
+        response = tenants_client.get_credentials(data_store_type=data_store_type)
+        writer.write(response)

peak/resources/tenants.py

@@ -31,7 +31,8 @@ from peak.session import Session
 class Tenant(BaseClient):
     """Tenant client class."""
 
-    BASE_ENDPOINT = "quota/api/v1"
+    QUOTA_BASE_ENDPOINT = "quota/api/v1"
+    CONNECTIONS_BASE_ENDPOINT = "connections/api/v1"
 
     def list_instance_options(
         self,
@@ -56,7 +57,7 @@ class Tenant(BaseClient):
             NotFoundException: The given image does not exist.
             InternalServerErrorException: The server failed to process the request.
         """
-        method, endpoint = HttpMethods.GET, f"{self.BASE_ENDPOINT}/settings/tenant-instance-options"
+        method, endpoint = HttpMethods.GET, f"{self.QUOTA_BASE_ENDPOINT}/settings/tenant-instance-options"
         params = {"entityType": entity_type}
 
         return self.session.create_request(  # type: ignore[no-any-return]
@@ -66,6 +67,40 @@ class Tenant(BaseClient):
             params=params,
         )
 
+    def get_credentials(
+        self,
+        data_store_type: Optional[str] = None,
+    ) -> Dict[str, Any]:
+        """Retrieve credentials for a given data store type.
+
+        REFERENCE:
+            🔗 `API Documentation <https://service.peak.ai/connections/api-docs/index.htm#/connections/get_api_v1_connections_credentials>`__
+
+        Args:
+            data_store_type (str): The type of the data store.
+                Allowed values are - data-warehouse.
+                Default - data-warehouse
+
+        Returns:
+            Dict[str, Any]: a dictionary containing the credentials for the data store.
+
+        Raises:
+            BadRequestException: The given request parameters are invalid.
+            UnauthorizedException: The credentials are invalid.
+            ForbiddenException: The user does not have permission to perform the operation.
+            NotFoundException: The given image does not exist.
+            InternalServerErrorException: The server failed to process the request.
+        """
+        method, endpoint = HttpMethods.GET, f"{self.CONNECTIONS_BASE_ENDPOINT}/connections/credentials"
+        params = {"type": data_store_type}
+
+        return self.session.create_request(  # type: ignore[no-any-return]
+            endpoint,
+            method,
+            content_type=ContentType.APPLICATION_JSON,
+            params=params,
+        )
+
 
 def get_client(session: Optional[Session] = None) -> Tenant:
     """Returns a Tenant client, If no session is provided, a default session is used.

{peak_sdk-1.8.0.dist-info → peak_sdk-1.9.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: peak-sdk
-Version: 1.8.0
+Version: 1.9.0
 Summary: Python SDK for interacting with the Peak platform
 Home-page: https://docs.peak.ai/sdk/latest/
 License: Apache-2.0

{peak_sdk-1.8.0.dist-info → peak_sdk-1.9.0.dist-info}/RECORD

@@ -1,6 +1,6 @@
 peak/__init__.py,sha256=yULGsiciVPNjdTO6uwEynsyLFiGBA2n7TT72DzKn9z8,1263
 peak/_metadata.py,sha256=ulvZZj-Vpjp6zVSUyw3vFjckTsX_QH0GPb0K1yln2io,25190
-peak/_version.py,sha256=sFvqkTtKF5mQzLwLLRh9eEwrry9-B2eEnVP7VUm9Uo4,886
+peak/_version.py,sha256=M4whDJj9RrzgffEaA0-SyoFtv4gSxWM-jH5gwkE8FGE,886
 peak/auth.py,sha256=KcqCqovY6sFiMGNAGP9k3AtCAXrZ-tYd7QP-PFYorX0,904
 peak/base_client.py,sha256=3cS8hZApONkojwdgzt_tbwSxwpFcn3xGwzVcmxkgOJc,1808
 peak/callbacks.py,sha256=OwAvipxU4kNNHr97J4LyuE3fs5JttXkrny0WIOKgJPU,3045
@@ -23,7 +23,7 @@ peak/cli/resources/alerts/emails.py,sha256=10ledYdyD6jQgTac62xn8yDodOLwoj8wD4s3H
 peak/cli/resources/artifacts.py,sha256=GiJ-yR1bvClDVO6EIHGqDyqsLGpmlUh9Bk_UPdUc-Co,12105
 peak/cli/resources/images.py,sha256=SwZC0y6pw3mq1HMiO9nPV3YzykbMuSIGt2TD47ZTZ1Y,43133
 peak/cli/resources/services.py,sha256=lBQX__-fEXx1xYFxEH6iq2T_jZQmNzWx3CJ1bwAzoEE,24295
-peak/cli/resources/tenants.py,sha256=Ng4KV7eXlC2vM4JUgKXb7Skr73erhgIYJL7-fcLvQwQ,2607
+peak/cli/resources/tenants.py,sha256=1vmNOwmLQdNV9RSVNaOZBRcAzHkOzIcVNz5ZH390O0c,4066
 peak/cli/resources/users.py,sha256=1qDv8bHYoIBHZ903Y5PBU_KIrMvOY4FrHPay4r0UF_c,2635
 peak/cli/resources/webapps.py,sha256=RyYxi2x8aJcQQ4Zowu0xqyMAmS8qlp0Mp-zodDKqkVw,21140
 peak/cli/resources/workflows.py,sha256=xJhXMB_j-1JZD8w-h4Sl9oe820WOvABoksOY6GS742w,52219
@@ -48,7 +48,7 @@ peak/resources/alerts.py,sha256=-RIfyCa4bkrb32poJlbKnsfLslpw49DyqvUOCTl-mUo,1237
 peak/resources/artifacts.py,sha256=lLVRRrznKKYVOA-pinWyqLpRd-AoFHGkFYwLT9aEGOU,14365
 peak/resources/images.py,sha256=7UB-jNEkE3HaqGLiTkv5wf5SU_9Y-jeFoMG_aMPT-y8,46715
 peak/resources/services.py,sha256=B9FJQD4eFGYdHbYbfHocfXm1LlRGil7yTmQPwTEpazM,16910
-peak/resources/tenants.py,sha256=J6lxfq1HXfyhA2EmSSSUHi4fRN6KU__yV7El1PVmvtk,2936
+peak/resources/tenants.py,sha256=amTs39KZ5hxDd7h3h1qo3TgYJOY1Eql0EhRgDBPlQ6Y,4380
 peak/resources/users.py,sha256=X9NfZo1dVvXQk_DCxzTKGG_VCsSmk7HzHdJrRR90mKQ,3428
 peak/resources/webapps.py,sha256=6rJDEj0dkQ8bPm0OnNchB7_Ec4eN9FBCUhD8r7YyX5U,13719
 peak/resources/workflows.py,sha256=ePDis4MFa7HZd2PCEt2Y-WfpdFY1hsOmJStiXxccVwc,52753
@@ -111,8 +111,8 @@ peak/tools/logging/log_level.py,sha256=Xtv4zZnYk4QBix_MfCRGim4yJbv7w2tTbNKe_s4Qm
 peak/tools/logging/logger.py,sha256=Jow5D3hqek-n0kHWNeL2nKbC9NfpfkWFT_FmPp-skTw,15653
 peak/tools/logging/utils.py,sha256=qmjdBbCc1Y6JDbSgNXfdnLnpkx7fb9kPw9u8uO3AAII,3330
 peak/validators.py,sha256=_nVAXF_AB443voKNjyQsCTQBAVVAqn7xCbCbBguohIs,2715
-peak_sdk-1.8.0.dist-info/LICENSE,sha256=W0jszenKx7YdFA7BDnyg8xDKXzCP8AperJb_PHh9paQ,11340
-peak_sdk-1.8.0.dist-info/METADATA,sha256=PHdlN3lVNZpaiiEQcnLOHXHqUYMs3l4mi3E7oms8sKs,7139
-peak_sdk-1.8.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-peak_sdk-1.8.0.dist-info/entry_points.txt,sha256=zHCEjuOTjkfmqivgEZQsPGm4zFA4W3Q_vKCjPr7W6lE,47
-peak_sdk-1.8.0.dist-info/RECORD,,
+peak_sdk-1.9.0.dist-info/LICENSE,sha256=W0jszenKx7YdFA7BDnyg8xDKXzCP8AperJb_PHh9paQ,11340
+peak_sdk-1.9.0.dist-info/METADATA,sha256=GulBZwrr94fmLmqWp0NCIMt8CK7mILEV-Jn8B4FJ9EY,7139
+peak_sdk-1.9.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+peak_sdk-1.9.0.dist-info/entry_points.txt,sha256=zHCEjuOTjkfmqivgEZQsPGm4zFA4W3Q_vKCjPr7W6lE,47
+peak_sdk-1.9.0.dist-info/RECORD,,