mirror of
https://github.com/ByteWelder/Tactility.git
synced 2026-02-18 02:43:15 +00:00
77 lines
2.4 KiB
Python
77 lines
2.4 KiB
Python
import os
|
|
import sys
|
|
import boto3
|
|
|
|
verbose = False
|
|
|
|
if sys.platform == "win32":
|
|
shell_color_red = ""
|
|
shell_color_orange = ""
|
|
shell_color_reset = ""
|
|
else:
|
|
shell_color_red = "\033[91m"
|
|
shell_color_orange = "\033[93m"
|
|
shell_color_reset = "\033[m"
|
|
|
|
def print_warning(message):
|
|
print(f"{shell_color_orange}WARNING: {message}{shell_color_reset}")
|
|
|
|
def print_error(message):
|
|
print(f"{shell_color_red}ERROR: {message}{shell_color_reset}")
|
|
|
|
def print_help():
|
|
print("Usage: python upload-files.py [path] [version] [cloudflareAccountId] [cloudflareTokenName] [cloudflareTokenValue]")
|
|
print("")
|
|
print("Options:")
|
|
print(" --verbose Show extra console output")
|
|
print(" --index-only Upload only index.json")
|
|
|
|
def exit_with_error(message):
|
|
print_error(message)
|
|
sys.exit(1)
|
|
|
|
def main(path: str, version: str, cloudflare_account_id, cloudflare_token_name: str, cloudflare_token_value: str, index_only: bool):
|
|
if not os.path.exists(path):
|
|
exit_with_error(f"Path not found: {path}")
|
|
s3 = boto3.client(
|
|
service_name="s3",
|
|
endpoint_url=f"https://{cloudflare_account_id}.r2.cloudflarestorage.com",
|
|
aws_access_key_id=cloudflare_token_name,
|
|
aws_secret_access_key=cloudflare_token_value,
|
|
region_name="auto"
|
|
)
|
|
files_to_upload = os.listdir(path)
|
|
counter = 1
|
|
total = len(files_to_upload)
|
|
for file_name in files_to_upload:
|
|
if not index_only or file_name == 'index.json':
|
|
object_path = f"firmware/{version}/{file_name}"
|
|
print(f"[{counter}/{total}] Uploading {file_name} to {object_path}")
|
|
file_path = os.path.join(path, file_name)
|
|
try:
|
|
s3.upload_file(file_path, "tactility", object_path)
|
|
except Exception as e:
|
|
exit_with_error(f"Failed to upload {file_name}: {str(e)}")
|
|
counter += 1
|
|
|
|
if __name__ == "__main__":
|
|
print("Tactility CDN Uploader")
|
|
if "--help" in sys.argv:
|
|
print_help()
|
|
sys.exit()
|
|
# Argument validation
|
|
if len(sys.argv) < 6:
|
|
print_help()
|
|
sys.exit()
|
|
if "--verbose" in sys.argv:
|
|
verbose = True
|
|
sys.argv.remove("--verbose")
|
|
main(
|
|
path=sys.argv[1],
|
|
version=sys.argv[2],
|
|
cloudflare_account_id=sys.argv[3],
|
|
cloudflare_token_name=sys.argv[4],
|
|
cloudflare_token_value=sys.argv[5],
|
|
index_only="--index-only" in sys.argv
|
|
)
|