import boto3
from dotenv import dotenv_values
from fastapi import UploadFile

from app.util.file_util import write_file_env


def _create_session(config: dict):
    return boto3.Session(
        aws_access_key_id=config['AWS_ACCESS_KEY_ID'],
        aws_secret_access_key=config['AWS_SECRET_ACCESS_KEY'],
        aws_session_token=config['AWS_SESSION_TOKEN'],
        region_name=config['AWS_REGION'],
    )


def get_session(path: str, file: UploadFile):
    write_file_env(path, file)
    config = dotenv_values(dotenv_path=path)
    return _create_session(config)