File size: 567 Bytes
02109ec
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
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)