terralab-cli 0.0.1__tar.gz

terralab_cli-0.0.1/LICENSE

@@ -0,0 +1,29 @@
+BSD 3-Clause License
+
+Copyright (c) 2024, Broad Institute
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+1. Redistributions of source code must retain the above copyright notice, this
+   list of conditions and the following disclaimer.
+
+2. Redistributions in binary form must reproduce the above copyright notice,
+   this list of conditions and the following disclaimer in the documentation
+   and/or other materials provided with the distribution.
+
+3. Neither the name of the copyright holder nor the names of its
+   contributors may be used to endorse or promote products derived from
+   this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

terralab_cli-0.0.1/PKG-INFO

@@ -0,0 +1,132 @@
+Metadata-Version: 2.1
+Name: terralab-cli
+Version: 0.0.1
+Summary: Command line interface for interacting with the Terra Scientific Pipelines Service, or Teaspoons.
+Author: Terra Scientific Services
+Author-email: teaspoons-developers@broadinstitute.org
+Requires-Python: >=3.9,<4.0
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: PyJWT (>=2.9.0,<3.0.0)
+Requires-Dist: click (>=8.1.7,<9.0.0)
+Requires-Dist: oauth2-cli-auth (>=1.5.0,<2.0.0)
+Requires-Dist: python-dotenv (>=1.0.1,<2.0.0)
+Requires-Dist: terra-scientific-pipelines-service-api-client (==0.0.0)
+Description-Content-Type: text/markdown
+
+# Teaspoons CLI
+
+[![Quality Gate Status](https://sonarcloud.io/api/project_badges/measure?project=DataBiosphere_terra-scientific-pipelines-service-cli&metric=alert_status)](https://sonarcloud.io/summary/new_code?id=DataBiosphere_terra-scientific-pipelines-service-cli)
+
+
+## Python CLI structure
+The CLI code is structured as follows:
+```
+terra-scientific-pipelines-service-cli
+├── terralab
+│   └── commands
+│   │   └── __init__.py
+│   │   └── auth_commands.py
+│   │   └── pipelines_commands.py
+│   └── logic
+│   │   └── __init__.py
+│   │   └── auth_logic.py
+│   │   └── pipelines_logic.py
+│   └── __init__.py
+│   └── auth_helper.py
+│   └── cli.py
+│   └── client.py
+│   └── config.py
+│   └── teaspoons
+├── tests
+│   └── commands
+│   │   └── test_auth_commands.py
+│   │   └── test_pipelines_commands.py
+│   └── logic
+│   │   └── test_auth_logic.py
+│   │   └── test_pipelines_logic.py
+│   └── __init__.py
+│   └── auth_helper.py
+│   └── cli.py
+│   └── client.py
+│   └── config.py
+│   └── terralab
+├── .gitignore
+├── .terralab-cli-config
+├── poetry.lock
+├── pyproject.toml
+├── README.md
+```
+
+In the `terralab` directory, we have the following files and subdirectories:
+- `auth_helper.py` contains the code for authenticating with the Terra Scientific Pipelines Service (Terra, via b2c).
+- `cli.py` assembles the CLI sub-modules that are defined in `commands/`.
+- `client.py` contains the code for wrapping API calls to the Terra Scientific Pipelines Service.
+- `config.py` contains the code for managing the CLI configuration via environment variables.
+- `terralab` is the entrypoint for the CLI. It contains the main function that is called when the CLI is run.
+- `utils.py` contains utility functions that are used across the CLI.
+- The `commands` directory contains the CLI sub-modules. This is effectively the controller layer for the CLI.
+- The `logic` directory contains the business logic for the CLI, including calling Terra Scientific Pipelines Service APIs via the thin `teaspoons_client`, 
+which is autogenerated and published by the [Terra Scientific Pipelines Service repository](https://github.com/DataBiosphere/terra-scientific-pipelines-service).
+
+In the `tests` directory, we have test files that can be run with pytest.
+
+## Using the CLI
+For now, the CLI requires poetry to be installed to run. See the [Development](#development) section for instructions on how to install poetry.
+
+To run the CLI, navigate to the root (terra-scientific-pipelines-service-cli) directory and run the following command:
+```bash
+./terralab COMMAND [ARGS]
+```
+
+For example, to authenticate with Terralab (via Terra b2c), run the following command:
+```bash
+./terralab auth login
+```
+
+To list the pipelines you can run using Terralab, run the following command:
+```bash
+./terralab pipelines list
+```
+
+See WIP documentation for the CLI [here](https://docs.google.com/document/d/1ovbcHCzdyuC8RjFfkVJZiuDTQ_UAVrglSxSGaZwppoY/edit?tab=t.0#heading=h.jfsr3j3x0zjr).
+
+
+## Development
+You'll need to have poetry installed to manage python dependencies. Instructions for installing poetry can be found [here](https://python-poetry.org/docs/).
+
+To install the CLI locally, run the following commands from the root project directory:
+```
+poetry lock      # only needed if you updated dependencies in pyproject.toml
+poetry install
+```
+You do not need to re-run these commands each time you update code locally, unless you've added dependencies in pyproject.toml.
+
+If you do update dependencies in `pyproject.toml`, run `poetry lock` and check in the resulting changes to `poetry.lock` along with the rest of 
+your code changes.
+
+To run the tests, execute the following command from the root project (terra-scientific-pipelines-service-cli) directory:
+```bash
+poetry run pytest
+```
+
+To run tests with a coverage report printed to the terminal:
+```bash
+poetry run pytest --cov-report term --cov=terralab
+```
+
+To run the formatter, execute the following command from the root project directory:
+```bash
+poetry run black .
+```
+
+To run the linter with fixes, execute the following command from the root project directory:
+```bash
+poetry run ruff check --fix
+```
+To run the linter as a check without fixes, omit the `--fix` flag.
+

terralab_cli-0.0.1/README.md

@@ -0,0 +1,111 @@
+# Teaspoons CLI
+
+[![Quality Gate Status](https://sonarcloud.io/api/project_badges/measure?project=DataBiosphere_terra-scientific-pipelines-service-cli&metric=alert_status)](https://sonarcloud.io/summary/new_code?id=DataBiosphere_terra-scientific-pipelines-service-cli)
+
+
+## Python CLI structure
+The CLI code is structured as follows:
+```
+terra-scientific-pipelines-service-cli
+├── terralab
+│   └── commands
+│   │   └── __init__.py
+│   │   └── auth_commands.py
+│   │   └── pipelines_commands.py
+│   └── logic
+│   │   └── __init__.py
+│   │   └── auth_logic.py
+│   │   └── pipelines_logic.py
+│   └── __init__.py
+│   └── auth_helper.py
+│   └── cli.py
+│   └── client.py
+│   └── config.py
+│   └── teaspoons
+├── tests
+│   └── commands
+│   │   └── test_auth_commands.py
+│   │   └── test_pipelines_commands.py
+│   └── logic
+│   │   └── test_auth_logic.py
+│   │   └── test_pipelines_logic.py
+│   └── __init__.py
+│   └── auth_helper.py
+│   └── cli.py
+│   └── client.py
+│   └── config.py
+│   └── terralab
+├── .gitignore
+├── .terralab-cli-config
+├── poetry.lock
+├── pyproject.toml
+├── README.md
+```
+
+In the `terralab` directory, we have the following files and subdirectories:
+- `auth_helper.py` contains the code for authenticating with the Terra Scientific Pipelines Service (Terra, via b2c).
+- `cli.py` assembles the CLI sub-modules that are defined in `commands/`.
+- `client.py` contains the code for wrapping API calls to the Terra Scientific Pipelines Service.
+- `config.py` contains the code for managing the CLI configuration via environment variables.
+- `terralab` is the entrypoint for the CLI. It contains the main function that is called when the CLI is run.
+- `utils.py` contains utility functions that are used across the CLI.
+- The `commands` directory contains the CLI sub-modules. This is effectively the controller layer for the CLI.
+- The `logic` directory contains the business logic for the CLI, including calling Terra Scientific Pipelines Service APIs via the thin `teaspoons_client`, 
+which is autogenerated and published by the [Terra Scientific Pipelines Service repository](https://github.com/DataBiosphere/terra-scientific-pipelines-service).
+
+In the `tests` directory, we have test files that can be run with pytest.
+
+## Using the CLI
+For now, the CLI requires poetry to be installed to run. See the [Development](#development) section for instructions on how to install poetry.
+
+To run the CLI, navigate to the root (terra-scientific-pipelines-service-cli) directory and run the following command:
+```bash
+./terralab COMMAND [ARGS]
+```
+
+For example, to authenticate with Terralab (via Terra b2c), run the following command:
+```bash
+./terralab auth login
+```
+
+To list the pipelines you can run using Terralab, run the following command:
+```bash
+./terralab pipelines list
+```
+
+See WIP documentation for the CLI [here](https://docs.google.com/document/d/1ovbcHCzdyuC8RjFfkVJZiuDTQ_UAVrglSxSGaZwppoY/edit?tab=t.0#heading=h.jfsr3j3x0zjr).
+
+
+## Development
+You'll need to have poetry installed to manage python dependencies. Instructions for installing poetry can be found [here](https://python-poetry.org/docs/).
+
+To install the CLI locally, run the following commands from the root project directory:
+```
+poetry lock      # only needed if you updated dependencies in pyproject.toml
+poetry install
+```
+You do not need to re-run these commands each time you update code locally, unless you've added dependencies in pyproject.toml.
+
+If you do update dependencies in `pyproject.toml`, run `poetry lock` and check in the resulting changes to `poetry.lock` along with the rest of 
+your code changes.
+
+To run the tests, execute the following command from the root project (terra-scientific-pipelines-service-cli) directory:
+```bash
+poetry run pytest
+```
+
+To run tests with a coverage report printed to the terminal:
+```bash
+poetry run pytest --cov-report term --cov=terralab
+```
+
+To run the formatter, execute the following command from the root project directory:
+```bash
+poetry run black .
+```
+
+To run the linter with fixes, execute the following command from the root project directory:
+```bash
+poetry run ruff check --fix
+```
+To run the linter as a check without fixes, omit the `--fix` flag.

terralab_cli-0.0.1/pyproject.toml

@@ -0,0 +1,71 @@
+[tool.poetry]
+name = "terralab-cli"
+version = "0.0.1"
+description = "Command line interface for interacting with the Terra Scientific Pipelines Service, or Teaspoons."
+authors = ["Terra Scientific Services <teaspoons-developers@broadinstitute.org>"]
+readme = "README.md"
+packages = [
+    {include = "terralab"}
+]
+
+[tool.poetry.dependencies]
+python = "^3.9"
+terra-scientific-pipelines-service-api-client = "0.0.0"
+python-dotenv = "^1.0.1"
+click = "^8.1.7"
+oauth2-cli-auth = "^1.5.0"
+PyJWT = "^2.9.0"
+
+[tool.poetry.group.dev.dependencies]
+pytest = "^8.3.3"
+pytest-cov = "^3.0.0"
+mockito = ">=1.4.0"
+pytest-mockito = ">=0.0.4"
+ruff = "^0.7.1"
+black = "^24.8.0"
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"
+
+[tool.pytest.ini_options]
+pythonpath = "terralab"
+
+[tool.coverage.run]
+branch = true
+relative_files = true  # allows sonarcloud GHA step to process the coverage files
+
+[tool.ruff]
+# Exclude a variety of commonly ignored directories.
+exclude = [
+    ".bzr",
+    ".direnv",
+    ".eggs",
+    ".git",
+    ".git-rewrite",
+    ".hg",
+    ".ipynb_checkpoints",
+    ".mypy_cache",
+    ".nox",
+    ".pants.d",
+    ".pyenv",
+    ".pytest_cache",
+    ".pytype",
+    ".ruff_cache",
+    ".svn",
+    ".tox",
+    ".venv",
+    ".vscode",
+    "__pypackages__",
+    "_build",
+    "buck-out",
+    "build",
+    "dist",
+    "node_modules",
+    "site-packages",
+    "venv",
+    "generated"
+]
+# Same as Black.
+line-length = 88
+indent-width = 4

terralab_cli-0.0.1/terralab/__init__.py

@@ -0,0 +1,10 @@
+"""
+Terralab: A CLI for the Terra Scientific Pipelines Service (Teaspoons).
+
+This package provides a command-line interface to interact with the Teaspoons service,
+built on top of an autogenerated thin client.
+"""
+
+__version__ = "0.0.1"
+__author__ = "Terra Scientific Services"
+__email__ = "teaspoons-developers@broadinstitute.org"

terralab_cli-0.0.1/terralab/auth_helper.py

@@ -0,0 +1,89 @@
+# auth_helper.py
+
+import jwt
+import logging
+import os
+import typing as t
+
+from oauth2_cli_auth import (
+    OAuth2ClientInfo,
+    OAuthCallbackHttpServer,
+    get_auth_url,
+    open_browser,
+    exchange_code_for_access_token,
+)
+
+from config import CliConfig
+
+
+LOGGER = logging.getLogger(__name__)
+
+
+def get_access_token_with_browser_open(client_info: OAuth2ClientInfo) -> str:
+    """
+    Note: this is overridden from the oauth2-cli-auth library to use a custom auth url
+
+    Provides a simplified API to:
+
+    - Spin up the callback server
+    - Open the browser with the authorization URL
+    - Wait for the code to arrive
+    - Get access token from code
+
+    :param client_info: Client Info for Oauth2 Interaction
+    :param server_port: Port of the local web server to spin up
+    :return: Access Token
+    """
+    server_port = CliConfig().server_port
+    callback_server = OAuthCallbackHttpServer(server_port)
+    auth_url = get_auth_url(client_info, callback_server.callback_url)
+    open_browser(f"{auth_url}&prompt=login")
+    code = callback_server.wait_for_code()
+    if code is None:
+        raise ValueError("No code could be obtained from browser callback page")
+    return exchange_code_for_access_token(
+        client_info, callback_server.callback_url, code
+    )
+
+
+def _validate_token(token: str) -> bool:
+    try:
+        # Attempt to read the token to ensure it is valid.  If it isn't, the file will be removed and None will be returned.
+        # Note: to simplify, not worrying about the signature of the token since that will be verified by the backend services
+        # This is just to ensure the token is not expired
+        jwt.decode(token, options={"verify_signature": False, "verify_exp": True})
+        return True
+    except jwt.ExpiredSignatureError:
+        LOGGER.debug("Token expired")
+        return False
+    except Exception as e:
+        LOGGER.error(f"Error validating token: {e}")
+        return False
+
+
+def _clear_local_token(token_file: str):
+    try:
+        os.remove(token_file)
+    except FileNotFoundError:
+        LOGGER.debug("No local token found to clean up")
+
+
+def _load_local_token(token_file: str) -> t.Optional[str]:
+    try:
+        with open(token_file, "r") as f:
+            token = f.read()
+            if _validate_token(token):
+                return token
+            else:
+                return None
+
+    except FileNotFoundError:
+        _clear_local_token(token_file)
+        return None
+
+
+def _save_local_token(token_file: str, token: str):
+    # Create the containing directory if it doesn't exist
+    os.makedirs(os.path.dirname(token_file), exist_ok=True)
+    with open(token_file, "w") as f:
+        f.write(token)

terralab_cli-0.0.1/terralab/cli.py

@@ -0,0 +1,37 @@
+# cli.py
+
+import click
+import logging
+
+from terralab import __version__, log
+from terralab.commands.auth_commands import auth
+from terralab.commands.pipelines_commands import pipelines
+
+
+# Context settings for commands, for overwriting some click defaults
+CONTEXT_SETTINGS = dict(help_option_names=["-h", "--help"])
+
+LOGGER = logging.getLogger(__name__)
+
+
+@click.group(name="terralab", context_settings=CONTEXT_SETTINGS)
+@click.version_option(__version__)
+@click.option(
+    "--debug",
+    is_flag=True,
+    help="DEBUG-level logging",
+)
+def cli(debug):
+    log.configure_logging(debug)
+    LOGGER.debug(
+        "Log level set to: %s", logging.getLevelName(logging.getLogger().level)
+    )
+
+
+cli.add_command(auth)
+cli.add_command(pipelines)
+# will add runs_app later
+
+
+if __name__ == "__main__":
+    cli()

terralab_cli-0.0.1/terralab/client.py

@@ -0,0 +1,43 @@
+# client.py
+
+import logging
+
+from teaspoons_client import Configuration, ApiClient
+from config import CliConfig
+from auth_helper import (
+    _load_local_token,
+    _validate_token,
+    _save_local_token,
+    get_access_token_with_browser_open,
+)
+
+
+LOGGER = logging.getLogger(__name__)
+
+
+def _get_api_client(token: str, api_url: str) -> ApiClient:
+    api_config = Configuration()
+    api_config.host = api_url
+    api_config.access_token = token
+    return ApiClient(configuration=api_config)
+
+
+class ClientWrapper:
+    """
+    Wrapper to ensure that the user is authenticated before running the callback and that provides the low level api client to be used
+    by subsequent commands
+    """
+
+    def __enter__(self):
+        cli_config = CliConfig()  # initialize the config from environment variables
+        token = _load_local_token(cli_config.token_file)
+        if not (token and _validate_token(token)):
+            LOGGER.info("No valid token found. Logging you in...")
+            token = get_access_token_with_browser_open(cli_config.client_info)
+        _save_local_token(cli_config.token_file, token)
+
+        return _get_api_client(token, cli_config.config["TEASPOONS_API_URL"])
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        # no action needed
+        pass

terralab_cli-0.0.1/terralab/commands/auth_commands.py

@@ -0,0 +1,22 @@
+# commands/auth_commands.py
+
+import click
+
+from logic import auth_logic
+
+
+@click.group()
+def auth():
+    """Commands for authenticating to Terralab"""
+
+
+@auth.command()
+def login():
+    """Authenticate with Terralab via browser login to Terra b2c"""
+    auth_logic.check_local_token_and_fetch_if_needed()
+
+
+@auth.command()
+def logout():
+    """Clear the local authentication token"""
+    auth_logic.clear_local_token()

terralab_cli-0.0.1/terralab/commands/pipelines_commands.py

@@ -0,0 +1,35 @@
+# commands/pipelines_commands.py
+
+import click
+import logging
+
+from logic import pipelines_logic
+from utils import _pretty_print, handle_api_exceptions
+
+LOGGER = logging.getLogger(__name__)
+
+
+@click.group()
+def pipelines():
+    """Commands for running Teaspooons pipelines"""
+
+
+@pipelines.command()
+@handle_api_exceptions
+def list():
+    """List all available pipelines"""
+    pipelines_list = pipelines_logic.list_pipelines()
+    LOGGER.info(
+        f"Found {len(pipelines_list)} available pipeline{'' if len(pipelines_list) == 1 else 's'}:"
+    )
+    for pipeline in pipelines_list:
+        LOGGER.info(f"- {pipeline.pipeline_name} ({pipeline.description})")
+
+
+@pipelines.command()
+@click.argument("pipeline_name")
+@handle_api_exceptions
+def get_info(pipeline_name: str):
+    """Get information about a specific pipeline"""
+    pipeline_info = pipelines_logic.get_pipeline_info(pipeline_name)
+    _pretty_print(pipeline_info)

terralab_cli-0.0.1/terralab/config.py

@@ -0,0 +1,24 @@
+# config.py
+
+from pathlib import Path
+from oauth2_cli_auth import OAuth2ClientInfo
+from dotenv import dotenv_values
+
+
+class CliConfig:
+    """A class to hold configuration information for the CLI"""
+
+    def __init__(self, config_file="../.terralab-cli-config"):
+        self.config = dotenv_values(config_file)
+
+        self.client_info = OAuth2ClientInfo.from_oidc_endpoint(
+            self.config["OAUTH_OPENID_CONFIGURATION_URI"],
+            client_id=self.config["OAUTH_CLIENT_ID"],
+            scopes=[f"openid+email+profile+{self.config['OAUTH_CLIENT_ID']}"],
+        )
+
+        self.server_port = int(self.config["SERVER_PORT"])
+
+        self.token_file = (
+            f'{Path.home()}/{self.config["LOCAL_STORAGE_PATH"]}/access_token'
+        )

terralab_cli-0.0.1/terralab/log.py

@@ -0,0 +1,8 @@
+# log.py
+
+import logging
+
+
+def configure_logging(debug: bool):
+    log_level = logging.DEBUG if debug else logging.INFO
+    logging.basicConfig(level=log_level, format="%(message)s")

terralab_cli-0.0.1/terralab/logic/auth_logic.py

@@ -0,0 +1,31 @@
+# logic/auth_logic.py
+
+import logging
+from auth_helper import (
+    get_access_token_with_browser_open,
+    _validate_token,
+    _save_local_token,
+    _load_local_token,
+    _clear_local_token,
+)
+from config import CliConfig
+
+LOGGER = logging.getLogger(__name__)
+
+
+def check_local_token_and_fetch_if_needed():
+    """Authenticate with Teaspoons via browser login to Terra b2c"""
+    cli_config = CliConfig()  # initialize the config from environment variables
+    token = _load_local_token(cli_config.token_file)
+    if token and _validate_token(token):
+        LOGGER.info("Already authenticated")
+        return
+    token = get_access_token_with_browser_open(cli_config.client_info)
+    _save_local_token(cli_config.token_file, token)
+
+
+def clear_local_token():
+    """Clear the local authentication token"""
+    cli_config = CliConfig()  # initialize the config from environment variables
+    _clear_local_token(cli_config.token_file)
+    LOGGER.info("Logged out")

terralab_cli-0.0.1/terralab/logic/pipelines_logic.py

@@ -0,0 +1,27 @@
+# logic/pipelines_logic.py
+
+from teaspoons_client import PipelinesApi
+from teaspoons_client.models.pipeline_with_details import PipelineWithDetails
+from teaspoons_client.models.pipeline import Pipeline
+
+from client import ClientWrapper
+
+
+def list_pipelines() -> list[Pipeline]:
+    """List all pipelines, returning a list of dictionaries."""
+    with ClientWrapper() as api_client:
+        pipeline_client = PipelinesApi(api_client=api_client)
+        pipelines = pipeline_client.get_pipelines()
+
+        result = []
+        for pipeline in pipelines.results:
+            result.append(pipeline)
+
+        return result
+
+
+def get_pipeline_info(pipeline_name: str) -> PipelineWithDetails:
+    """Get the details of a pipeline, returning a dictionary."""
+    with ClientWrapper() as api_client:
+        pipeline_client = PipelinesApi(api_client=api_client)
+        return pipeline_client.get_pipeline_details(pipeline_name=pipeline_name)

terralab_cli-0.0.1/terralab/terralab

@@ -0,0 +1,6 @@
+#!/usr/bin/env poetry run python
+
+from cli import cli
+
+if __name__ == '__main__':
+    cli()

terralab_cli-0.0.1/terralab/utils.py

@@ -0,0 +1,38 @@
+# utils.py
+
+import json
+import logging
+
+from functools import wraps
+from pydantic import BaseModel
+
+from teaspoons_client.exceptions import ApiException
+
+
+LOGGER = logging.getLogger(__name__)
+
+
+def _pretty_print(obj: BaseModel):
+    """
+    Prints a pydantic model in a pretty format to the console
+    """
+    try:
+        LOGGER.info(json.dumps(obj.model_dump(), indent=4))
+    except Exception:
+        LOGGER.error(obj)
+
+
+def handle_api_exceptions(func):
+    @wraps(func)
+    def wrapper(*args, **kwargs):
+        try:
+            return func(*args, **kwargs)
+        except ApiException as e:
+            formatted_message = f"API call failed with status code {e.status} ({e.reason}): {json.loads(e.body)['message']}"
+            LOGGER.error(formatted_message)
+            exit(1)
+        except Exception as e:
+            LOGGER.error(str(e))
+            exit(1)
+
+    return wrapper