From ee94d2f5dbaa0bd3fd0df26871d255ffe5b03bdb Mon Sep 17 00:00:00 2001
From: Alex Bezdieniezhnykh <zxsanny@gmail.com>
Date: Tue, 4 Mar 2025 13:46:49 +0200
Subject: [PATCH] add cdn_manager.py and build script

---
 build/build_downloader.cmd |  5 +++
 build/cdn_manager.py       | 80 ++++++++++++++++++++++++++++++++++++++
 build/requirements.txt     |  3 ++
 3 files changed, 88 insertions(+)
 create mode 100644 build/build_downloader.cmd
 create mode 100644 build/cdn_manager.py
 create mode 100644 build/requirements.txt

diff --git a/build/build_downloader.cmd b/build/build_downloader.cmd
new file mode 100644
index 0000000..d8a0e03
--- /dev/null
+++ b/build/build_downloader.cmd
@@ -0,0 +1,5 @@
+pyinstaller --onefile --collect-all boto3 cdn_manager.py
+move dist\cdn_manager.exe .\cdn_manager.exe
+rmdir /s /q dist
+rmdir /s /q build
+
diff --git a/build/cdn_manager.py b/build/cdn_manager.py
new file mode 100644
index 0000000..60e820b
--- /dev/null
+++ b/build/cdn_manager.py
@@ -0,0 +1,80 @@
+import io
+import sys
+
+import boto3
+import yaml
+import os
+
+
+class CDNCredentials:
+    def __init__(self, host, downloader_access_key, downloader_access_secret, uploader_access_key, uploader_access_secret):
+        self.host = host
+        self.downloader_access_key = downloader_access_key
+        self.downloader_access_secret = downloader_access_secret
+        self.uploader_access_key = uploader_access_key
+        self.uploader_access_secret = uploader_access_secret
+
+
+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)
+        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)
+
+    def upload(self, bucket: str, filename: str, file_bytes: bytearray):
+        try:
+            self.upload_client.upload_fileobj(io.BytesIO(file_bytes), bucket, filename)
+            print(f'uploaded {filename} ({len(file_bytes)} bytes) to the {bucket}')
+            return True
+        except Exception as e:
+            print(e)
+            return False
+
+    def upload(self, bucket: str, filename: str, file_bytes: bytearray):
+        try:
+            self.upload_client.upload
+            self.upload_client.upload_fileobj(io.BytesIO(file_bytes), bucket, filename)
+            print(f'uploaded {filename} ({len(file_bytes)} bytes) to the {bucket}')
+            return True
+        except Exception as e:
+            print(e)
+            return False
+
+    def download(self, bucket: str, filename: str):
+        try:
+            self.download_client.download_file(bucket, filename, filename)
+            print(f'downloaded {filename} from the {bucket} to current folder')
+            return True
+        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)
+
diff --git a/build/requirements.txt b/build/requirements.txt
new file mode 100644
index 0000000..acfabc8
--- /dev/null
+++ b/build/requirements.txt
@@ -0,0 +1,3 @@
+boto3
+pyinstaller
+pyyaml
\ No newline at end of file