Source code for s3pathlib.core.resolve_s3_client

# -*- coding: utf-8 -*-

"""
Resolve what s3 client to use for API call.
"""

import typing as T

from ..aws import Context

if T.TYPE_CHECKING:  # pragma: no cover
    from boto_session_manager import BotoSesManager
    from mypy_boto3_s3 import S3Client



[docs]
def resolve_s3_client(
    context: Context,
    bsm: T.Optional[T.Union["BotoSesManager", "S3Client"]] = None,
) -> "S3Client":
    """
    Figure out the final boto session to use for API call.
    If ``BotoSesManager`` is defined, then prioritize to use it.
    If bsm is an pre-defined s3 client, then use it.
    """
    if bsm is None:
        return context.s3_client
    else:
        try:
            return bsm.s3_client
        except AttributeError:
            return bsm