mirror of
https://github.com/zhinianboke/xianyu-auto-reply.git
synced 2025-08-02 04:27:36 +08:00
291 lines
10 KiB
Python
291 lines
10 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
关键词回复功能测试脚本
|
||
用于验证关键词匹配是否正常工作
|
||
"""
|
||
|
||
import asyncio
|
||
import sys
|
||
import os
|
||
|
||
# 添加项目根目录到Python路径
|
||
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
||
|
||
from db_manager import db_manager
|
||
from XianyuAutoAsync import XianyuLive
|
||
from loguru import logger
|
||
|
||
def test_keyword_database():
|
||
"""测试关键词数据库操作"""
|
||
print("🗄️ 开始测试关键词数据库操作...")
|
||
|
||
test_cookie_id = "test_cookie_001"
|
||
|
||
# 1. 清理测试数据
|
||
print("\n1️⃣ 清理测试数据...")
|
||
try:
|
||
with db_manager.lock:
|
||
cursor = db_manager.conn.cursor()
|
||
cursor.execute("DELETE FROM keywords WHERE cookie_id = ?", (test_cookie_id,))
|
||
db_manager.conn.commit()
|
||
print(" ✅ 测试数据清理完成")
|
||
except Exception as e:
|
||
print(f" ❌ 清理测试数据失败: {e}")
|
||
return False
|
||
|
||
# 2. 添加测试关键词
|
||
print("\n2️⃣ 添加测试关键词...")
|
||
test_keywords = [
|
||
("你好", "您好！欢迎咨询，有什么可以帮助您的吗？"),
|
||
("价格", "这个商品的价格很优惠哦，{send_user_name}！"),
|
||
("包邮", "全国包邮，放心购买！"),
|
||
("发货", "我们会在24小时内发货"),
|
||
("退换", "支持7天无理由退换货")
|
||
]
|
||
|
||
try:
|
||
success = db_manager.save_keywords(test_cookie_id, test_keywords)
|
||
if success:
|
||
print(f" ✅ 成功添加 {len(test_keywords)} 个关键词")
|
||
else:
|
||
print(" ❌ 添加关键词失败")
|
||
return False
|
||
except Exception as e:
|
||
print(f" ❌ 添加关键词异常: {e}")
|
||
return False
|
||
|
||
# 3. 验证关键词保存
|
||
print("\n3️⃣ 验证关键词保存...")
|
||
try:
|
||
saved_keywords = db_manager.get_keywords(test_cookie_id)
|
||
if len(saved_keywords) == len(test_keywords):
|
||
print(f" ✅ 关键词保存验证成功: {len(saved_keywords)} 个")
|
||
for keyword, reply in saved_keywords:
|
||
print(f" '{keyword}' -> '{reply[:30]}...'")
|
||
else:
|
||
print(f" ❌ 关键词数量不匹配: 期望 {len(test_keywords)}, 实际 {len(saved_keywords)}")
|
||
return False
|
||
except Exception as e:
|
||
print(f" ❌ 验证关键词异常: {e}")
|
||
return False
|
||
|
||
print("\n✅ 关键词数据库操作测试完成！")
|
||
return True
|
||
|
||
async def test_keyword_matching():
|
||
"""测试关键词匹配功能"""
|
||
print("\n🔍 开始测试关键词匹配功能...")
|
||
|
||
test_cookie_id = "test_cookie_001"
|
||
|
||
# 创建一个简化的测试类
|
||
class TestKeywordMatcher:
|
||
def __init__(self, cookie_id):
|
||
self.cookie_id = cookie_id
|
||
|
||
async def get_keyword_reply(self, send_user_name: str, send_user_id: str, send_message: str) -> str:
|
||
"""获取关键词匹配回复"""
|
||
try:
|
||
from db_manager import db_manager
|
||
|
||
# 获取当前账号的关键词列表
|
||
keywords = db_manager.get_keywords(self.cookie_id)
|
||
|
||
if not keywords:
|
||
print(f" 调试: 账号 {self.cookie_id} 没有配置关键词")
|
||
return None
|
||
|
||
# 遍历关键词，查找匹配
|
||
for keyword, reply in keywords:
|
||
if keyword.lower() in send_message.lower():
|
||
# 进行变量替换
|
||
try:
|
||
formatted_reply = reply.format(
|
||
send_user_name=send_user_name,
|
||
send_user_id=send_user_id,
|
||
send_message=send_message
|
||
)
|
||
print(f" 调试: 关键词匹配成功: '{keyword}' -> {formatted_reply}")
|
||
return f"[关键词回复] {formatted_reply}"
|
||
except Exception as format_error:
|
||
print(f" 调试: 关键词回复变量替换失败: {format_error}")
|
||
# 如果变量替换失败，返回原始内容
|
||
return f"[关键词回复] {reply}"
|
||
|
||
print(f" 调试: 未找到匹配的关键词: {send_message}")
|
||
return None
|
||
|
||
except Exception as e:
|
||
print(f" 调试: 获取关键词回复失败: {e}")
|
||
return None
|
||
|
||
# 创建测试实例
|
||
test_matcher = TestKeywordMatcher(test_cookie_id)
|
||
|
||
# 测试消息和期望结果
|
||
test_cases = [
|
||
{
|
||
"message": "你好",
|
||
"expected_keyword": "你好",
|
||
"should_match": True
|
||
},
|
||
{
|
||
"message": "请问价格多少？",
|
||
"expected_keyword": "价格",
|
||
"should_match": True
|
||
},
|
||
{
|
||
"message": "包邮吗？",
|
||
"expected_keyword": "包邮",
|
||
"should_match": True
|
||
},
|
||
{
|
||
"message": "什么时候发货？",
|
||
"expected_keyword": "发货",
|
||
"should_match": True
|
||
},
|
||
{
|
||
"message": "可以退换吗？",
|
||
"expected_keyword": "退换",
|
||
"should_match": True
|
||
},
|
||
{
|
||
"message": "这是什么材质的？",
|
||
"expected_keyword": None,
|
||
"should_match": False
|
||
}
|
||
]
|
||
|
||
success_count = 0
|
||
|
||
for i, test_case in enumerate(test_cases, 1):
|
||
print(f"\n{i}️⃣ 测试消息: '{test_case['message']}'")
|
||
|
||
try:
|
||
reply = await test_matcher.get_keyword_reply(
|
||
send_user_name="测试用户",
|
||
send_user_id="test_user_001",
|
||
send_message=test_case['message']
|
||
)
|
||
|
||
if test_case['should_match']:
|
||
if reply:
|
||
print(f" ✅ 匹配成功: {reply}")
|
||
if test_case['expected_keyword'] in reply or test_case['expected_keyword'] in test_case['message']:
|
||
success_count += 1
|
||
else:
|
||
print(f" ⚠️ 匹配的关键词不符合预期")
|
||
else:
|
||
print(f" ❌ 期望匹配但未匹配")
|
||
else:
|
||
if reply:
|
||
print(f" ❌ 不应该匹配但却匹配了: {reply}")
|
||
else:
|
||
print(f" ✅ 正确未匹配")
|
||
success_count += 1
|
||
|
||
except Exception as e:
|
||
print(f" ❌ 测试异常: {e}")
|
||
|
||
print(f"\n📊 测试结果: {success_count}/{len(test_cases)} 个测试通过")
|
||
|
||
if success_count == len(test_cases):
|
||
print("✅ 关键词匹配功能测试完成！")
|
||
return True
|
||
else:
|
||
print("❌ 部分测试失败")
|
||
return False
|
||
|
||
def test_reply_priority():
|
||
"""测试回复优先级"""
|
||
print("\n🎯 开始测试回复优先级...")
|
||
|
||
test_cookie_id = "test_cookie_001"
|
||
|
||
# 检查AI回复状态
|
||
print("\n1️⃣ 检查AI回复状态...")
|
||
try:
|
||
ai_settings = db_manager.get_ai_reply_settings(test_cookie_id)
|
||
ai_enabled = ai_settings.get('ai_enabled', False)
|
||
print(f" AI回复状态: {'启用' if ai_enabled else '禁用'}")
|
||
except Exception as e:
|
||
print(f" ❌ 检查AI回复状态失败: {e}")
|
||
return False
|
||
|
||
# 检查关键词数量
|
||
print("\n2️⃣ 检查关键词配置...")
|
||
try:
|
||
keywords = db_manager.get_keywords(test_cookie_id)
|
||
print(f" 关键词数量: {len(keywords)} 个")
|
||
if len(keywords) > 0:
|
||
print(" 关键词列表:")
|
||
for keyword, reply in keywords[:3]: # 只显示前3个
|
||
print(f" '{keyword}' -> '{reply[:30]}...'")
|
||
except Exception as e:
|
||
print(f" ❌ 检查关键词配置失败: {e}")
|
||
return False
|
||
|
||
# 检查默认回复
|
||
print("\n3️⃣ 检查默认回复配置...")
|
||
try:
|
||
default_reply = db_manager.get_default_reply(test_cookie_id)
|
||
if default_reply and default_reply.get('enabled', False):
|
||
print(f" 默认回复: 启用")
|
||
print(f" 默认回复内容: {default_reply.get('reply_content', '')[:50]}...")
|
||
else:
|
||
print(f" 默认回复: 禁用")
|
||
except Exception as e:
|
||
print(f" ❌ 检查默认回复配置失败: {e}")
|
||
return False
|
||
|
||
print("\n📋 回复优先级说明:")
|
||
print(" 1. API回复 (最高优先级)")
|
||
print(" 2. AI回复 (如果启用)")
|
||
print(" 3. 关键词匹配 (如果AI禁用)")
|
||
print(" 4. 默认回复 (最低优先级)")
|
||
|
||
if not ai_enabled and len(keywords) > 0:
|
||
print("\n✅ 当前配置下，关键词匹配应该正常工作！")
|
||
return True
|
||
elif ai_enabled:
|
||
print("\n⚠️ 当前AI回复已启用，关键词匹配会被跳过")
|
||
print(" 如需测试关键词匹配，请先禁用AI回复")
|
||
return True
|
||
else:
|
||
print("\n⚠️ 当前没有配置关键词，将使用默认回复")
|
||
return True
|
||
|
||
async def main():
|
||
"""主测试函数"""
|
||
print("🚀 关键词回复功能测试开始")
|
||
print("=" * 50)
|
||
|
||
# 测试数据库操作
|
||
db_ok = test_keyword_database()
|
||
if not db_ok:
|
||
print("\n❌ 数据库操作测试失败")
|
||
return
|
||
|
||
# 测试关键词匹配
|
||
match_ok = await test_keyword_matching()
|
||
if not match_ok:
|
||
print("\n❌ 关键词匹配测试失败")
|
||
return
|
||
|
||
# 测试回复优先级
|
||
priority_ok = test_reply_priority()
|
||
if not priority_ok:
|
||
print("\n❌ 回复优先级测试失败")
|
||
return
|
||
|
||
print("\n" + "=" * 50)
|
||
print("🎉 所有测试通过！关键词回复功能正常！")
|
||
print("\n📋 使用说明:")
|
||
print("1. 在Web界面的'自动回复'页面配置关键词")
|
||
print("2. 确保AI回复已禁用（如果要使用关键词匹配）")
|
||
print("3. 发送包含关键词的消息进行测试")
|
||
print("4. 关键词匹配支持变量替换：{send_user_name}, {send_user_id}, {send_message}")
|
||
|
||
if __name__ == "__main__":
|
||
asyncio.run(main())
|