148 lines
4.5 KiB
Python
148 lines
4.5 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
Upload GUI installer to KS3 object storage
|
|
- Target directory: installers/
|
|
- Set proper HTTP headers for browser download
|
|
"""
|
|
import os
|
|
import sys
|
|
import logging
|
|
from pathlib import Path
|
|
|
|
try:
|
|
from ks3.connection import Connection
|
|
except ImportError:
|
|
print("ERROR: ks3sdk not installed. Please run: pip install ks3sdk")
|
|
sys.exit(1)
|
|
|
|
# Configure logging
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format='%(asctime)s [%(levelname)s] %(message)s'
|
|
)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# KS3 Configuration (using actual project config)
|
|
KS3_ACCESS_KEY = 'AKLT0Ey7Nq7ZSXykki0X0RGG'
|
|
KS3_SECRET_KEY = 'OONxkt9wwJa1FIco21vCbirs1HB6AGzDWdRyV0k2'
|
|
KS3_ENDPOINT = 'ks3-cn-guangzhou.ksyuncs.com' # Guangzhou region
|
|
KS3_BUCKET = 'shuidrop-chat-server'
|
|
KS3_IS_SECURE = True # Use HTTPS
|
|
KS3_PREFIX = 'installers/'
|
|
|
|
|
|
def get_ks3_connection():
|
|
"""Get KS3 connection"""
|
|
try:
|
|
connection = Connection(
|
|
KS3_ACCESS_KEY,
|
|
KS3_SECRET_KEY,
|
|
host=KS3_ENDPOINT,
|
|
is_secure=KS3_IS_SECURE,
|
|
timeout=300,
|
|
)
|
|
logger.info(f"KS3 connection established: {KS3_ENDPOINT}")
|
|
return connection
|
|
except Exception as e:
|
|
logger.error(f"KS3 connection failed: {e}")
|
|
return None
|
|
|
|
|
|
def find_latest_installer():
|
|
"""Find the latest generated installer"""
|
|
project_root = Path(__file__).parent.parent.parent
|
|
installer_dir = project_root / 'installer' / 'output'
|
|
|
|
if not installer_dir.exists():
|
|
logger.error(f"Installer directory not found: {installer_dir}")
|
|
return None
|
|
|
|
installers = list(installer_dir.glob('*.exe'))
|
|
if not installers:
|
|
logger.error(f"No installer files found in: {installer_dir}")
|
|
return None
|
|
|
|
latest_installer = max(installers, key=lambda p: p.stat().st_mtime)
|
|
file_size_mb = latest_installer.stat().st_size / 1024 / 1024
|
|
|
|
logger.info(f"Found installer: {latest_installer.name}")
|
|
logger.info(f"File size: {file_size_mb:.2f} MB")
|
|
|
|
return latest_installer
|
|
|
|
|
|
def upload_installer(connection, installer_path):
|
|
"""Upload installer to KS3"""
|
|
try:
|
|
bucket = connection.get_bucket(KS3_BUCKET)
|
|
ks3_key = f"{KS3_PREFIX}{installer_path.name}"
|
|
|
|
logger.info(f"Starting upload to KS3...")
|
|
logger.info(f"Target path: {ks3_key}")
|
|
|
|
key = bucket.new_key(ks3_key)
|
|
|
|
# Upload file with public read permission (reference: bulk_upload_static_to_ks3.py)
|
|
key.set_contents_from_filename(
|
|
str(installer_path),
|
|
headers={
|
|
'Content-Type': 'application/octet-stream',
|
|
'Content-Disposition': f'attachment; filename="{installer_path.name}"',
|
|
'Cache-Control': 'public, max-age=3600',
|
|
'x-kss-storage-class': 'STANDARD',
|
|
'x-kss-acl': 'public-read' # Set public read permission in headers
|
|
},
|
|
policy='public-read' # Set ACL policy
|
|
)
|
|
|
|
# Generate download URL (using KS3 third-level domain format)
|
|
# Format: https://{bucket}.{endpoint}/{key}
|
|
protocol = 'https' if KS3_IS_SECURE else 'http'
|
|
download_url = f"{protocol}://{KS3_BUCKET}.{KS3_ENDPOINT}/{ks3_key}"
|
|
|
|
logger.info(f"Upload successful!")
|
|
logger.info(f"Download URL: {download_url}")
|
|
|
|
return download_url
|
|
except Exception as e:
|
|
logger.error(f"Upload failed: {e}")
|
|
import traceback
|
|
logger.error(traceback.format_exc())
|
|
return None
|
|
|
|
|
|
def main():
|
|
"""Main function"""
|
|
logger.info("=" * 70)
|
|
logger.info("Starting GUI installer upload to KS3")
|
|
logger.info(f"KS3 Endpoint: {KS3_ENDPOINT}")
|
|
logger.info(f"Bucket: {KS3_BUCKET}")
|
|
logger.info(f"Target directory: {KS3_PREFIX}")
|
|
logger.info("=" * 70)
|
|
|
|
installer_path = find_latest_installer()
|
|
if not installer_path:
|
|
return 1
|
|
|
|
connection = get_ks3_connection()
|
|
if not connection:
|
|
return 1
|
|
|
|
download_url = upload_installer(connection, installer_path)
|
|
if not download_url:
|
|
return 1
|
|
|
|
logger.info("=" * 70)
|
|
logger.info("Task completed successfully!")
|
|
logger.info(f"Installer: {installer_path.name}")
|
|
logger.info(f"Download URL: {download_url}")
|
|
logger.info("=" * 70)
|
|
|
|
return 0
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main())
|
|
|