"""Connectors."""

from functools import cache

import boto3
from content_utils.connectors.opensearch import LambdaOpensearchConnector

from src import config


@cache
def get_os_connector():
    """Get OpenSearch connector."""
    os_opts = {}
    if config.OPENSEARCH_PORT:
        os_opts.update(port=int(config.OPENSEARCH_PORT))

    return LambdaOpensearchConnector(
        config.OPENSEARCH_ENDPOINT,
        config.app_logger,
        **os_opts,
    )


@cache
def get_sqs_connector():
    """Get SQS connector."""
    return boto3.client('sqs')