2026-1-6
This commit is contained in:
“shengyudong”
238
快速配置测试.py
Normal file
238
快速配置测试.py
Normal file
@@ -0,0 +1,238 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
快速配置和启动后端接口性能测试
|
||||
|
||||
这个脚本提供交互式配置界面，帮助快速设置测试参数
|
||||
"""
|
||||
import sys
|
||||
import os
|
||||
|
||||
def print_header():
|
||||
"""打印头部"""
|
||||
print("\n" + "="*80)
|
||||
print("后端接口性能测试 - 快速配置向导")
|
||||
print("="*80 + "\n")
|
||||
|
||||
def get_api_url():
|
||||
"""获取API地址"""
|
||||
print("1. 配置主服务地址")
|
||||
print("-" * 80)
|
||||
default_url = "http://127.0.0.1:8216"
|
||||
url = input(f"请输入主服务地址 (默认: {default_url}): ").strip()
|
||||
return url if url else default_url
|
||||
|
||||
def get_search_url():
|
||||
"""获取搜索服务地址"""
|
||||
print("\n2. 配置搜索服务地址")
|
||||
print("-" * 80)
|
||||
default_url = "http://127.0.0.1:8321"
|
||||
url = input(f"请输入搜索服务地址 (默认: {default_url}): ").strip()
|
||||
return url if url else default_url
|
||||
|
||||
def get_credentials():
|
||||
"""获取登录凭证"""
|
||||
print("\n3. 配置登录信息")
|
||||
print("-" * 80)
|
||||
default_username = "13621242430"
|
||||
username = input(f"请输入用户名 (默认: {default_username}): ").strip()
|
||||
username = username if username else default_username
|
||||
|
||||
default_password = "admin123"
|
||||
password = input(f"请输入密码 (默认: {default_password}): ").strip()
|
||||
password = password if password else default_password
|
||||
|
||||
return username, password
|
||||
|
||||
def select_modules():
|
||||
"""选择测试模块"""
|
||||
print("\n4. 选择测试模块")
|
||||
print("-" * 80)
|
||||
|
||||
modules = {
|
||||
'basic': {'name': '基础接口', 'default': True},
|
||||
'auth': {'name': '认证接口', 'default': True},
|
||||
'dashboard': {'name': '工作台', 'default': True},
|
||||
'enterprise': {'name': '企业管理', 'default': True},
|
||||
'image': {'name': '图片管理', 'default': True},
|
||||
'log': {'name': '日志管理', 'default': True},
|
||||
'article': {'name': '文章管理', 'default': True},
|
||||
'employee': {'name': '员工管理', 'default': True},
|
||||
'author': {'name': '作者管理', 'default': True},
|
||||
'search': {'name': '搜索服务', 'default': False},
|
||||
}
|
||||
|
||||
print("可用的测试模块：")
|
||||
for i, (key, info) in enumerate(modules.items(), 1):
|
||||
status = "✓" if info['default'] else " "
|
||||
print(f" [{status}] {i:2}. {info['name']:12} ({key})")
|
||||
|
||||
print("\n选择操作：")
|
||||
print(" a - 全部启用")
|
||||
print(" n - 全部禁用")
|
||||
print(" d - 使用默认配置（推荐）")
|
||||
print(" c - 自定义选择")
|
||||
|
||||
choice = input("\n请选择 (默认: d): ").strip().lower()
|
||||
|
||||
if choice == 'a':
|
||||
# 全部启用
|
||||
return {key: True for key in modules.keys()}
|
||||
elif choice == 'n':
|
||||
# 全部禁用
|
||||
return {key: False for key in modules.keys()}
|
||||
elif choice == 'c':
|
||||
# 自定义选择
|
||||
print("\n请输入要启用的模块编号，用逗号分隔（如: 1,2,3）")
|
||||
print("或输入要禁用的模块编号前加'-'（如: 1,2,3,-10）")
|
||||
selected = input("模块编号: ").strip()
|
||||
|
||||
result = {key: info['default'] for key, info in modules.items()}
|
||||
|
||||
if selected:
|
||||
module_list = list(modules.keys())
|
||||
for s in selected.split(','):
|
||||
s = s.strip()
|
||||
if s.startswith('-'):
|
||||
# 禁用
|
||||
try:
|
||||
idx = int(s[1:]) - 1
|
||||
if 0 <= idx < len(module_list):
|
||||
result[module_list[idx]] = False
|
||||
except:
|
||||
pass
|
||||
else:
|
||||
# 启用
|
||||
try:
|
||||
idx = int(s) - 1
|
||||
if 0 <= idx < len(module_list):
|
||||
result[module_list[idx]] = True
|
||||
except:
|
||||
pass
|
||||
|
||||
return result
|
||||
else:
|
||||
# 默认配置
|
||||
return {key: info['default'] for key, info in modules.items()}
|
||||
|
||||
def generate_config(api_url, search_url, username, password, modules):
|
||||
"""生成配置代码"""
|
||||
config_lines = [
|
||||
"# 自动生成的配置",
|
||||
f"API_BASE_URL = \"{api_url}\"",
|
||||
f"SEARCH_API_BASE_URL = \"{search_url}\"",
|
||||
f"USERNAME = \"{username}\"",
|
||||
f"PASSWORD = \"{password}\"",
|
||||
"",
|
||||
"TEST_MODULES = {",
|
||||
]
|
||||
|
||||
for key, enabled in modules.items():
|
||||
config_lines.append(f" '{key}': {enabled},")
|
||||
|
||||
config_lines.append("}")
|
||||
|
||||
return "\n".join(config_lines)
|
||||
|
||||
def update_script(config):
|
||||
"""更新脚本配置"""
|
||||
script_file = "测试整体后端接口性能.py"
|
||||
|
||||
if not os.path.exists(script_file):
|
||||
print(f"\n✗ 错误：找不到文件 {script_file}")
|
||||
return False
|
||||
|
||||
try:
|
||||
with open(script_file, 'r', encoding='utf-8') as f:
|
||||
content = f.read()
|
||||
|
||||
# 查找配置区域
|
||||
start_marker = "# API配置"
|
||||
end_marker = "# 测试结果存储"
|
||||
|
||||
start_idx = content.find(start_marker)
|
||||
end_idx = content.find(end_marker)
|
||||
|
||||
if start_idx == -1 or end_idx == -1:
|
||||
print("\n✗ 错误：无法找到配置区域")
|
||||
return False
|
||||
|
||||
# 替换配置
|
||||
new_content = (
|
||||
content[:start_idx] +
|
||||
config + "\n\n" +
|
||||
content[end_idx:]
|
||||
)
|
||||
|
||||
# 保存
|
||||
with open(script_file, 'w', encoding='utf-8') as f:
|
||||
f.write(new_content)
|
||||
|
||||
print(f"\n✓ 配置已更新到 {script_file}")
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
print(f"\n✗ 更新配置失败: {e}")
|
||||
return False
|
||||
|
||||
def main():
|
||||
"""主函数"""
|
||||
print_header()
|
||||
|
||||
# 获取配置
|
||||
api_url = get_api_url()
|
||||
search_url = get_search_url()
|
||||
username, password = get_credentials()
|
||||
modules = select_modules()
|
||||
|
||||
# 显示配置摘要
|
||||
print("\n" + "="*80)
|
||||
print("配置摘要")
|
||||
print("="*80)
|
||||
print(f"主服务地址: {api_url}")
|
||||
print(f"搜索服务地址: {search_url}")
|
||||
print(f"登录用户: {username}")
|
||||
print(f"登录密码: {'*' * len(password)}")
|
||||
|
||||
enabled_modules = [k for k, v in modules.items() if v]
|
||||
disabled_modules = [k for k, v in modules.items() if not v]
|
||||
|
||||
print(f"\n启用的模块 ({len(enabled_modules)}个): {', '.join(enabled_modules)}")
|
||||
if disabled_modules:
|
||||
print(f"禁用的模块 ({len(disabled_modules)}个): {', '.join(disabled_modules)}")
|
||||
|
||||
# 确认
|
||||
print("\n" + "="*80)
|
||||
confirm = input("确认配置并更新脚本？(Y/n): ").strip().lower()
|
||||
|
||||
if confirm in ['', 'y', 'yes']:
|
||||
# 生成配置
|
||||
config = generate_config(api_url, search_url, username, password, modules)
|
||||
|
||||
# 更新脚本
|
||||
if update_script(config):
|
||||
print("\n" + "="*80)
|
||||
print("✓ 配置完成！")
|
||||
print("="*80)
|
||||
print("\n现在可以运行测试脚本：")
|
||||
print(" python 测试整体后端接口性能.py")
|
||||
print("\n或者使用命令：")
|
||||
print(" python -m 测试整体后端接口性能")
|
||||
return 0
|
||||
else:
|
||||
return 1
|
||||
else:
|
||||
print("\n✗ 已取消配置")
|
||||
return 0
|
||||
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
sys.exit(main())
|
||||
except KeyboardInterrupt:
|
||||
print("\n\n✗ 配置被用户中断")
|
||||
sys.exit(1)
|
||||
except Exception as e:
|
||||
print(f"\n✗ 配置过程出现异常: {e}")
|
||||
import traceback
|
||||
traceback.print_exc()
|
||||
sys.exit(1)
|
||||
Reference in New Issue
Block a user