import boto3

from .. import config

client = boto3.client("s3")


def get_object(bucket: str, key: str):
    return client.get_object(
        Bucket=bucket,
        Key=key,
        ExpectedBucketOwner=config.AWS_ACCOUNT_ID,
    )["Body"]


def upload_file(filename: str, bucket: str, key: str):
    return client.upload_file(
        Filename=filename,
        Bucket=bucket,
        Key=key,
        ExtraArgs={"ExpectedBucketOwner": config.AWS_ACCOUNT_ID},
    )