airbyte.destinations.util

  1# Copyright (c) 2024 Airbyte, Inc., all rights reserved.
  2"""Destination utilities.
  3
  4For usage examples, see the `airbyte.destinations` module documentation.
  5"""
  6
  7from __future__ import annotations
  8
  9from typing import TYPE_CHECKING, Any
 10
 11from airbyte._executors.util import get_connector_executor
 12from airbyte.destinations.base import Destination
 13
 14
 15if TYPE_CHECKING:
 16    from pathlib import Path
 17
 18    from airbyte.callbacks import ConfigChangeCallback
 19
 20
 21def get_destination(  # noqa: PLR0913 # Too many arguments
 22    name: str,
 23    config: dict[str, Any] | None = None,
 24    *,
 25    config_change_callback: ConfigChangeCallback | None = None,
 26    version: str | None = None,
 27    pip_url: str | None = None,
 28    local_executable: Path | str | None = None,
 29    docker_image: str | bool | None = None,
 30    use_host_network: bool = False,
 31    install_if_missing: bool = True,
 32) -> Destination:
 33    """Get a connector by name and version.
 34
 35    Args:
 36        name: connector name
 37        config: connector config - if not provided, you need to set it later via the set_config
 38            method.
 39        config_change_callback: callback function to be called when the connector config changes.
 40        streams: list of stream names to select for reading. If set to "*", all streams will be
 41            selected. If not provided, you can set it later via the `select_streams()` or
 42            `select_all_streams()` method.
 43        version: connector version - if not provided, the currently installed version will be used.
 44            If no version is installed, the latest available version will be used. The version can
 45            also be set to "latest" to force the use of the latest available version.
 46        pip_url: connector pip URL - if not provided, the pip url will be inferred from the
 47            connector name.
 48        local_executable: If set, the connector will be assumed to already be installed and will be
 49            executed using this path or executable name. Otherwise, the connector will be installed
 50            automatically in a virtual environment.
 51        docker_image: If set, the connector will be executed using Docker. You can specify `True`
 52            to use the default image for the connector, or you can specify a custom image name.
 53            If `version` is specified and your image name does not already contain a tag
 54            (e.g. `my-image:latest`), the version will be appended as a tag (e.g. `my-image:0.1.0`).
 55        use_host_network: If set, along with docker_image, the connector will be executed using
 56            the host network. This is useful for connectors that need to access resources on
 57            the host machine, such as a local database. This parameter is ignored when
 58            `docker_image` is not set.
 59        install_if_missing: Whether to install the connector if it is not available locally. This
 60            parameter is ignored when local_executable is set.
 61    """
 62    return Destination(
 63        name=name,
 64        config=config,
 65        config_change_callback=config_change_callback,
 66        executor=get_connector_executor(
 67            name=name,
 68            version=version,
 69            pip_url=pip_url,
 70            local_executable=local_executable,
 71            docker_image=docker_image,
 72            use_host_network=use_host_network,
 73            install_if_missing=install_if_missing,
 74        ),
 75    )
 76
 77
 78def get_noop_destination() -> Destination:
 79    """Get a devnull (no-op) destination.
 80
 81    This is useful for performance benchmarking of sources, without
 82    adding the overhead of writing data to a real destination.
 83    """
 84    return get_destination(
 85        "destination-dev-null",
 86        config={
 87            "test_destination": {
 88                "test_destination_type": "LOGGING",
 89                "logging_config": {
 90                    "logging_type": "FirstN",
 91                    "max_entry_count": 100,
 92                },
 93            }
 94        },
 95        docker_image=True,
 96    )
 97
 98
 99__all__ = [
100    "get_destination",
101    "get_noop_destination",
102]