add pytest in makefile

publish.py

@@ -0,0 +1,164 @@
+#!/usr/bin/env python3
+"""
+RustFS S3 Storage Toolkit 发布脚本
+支持交互式和自动化发布
+"""
+
+import subprocess
+import sys
+import os
+import argparse
+from pathlib import Path
+
+
+def run_command(command, description):
+    """运行命令并显示结果"""
+    print(f"🔧 {description}...")
+    try:
+        result = subprocess.run(command, shell=True, check=True, capture_output=True, text=True)
+        print(f"✅ {description}成功")
+        if result.stdout.strip():
+            print(f"输出: {result.stdout.strip()}")
+        return True
+    except subprocess.CalledProcessError as e:
+        print(f"❌ {description}失败: {e}")
+        if e.stdout:
+            print(f"输出: {e.stdout}")
+        if e.stderr:
+            print(f"错误: {e.stderr}")
+        return False
+
+
+def main():
+    """主发布流程"""
+    parser = argparse.ArgumentParser(description="RustFS S3 Storage Toolkit 发布脚本")
+    parser.add_argument("--auto", action="store_true", help="自动发布到 PyPI (跳过交互)")
+    parser.add_argument("--test", action="store_true", help="发布到 TestPyPI")
+    parser.add_argument("--check-only", action="store_true", help="仅构建和检查，不发布")
+    args = parser.parse_args()
+
+    print("🚀 RustFS S3 Storage Toolkit 发布脚本")
+    print("=" * 50)
+
+    # 检查当前目录
+    current_dir = Path.cwd()
+    if not (current_dir / "pyproject.toml").exists():
+        print("❌ 请在项目根目录运行此脚本")
+        sys.exit(1)
+
+    print(f"✅ 项目目录: {current_dir}")
+
+    # 检查 PyPI 配置
+    pypirc_path = Path.home() / ".pypirc"
+    if not pypirc_path.exists() and not args.check_only:
+        print("⚠️  警告: 未找到 ~/.pypirc 配置文件")
+        print("请先配置 PyPI 认证信息")
+        sys.exit(1)
+    
+    # 1. 清理和构建
+    print("\n🧹 清理旧文件...")
+    cleanup_commands = [
+        "rm -rf dist/",
+        "rm -rf build/",
+        "rm -rf *.egg-info/",
+        "rm -rf src/*.egg-info/",
+    ]
+    for command in cleanup_commands:
+        subprocess.run(command, shell=True, capture_output=True)
+    print("✅ 清理完成")
+
+    print("\n📦 开始构建...")
+    if not run_command("python -m build", "构建包"):
+        print("❌ 构建失败，无法继续发布")
+        sys.exit(1)
+    
+    # 2. 检查构建结果
+    dist_dir = current_dir / "dist"
+    if not dist_dir.exists() or not list(dist_dir.glob("*")):
+        print("❌ 没有找到构建文件")
+        sys.exit(1)
+    
+    built_files = list(dist_dir.glob("*"))
+    print(f"\n📦 构建文件:")
+    for file in built_files:
+        print(f"   - {file.name}")
+
+    # 3. 验证包
+    print("\n🔍 验证包格式...")
+    if not run_command("python -m twine check dist/*", "验证包"):
+        print("❌ 包验证失败")
+        sys.exit(1)
+
+    # 如果只是检查，到此结束
+    if args.check_only:
+        print("\n✅ 构建和验证完成！")
+        return
+
+    # 4. 确定发布类型
+    if args.test:
+        publish_type = "test"
+    elif args.auto:
+        publish_type = "pypi"
+    else:
+        # 交互式选择
+        print("\n🎯 选择发布类型:")
+        print("1. 测试发布 (TestPyPI)")
+        print("2. 正式发布 (PyPI)")
+        print("3. 取消")
+
+        choice = input("\n请选择 (1/2/3): ").strip()
+        if choice == "1":
+            publish_type = "test"
+        elif choice == "2":
+            publish_type = "pypi"
+        else:
+            print("❌ 发布已取消")
+            return
+
+    # 5. 执行发布
+    if publish_type == "test":
+        # 测试发布
+        print("\n🧪 发布到 TestPyPI...")
+        if run_command("python -m twine upload --repository testpypi dist/*", "测试发布"):
+            print("\n🎉 测试发布成功！")
+            print("📖 测试安装命令:")
+            print("pip install --index-url https://test.pypi.org/simple/ rustfs-s3-toolkit")
+        else:
+            print("❌ 测试发布失败")
+            sys.exit(1)
+
+    elif publish_type == "pypi":
+        # 正式发布
+        if not args.auto:
+            print("\n⚠️ 确认正式发布到 PyPI？")
+            confirm = input("输入 'yes' 确认: ").strip().lower()
+            if confirm != "yes":
+                print("❌ 发布已取消")
+                return
+
+        print("\n🚀 发布到 PyPI...")
+        if run_command("python -m twine upload dist/*", "正式发布"):
+            print("\n🎉 正式发布成功！")
+            print("📖 安装命令:")
+            print("pip install rustfs-s3-toolkit")
+            print("\n🔗 PyPI 链接:")
+            print("https://pypi.org/project/rustfs-s3-toolkit/")
+        else:
+            print("❌ 正式发布失败")
+            sys.exit(1)
+
+    print("\n💡 后续步骤:")
+    print("1. 检查 PyPI 页面确认发布成功")
+    print("2. 测试安装: pip install rustfs-s3-toolkit")
+    print("3. 更新版本号准备下次发布")
+
+
+if __name__ == "__main__":
+    try:
+        main()
+    except KeyboardInterrupt:
+        print("\n❌ 发布已中断")
+        sys.exit(1)
+    except Exception as e:
+        print(f"\n❌ 发布失败: {e}")
+        sys.exit(1)