mirror of
https://github.com/Hommy-master/capcut-mate.git
synced 2026-08-29 07:31:26 +08:00
Hommy
98 lines
3.1 KiB
Python
98 lines
3.1 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
基本测试 search_sticker 功能
|
|
"""
|
|
|
|
import json
|
|
import random
|
|
import os
|
|
|
|
|
|
def search_sticker(keyword: str, sticker_config_path: str) -> list:
|
|
"""
|
|
搜索贴纸的业务逻辑（简化版）
|
|
|
|
Args:
|
|
keyword: 搜索关键词
|
|
sticker_config_path: 贴纸配置文件路径
|
|
|
|
Returns:
|
|
list: 贴纸数据列表，最多返回50条记录
|
|
"""
|
|
print(f"Searching stickers with keyword: {keyword}")
|
|
|
|
# 从配置文件中读取贴纸数据
|
|
try:
|
|
with open(sticker_config_path, 'r', encoding='utf-8') as f:
|
|
sticker_data = json.load(f)
|
|
except FileNotFoundError:
|
|
print(f"Sticker config file not found: {sticker_config_path}")
|
|
return []
|
|
except json.JSONDecodeError as e:
|
|
print(f"Failed to parse sticker config file: {e}")
|
|
return []
|
|
except Exception as e:
|
|
print(f"Failed to read sticker config file: {e}")
|
|
return []
|
|
|
|
# 根据关键词过滤数据（简单匹配标题）
|
|
filtered_data = []
|
|
for item in sticker_data:
|
|
if keyword in item["title"]:
|
|
filtered_data.append(item)
|
|
|
|
print(f"Found {len(filtered_data)} stickers matching keyword: {keyword}")
|
|
|
|
# 如果没有找到匹配的贴纸，随机返回50条记录
|
|
if not filtered_data:
|
|
print("No matching stickers found, returning 50 random stickers")
|
|
# 确保不会超出总数据量
|
|
sample_size = min(50, len(sticker_data))
|
|
filtered_data = random.sample(sticker_data, sample_size)
|
|
|
|
# 如果找到的贴纸超过50条，只返回前50条
|
|
if len(filtered_data) > 50:
|
|
print(f"Too many matching stickers ({len(filtered_data)}), returning only 50")
|
|
filtered_data = filtered_data[:50]
|
|
|
|
return filtered_data
|
|
|
|
|
|
def test_search_sticker():
|
|
"""测试搜索贴纸功能"""
|
|
|
|
print("测试搜索贴纸功能")
|
|
|
|
# 设置贴纸配置文件路径
|
|
sticker_config_path = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "config", "sticker.json")
|
|
|
|
# 测试正常搜索
|
|
print("\n1. 测试正常搜索 '梦幻':")
|
|
result = search_sticker("梦幻", sticker_config_path)
|
|
print(f" 找到 {len(result)} 条记录")
|
|
if result:
|
|
print(f" 第一条记录标题: {result[0]['title']}")
|
|
|
|
# 测试搜索不到内容的情况
|
|
print("\n2. 测试搜索不存在的关键词 '不存在的关键词':")
|
|
result = search_sticker("不存在的关键词", sticker_config_path)
|
|
print(f" 找到 {len(result)} 条记录（应该是随机的50条）")
|
|
if result:
|
|
print(f" 第一条记录标题: {result[0]['title']}")
|
|
|
|
# 测试空关键词
|
|
print("\n3. 测试空关键词:")
|
|
result = search_sticker("", sticker_config_path)
|
|
print(f" 找到 {len(result)} 条记录")
|
|
if len(result) > 50:
|
|
print(" 应该只返回前50条记录")
|
|
else:
|
|
print(f" 返回了 {len(result)} 条记录")
|
|
|
|
print("\n测试完成!")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_search_sticker() |