152 lines
4.5 KiB
Python
152 lines
4.5 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
应用打包脚本
|
||
根据 app.yaml 中列出的文件，打包成 zip 文件
|
||
版本号从 version.py 中读取
|
||
"""
|
||
import os
|
||
import yaml
|
||
import zipfile
|
||
from datetime import datetime
|
||
import sys
|
||
|
||
# 添加当前目录到路径，以便导入 version 模块
|
||
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
||
|
||
|
||
def load_app_yaml(yaml_path='app.yaml'):
|
||
"""加载 app.yaml 文件"""
|
||
try:
|
||
with open(yaml_path, 'r', encoding='utf-8') as f:
|
||
return yaml.safe_load(f)
|
||
except Exception as e:
|
||
print(f"[ERROR] 读取 {yaml_path} 失败: {e}")
|
||
return None
|
||
|
||
|
||
def check_files_exist(files, base_dir='.'):
|
||
"""检查文件是否存在"""
|
||
missing_files = []
|
||
existing_files = []
|
||
|
||
for file_path in files:
|
||
full_path = os.path.join(base_dir, file_path)
|
||
if os.path.exists(full_path):
|
||
existing_files.append(file_path)
|
||
else:
|
||
missing_files.append(file_path)
|
||
|
||
return existing_files, missing_files
|
||
|
||
|
||
def get_version_from_version_py():
|
||
"""从 version.py 读取版本号"""
|
||
try:
|
||
from version import VERSION
|
||
return VERSION
|
||
except ImportError:
|
||
print("[WARNING] 无法导入 version.py，使用默认版本号 1.0.0")
|
||
return '1.0.0'
|
||
except Exception as e:
|
||
print(f"[WARNING] 读取 version.py 失败: {e}，使用默认版本号 1.0.0")
|
||
return '1.0.0'
|
||
|
||
|
||
def create_zip_package(app_info, files, output_dir='.', base_dir='.'):
|
||
"""创建 zip 打包文件"""
|
||
# 生成输出文件名：{name}_v{version}_{timestamp}.zip
|
||
# 版本号从 version.py 读取，而不是从 app.yaml
|
||
app_name = app_info.get('name', 'app')
|
||
version = get_version_from_version_py() # 从 version.py 读取版本号
|
||
timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
|
||
zip_filename = f"{app_name}_v{version}_{timestamp}.zip"
|
||
zip_path = os.path.join(output_dir, zip_filename)
|
||
|
||
print(f"[INFO] 开始打包: {zip_filename}")
|
||
print(f"[INFO] 包含文件数: {len(files)}")
|
||
|
||
try:
|
||
with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
|
||
for file_path in files:
|
||
full_path = os.path.join(base_dir, file_path)
|
||
# 使用相对路径作为 zip 内的路径
|
||
zipf.write(full_path, file_path)
|
||
print(f" ✓ {file_path}")
|
||
|
||
# 获取文件大小
|
||
file_size = os.path.getsize(zip_path)
|
||
file_size_mb = file_size / (1024 * 1024)
|
||
|
||
print(f"\n[SUCCESS] 打包完成!")
|
||
print(f" 文件名: {zip_filename}")
|
||
print(f" 文件大小: {file_size_mb:.2f} MB ({file_size:,} 字节)")
|
||
print(f" 文件路径: {os.path.abspath(zip_path)}")
|
||
|
||
return zip_path
|
||
except Exception as e:
|
||
print(f"[ERROR] 打包失败: {e}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
return None
|
||
|
||
|
||
def main():
|
||
"""主函数"""
|
||
print("=" * 60)
|
||
print("应用打包脚本")
|
||
print("=" * 60)
|
||
|
||
# 1. 加载 app.yaml
|
||
app_info = load_app_yaml('app.yaml')
|
||
if app_info is None:
|
||
return
|
||
|
||
# 从 version.py 读取版本号
|
||
version = get_version_from_version_py()
|
||
|
||
print(f"\n[INFO] 应用信息:")
|
||
print(f" ID: {app_info.get('id', 'N/A')}")
|
||
print(f" 名称: {app_info.get('name', 'N/A')}")
|
||
print(f" 版本: {version} (来自 version.py)")
|
||
print(f" 作者: {app_info.get('author', 'N/A')}")
|
||
if app_info.get('version') != version:
|
||
print(f" [注意] app.yaml 中的版本 ({app_info.get('version', 'N/A')}) 与 version.py 不一致")
|
||
|
||
# 2. 获取文件列表
|
||
files = app_info.get('files', [])
|
||
if not files:
|
||
print("[ERROR] app.yaml 中没有找到 files 列表")
|
||
return
|
||
|
||
print(f"\n[INFO] 文件列表 ({len(files)} 个文件):")
|
||
|
||
# 3. 检查文件是否存在
|
||
existing_files, missing_files = check_files_exist(files)
|
||
|
||
if missing_files:
|
||
print(f"\n[WARNING] 以下文件不存在，将被跳过:")
|
||
for f in missing_files:
|
||
print(f" ✗ {f}")
|
||
|
||
if not existing_files:
|
||
print("\n[ERROR] 没有找到任何有效文件，无法打包")
|
||
return
|
||
|
||
print(f"\n[INFO] 找到 {len(existing_files)} 个有效文件")
|
||
|
||
# 4. 创建 zip 包
|
||
zip_path = create_zip_package(app_info, existing_files)
|
||
|
||
if zip_path:
|
||
print("\n" + "=" * 60)
|
||
print("打包成功完成!")
|
||
print("=" * 60)
|
||
else:
|
||
print("\n" + "=" * 60)
|
||
print("打包失败!")
|
||
print("=" * 60)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main() |