WebsocketTest 1.0.11__py3-none-any.whl → 1.0.12__py3-none-any.whl

WebsocketTest/allure_installer.py

@@ -0,0 +1,163 @@
+import os
+import sys
+import zipfile
+import subprocess
+import winreg
+from pathlib import Path
+
+# 配置参数
+ALLURE_VERSION = "2.30.0"
+PACKAGE_NAME = "WebsocketTest"  # 您的包名
+LOCAL_ZIP_RELATIVE_PATH = os.path.join("libs", f"allure-{ALLURE_VERSION}.zip")  # 包内相对路径
+INSTALL_DIR = os.path.expanduser("~/.allure")  # 安装到用户目录
+ALLURE_BIN_DIR = os.path.join(INSTALL_DIR, f"allure-{ALLURE_VERSION}", "bin")
+
+def is_allure_installed():
+    """检查Allure是否在PATH中且可执行"""
+    try:
+        # 同时检查版本避免找到无效安装
+        result = subprocess.run(["allure", "--version"], 
+                              check=True,
+                              stdout=subprocess.PIPE,
+                              stderr=subprocess.PIPE,
+                              text=True)
+        installed_version = result.stdout.strip()
+        if ALLURE_VERSION not in installed_version:
+            print(f"⚠️ 发现不匹配的Allure版本: {installed_version} (需要 {ALLURE_VERSION})")
+            return False
+        return True
+    except (subprocess.CalledProcessError, FileNotFoundError):
+        return False
+
+def get_local_zip_path():
+    """获取ZIP文件的绝对路径（兼容开发模式和正式安装）"""
+    # 尝试从包内获取
+    try:
+        import importlib.resources as pkg_resources
+        with pkg_resources.path(PACKAGE_NAME, LOCAL_ZIP_RELATIVE_PATH) as zip_path:
+            return str(zip_path)
+    except:
+        pass
+    
+    # 回退方案：从当前工作目录查找
+    base_dirs = [
+        os.getcwd(),
+        os.path.dirname(os.path.abspath(__file__)),
+        sys.prefix
+    ]
+    
+    for base in base_dirs:
+        zip_path = os.path.join(base, PACKAGE_NAME, LOCAL_ZIP_RELATIVE_PATH)
+        if os.path.exists(zip_path):
+            return zip_path
+    
+    raise FileNotFoundError(f"Allure ZIP文件未在任何位置找到: {LOCAL_ZIP_RELATIVE_PATH}")
+
+def install_allure():
+    """从本地ZIP安装Allure"""
+    try:
+        zip_path = get_local_zip_path()
+        print(f"🔍 找到Allure ZIP文件: {zip_path}")
+
+        # 创建安装目录
+        os.makedirs(INSTALL_DIR, exist_ok=True)
+        print(f"📦 解压到: {INSTALL_DIR}")
+
+        # 解压ZIP文件（使用更安全的提取方法）
+        with zipfile.ZipFile(zip_path, 'r') as zip_ref:
+            for file in zip_ref.namelist():
+                # 防止Zip Slip攻击
+                safe_path = os.path.join(INSTALL_DIR, *file.split('/'))
+                if file.endswith('/'):
+                    os.makedirs(safe_path, exist_ok=True)
+                else:
+                    with open(safe_path, 'wb') as f:
+                        f.write(zip_ref.read(file))
+
+        # 设置权限（特别是Linux/macOS）
+        if sys.platform != "win32":
+            os.chmod(os.path.join(ALLURE_BIN_DIR, "allure"), 0o755)
+
+        # 更新PATH
+        if add_to_user_path(ALLURE_BIN_DIR):
+            # 立即在当前进程生效
+            os.environ["PATH"] = f"{ALLURE_BIN_DIR}{os.pathsep}{os.environ.get('PATH', '')}"
+        
+        return True
+    except Exception as e:
+        print(f"❌ 安装失败: {type(e).__name__}: {e}", file=sys.stderr)
+        return False
+
+def add_to_user_path(path):
+    """添加到用户级PATH环境变量"""
+    try:
+        if sys.platform == "win32":
+            # Windows注册表操作
+            with winreg.OpenKey(
+                winreg.HKEY_CURRENT_USER,
+                r"Environment",
+                0,
+                winreg.KEY_READ | winreg.KEY_WRITE,
+            ) as key:
+                current_path, _ = winreg.QueryValueEx(key, "Path")
+                
+                if path in current_path.split(os.pathsep):
+                    return False
+                
+                new_path = f"{current_path}{os.pathsep}{path}" if current_path else path
+                winreg.SetValueEx(key, "Path", 0, winreg.REG_EXPAND_SZ, new_path)
+            
+            # 刷新环境变量
+            subprocess.run(
+                'powershell -command "[Environment]::SetEnvironmentVariable(\'Path\', $env:Path + \';{}\', \'User\')"'
+                .format(path),
+                shell=True,
+                check=True
+            )
+        else:
+            # Linux/macOS: 修改shell配置文件
+            shell_config = os.path.expanduser("~/.bashrc")
+            if not os.path.exists(shell_config):
+                shell_config = os.path.expanduser("~/.zshrc")
+            
+            with open(shell_config, "a") as f:
+                f.write(f"\nexport PATH=\"$PATH:{path}\"\n")
+        
+        print(f"✅ 已添加PATH: {path}")
+        return True
+    except Exception as e:
+        print(f"⚠️ 添加PATH失败: {e}\n请手动添加 {path} 到环境变量", file=sys.stderr)
+        return False
+
+def ensure_allure():
+    """确保Allure已安装"""
+    if is_allure_installed():
+        print(f"✅ Allure {ALLURE_VERSION} 已安装")
+        return True
+    
+    print("🔧 检测到Allure未安装，开始自动安装...")
+    if install_allure():
+        if not is_allure_installed():
+            print("""
+            \n⚠️ 安装成功但Allure仍未识别，可能是因为：
+            1. 需要重启终端使PATH生效
+            2. 尝试手动运行: {}
+            """.format(os.path.join(ALLURE_BIN_DIR, "allure" + (".bat" if sys.platform == "win32" else ""))))
+            return False
+        return True
+    else:
+        print(f"""
+        \n❌ 自动安装失败，请手动操作：
+        1. 解压 {get_local_zip_path()} 到任意目录（推荐 {INSTALL_DIR}）
+        2. 将解压后的bin目录添加到PATH:
+           - Windows: 添加 {ALLURE_BIN_DIR} 到系统环境变量
+           - Linux/macOS: 在~/.bashrc或~/.zshrc中添加:
+             export PATH="$PATH:{ALLURE_BIN_DIR}"
+        3. 运行 `allure --version` 验证
+        """)
+        return False
+
+if __name__ == "__main__":
+    if ensure_allure():
+        sys.exit(0)
+    sys.exit(1)

