Spaces:
Sleeping
Sleeping
File size: 759 Bytes
2f2406a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
import logging
import boto3
from botocore.exceptions import ClientError
import os
ACCESS_KEY_ID = os.environ.get("ACCESS_KEY_ID")
SECRET_ACCESS_KEY = os.environ.get("SECRET_ACCESS_KEY")
session = boto3.Session(ACCESS_KEY_ID, SECRET_ACCESS_KEY)
def upload_file(file_name, bucket, folder, object_name=None):
# If S3 object_name was not specified, use file_name
if object_name is None:
object_name = os.path.basename(file_name)
# Upload the file
s3_client = session.client('s3')
try:
response = s3_client.upload_file(file_name, bucket, f"{folder}/"+object_name)
except ClientError as e:
logging.error(e)
return False
url = f'{os.environ.get("RESULT_URL")}{folder}/{object_name}'
return url |