azaion/ai-training
1
0
Fork 0
mirror of https://github.com/azaion/ai-training.git synced 2026-04-22 11:16:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

update cdn manager

Browse Source 
download big engine file
This commit is contained in:
Alex Bezdieniezhnykh
2025-04-18 13:13:19 +03:00
parent adcbdb4915
commit 9017fa7469
3 changed files with 12 additions and 301 deletions
Download Patch File Download Diff File Expand all files Collapse all files
cdn_manager.py
+4 -30
View File
@@ -19,11 +19,11 @@ class CDNManager:
def __init__(self, credentials: CDNCredentials):
self.creds = credentials
self.download_client = boto3.client('s3', endpoint_url=self.creds.host,
aws_access_key_id=self.creds.downloader_access_key,
aws_secret_access_key=self.creds.downloader_access_secret)
aws_access_key_id=self.creds.downloader_access_key,
aws_secret_access_key=self.creds.downloader_access_secret)
self.upload_client = boto3.client('s3', endpoint_url=self.creds.host,
aws_access_key_id=self.creds.uploader_access_key,
aws_secret_access_key=self.creds.uploader_access_secret)
aws_access_key_id=self.creds.uploader_access_key,
aws_secret_access_key=self.creds.uploader_access_secret)
def upload(self, bucket: str, filename: str, file_bytes: bytearray):
try:
@@ -42,29 +42,3 @@ class CDNManager:
except Exception as e:
print(e)
return False
config_file = os.path.join('downloader_config.yaml')
with open(config_file, 'r') as f:
config = yaml.safe_load(f)
cdn_manager = CDNManager(CDNCredentials(
config['host'],
config['downloader_access_key'],
config['downloader_access_secret'],
config['uploader_access_key'],
config['uploader_access_secret']
))
input_action = sys.argv[1]
input_bucket = sys.argv[2]
input_filename = sys.argv[3]
if len(sys.argv) > 4: # 0 is this script's path, hence 5 args is max
input_path = sys.argv[4]
if input_action == 'upload':
with open(input_path, 'rb') as f:
cdn_manager.upload(input_bucket, input_filename, bytearray(f.read()))
elif input_action == 'download':
cdn_manager.download(input_bucket, input_filename)