mirror of
https://github.com/Monadical-SAS/reflector.git
synced 2025-12-21 12:49:06 +00:00
52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
import sys
|
|
|
|
import boto3
|
|
import botocore
|
|
|
|
from .log_utils import logger
|
|
from .run_utils import config
|
|
|
|
BUCKET_NAME = config["DEFAULT"]["BUCKET_NAME"]
|
|
|
|
s3 = boto3.client('s3',
|
|
aws_access_key_id=config["DEFAULT"]["AWS_ACCESS_KEY"],
|
|
aws_secret_access_key=config["DEFAULT"]["AWS_SECRET_KEY"])
|
|
|
|
|
|
def upload_files(files_to_upload):
|
|
"""
|
|
Upload a list of files to the configured S3 bucket
|
|
:param files_to_upload: List of files to upload
|
|
:return: None
|
|
"""
|
|
for KEY in files_to_upload:
|
|
logger.info("Uploading file " + KEY)
|
|
try:
|
|
s3.upload_file(KEY, BUCKET_NAME, KEY)
|
|
except botocore.exceptions.ClientError as e:
|
|
print(e.response)
|
|
|
|
|
|
def download_files(files_to_download):
|
|
"""
|
|
Download a list of files from the configured S3 bucket
|
|
:param files_to_download: List of files to download
|
|
:return: None
|
|
"""
|
|
for KEY in files_to_download:
|
|
logger.info("Downloading file " + KEY)
|
|
try:
|
|
s3.download_file(BUCKET_NAME, KEY, KEY)
|
|
except botocore.exceptions.ClientError as e:
|
|
if e.response['Error']['Code'] == "404":
|
|
print("The object does not exist.")
|
|
else:
|
|
raise
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if sys.argv[1] == "download":
|
|
download_files([sys.argv[2]])
|
|
elif sys.argv[1] == "upload":
|
|
upload_files([sys.argv[2]])
|