WebsocketTest/run_tests.py

@@ -1,7 +1,7 @@
 import subprocess
 import argparse
 import shutil
-import time
+import time,platform
 import webbrowser
 from urllib.request import urlopen
 from urllib.error import URLError
@@ -93,6 +93,33 @@ class TestRunner:
             logger.error(f"Tests failed with exit code {e.returncode}")
         except subprocess.TimeoutExpired:
             logger.error("Test execution timed out after 1 hour")
+def is_allure_installed():
+    """检查 Allure 是否已安装"""
+    try:
+        # 尝试运行 allure --version
+        result = subprocess.run(
+            ["allure", "--version"],
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE,
+            text=True,
+        )
+        return result.returncode == 0
+    except FileNotFoundError:
+        return False
+
+def install_allure():
+    """根据操作系统自动安装 Allure"""
+    system = platform.system().lower()
+    
+    if system == "windows":
+        # Windows 使用 scoop 安装
+        if not shutil.which("scoop"):
+            print("⚠️ 请先安装 Scoop (https://scoop.sh/)")
+            exit(1)
+        subprocess.run(["scoop", "install", "allure"], check=True)
+    else:
+        print(f"❌ 不支持的操作系统: {system}")
+        exit(1)
 
     def generate_allure_report(self) -> bool:
         """生成Allure报告"""

