239 lines
7.5 KiB
Python
239 lines
7.5 KiB
Python
#!/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)
|