Source code for io_collection.keys.copy_key

import shutil
from pathlib import Path

import boto3




def copy_key(location: str, old_key: str, new_key: str) -> None:
    """
    Copy object key at specified location.

    Method will copy the object in an S3 bucket if the location begins with the
    **s3://** protocol, otherwise it assumes the location is a local path.

    Parameters
    ----------
    location
        Object location (local path or S3 bucket).
    old_key
        Old object key.
    new_key
        New object key.
    """

    if location[:5] == "s3://":
        _copy_key_on_s3(location[5:], old_key, new_key)
    else:
        _copy_key_on_fs(location, old_key, new_key)



def _copy_key_on_fs(path: str, old_key: str, new_key: str) -> None:
    """
    Copy object key on local file system.

    Parameters
    ----------
    path
        Local object path.
    old_key
        Old object key.
    new_key
        New object key.
    """

    full_old_path = Path(path) / old_key
    full_new_path = Path(path) / new_key
    full_new_path.parent.mkdir(parents=True, exist_ok=True)
    shutil.copyfile(full_old_path, full_new_path)


def _copy_key_on_s3(bucket: str, old_key: str, new_key: str) -> None:
    """
    Copy object key in AWS S3 bucket.

    Parameters
    ----------
    bucket
        AWS S3 bucket name.
    old_key
        Old object key.
    new_key
        New object key.
    """

    s3_client = boto3.client("s3")
    s3_client.copy_object(Bucket=bucket, CopySource=f"{bucket}/{old_key}", Key=new_key)