{websockettest-1.0.11.dist-info → websockettest-1.0.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: WebsocketTest
-Version: 1.0.11
+Version: 1.0.12
 Summary: websocket api autotest
 Requires-Dist: allure_python_commons==2.13.5
 Requires-Dist: numpy==2.2.4

{websockettest-1.0.11.dist-info → websockettest-1.0.12.dist-info}/RECORD

@@ -1,7 +1,8 @@
 WebsocketTest/__init__.py,sha256=u71SAVmbgsyp0K21kilo7pIDgeyxsaHAi93clC0OIPQ,556
+WebsocketTest/allure_installer.py,sha256=OicATAGPC2EBMCxjoKlgFFJA8lfT-qDjakLWwDA8IAU,6194
 WebsocketTest/cli.py,sha256=HcnjgRLM404fVszrSRAf7G1y9XLJ_ZetVtRIwavS4xE,987
 WebsocketTest/conftest.py,sha256=xD7IK7airvGH9cDbES-G9PaKUdPazVuv3utB_UfRV-8,1875
-WebsocketTest/run_tests.py,sha256=kKdanw4dWdp3IY3Bm0-GVDpMgEu5mBbTQ7hbJ1-gusM,6227
+WebsocketTest/run_tests.py,sha256=xIe1QpWUT2LKEQ9yFoDQVZlV4epYhe2YqTktR-jAKis,7099
 WebsocketTest/caseScript/Aqua.py,sha256=dTjVd4FPka5lMUvxa1DsGTJ6gx7NRo5NGNunrue3d0M,7439
 WebsocketTest/caseScript/Gateway.py,sha256=4_BlJPbn6X_eFHYz2orn3fXPQMPz5jRdUlj6tXkeX2Q,11748
 WebsocketTest/caseScript/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -14,6 +15,7 @@ WebsocketTest/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hS
 WebsocketTest/common/assertUtils.py,sha256=poyL6J_WXb2BJvaABCILfiWHN7gHSTRcMWvpjk224yU,4713
 WebsocketTest/common/logger.py,sha256=JY7dJLVObKoA6E-5FJ0EMX_EyN8Otg4_4FYywKVSkcg,367
 WebsocketTest/common/utils.py,sha256=DfRwSqbci1w-omUjO4TlkqTqlc-_69Gs9JQtwL3-YSo,8704
+WebsocketTest/libs/allure-2.30.0.zip,sha256=SEc2gYElQbTQrIcpJ5gLIWptfPW8kPITU8gs4wezZho,26551281
 WebsocketTest/templates/basic/conftest.py,sha256=y2_2961mhCYSRMp5X9xLq3nOmRzaTSgD3GDdrlTqxn4,1885
 WebsocketTest/templates/basic/pytest.ini,sha256=An32MB2Yy2L4EVP9WoczW8x0ISAXsgG0tlSLta-cDcs,589
 WebsocketTest/templates/basic/config/svw/live.yml,sha256=dqTMDubVYO0lt2kPoHgbZW0ZVMaLO7-OjfBmHedqJuU,31
@@ -80,8 +82,8 @@ WebsocketTest/templates/basic/testcase/__pycache__/test_gateway.cpython-310-pyte
 WebsocketTest/templates/basic/testcase/__pycache__/test_gateway1.cpython-310-pytest-8.2.2.pyc,sha256=9_jiZtyNwQfXe-Jz1acHxrwz5snYfBngqgTFtIWhRNA,5684
 WebsocketTest/templates/basic/testcase/__pycache__/test_gatewaybak.cpython-310-pytest-8.2.2.pyc,sha256=L50gNeQvSuFq2e-arnmKqYJR75ZrThiVaiIow5UUTjo,5587
 WebsocketTest/templates/basic/testcase/__pycache__/test_limin.cpython-310-pytest-8.2.2.pyc,sha256=hx7j0GNxlgTscC36dUBHeo401Mo3bRt1cq6t7e7dDSA,3434
-websockettest-1.0.11.dist-info/METADATA,sha256=Ti0eOTADdn_lwVptuKeUbAbUuGtEF4rRHPlLwYRH7Bs,334
-websockettest-1.0.11.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-websockettest-1.0.11.dist-info/entry_points.txt,sha256=9-0RmZ0nVgfDKaDFyk5xv8arp_us-7w5Negvis1CvYc,46
-websockettest-1.0.11.dist-info/top_level.txt,sha256=2iF1gZSbXLjVFOe5ZKQiCLC1FzAwhcQUM88yGi-vrCU,14
-websockettest-1.0.11.dist-info/RECORD,,
+websockettest-1.0.12.dist-info/METADATA,sha256=q9VGwnji09k25ZtJ9F85Qo9qImQvv_X3-tpdqM2zZyM,334
+websockettest-1.0.12.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+websockettest-1.0.12.dist-info/entry_points.txt,sha256=9-0RmZ0nVgfDKaDFyk5xv8arp_us-7w5Negvis1CvYc,46
+websockettest-1.0.12.dist-info/top_level.txt,sha256=2iF1gZSbXLjVFOe5ZKQiCLC1FzAwhcQUM88yGi-vrCU,14
+websockettest-1.0.12.dist-info/RECORD,,