142 lines
4.2 KiB
Python
142 lines
4.2 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
|
|
from ks3.connection import Connection
|
|
|
|
# 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)
|
|
|
|
with open(installer_path, 'rb') as f:
|
|
key.set_contents_from_file(
|
|
f,
|
|
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'
|
|
}
|
|
)
|
|
|
|
key.set_acl('public-read')
|
|
|
|
# Generate download URL (using HTTPS)
|
|
protocol = 'https' if KS3_IS_SECURE else 'http'
|
|
download_url = f"{protocol}://{KS3_ENDPOINT}/{KS3_BUCKET}/{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())
|
|
|