89 lines
2.6 KiB
Python
89 lines
2.6 KiB
Python
|
#!/usr/bin/env python
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
直接测试同步API的Pydantic序列化
|
|||
|
|
不通过HTTP，直接模拟返回值构建过程
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
from database import User, get_db, init_database
|
|||
|
|
from app import SyncUsersResponse
|
|||
|
|
from pydantic import ValidationError
|
|||
|
|
import json
|
|||
|
|
|
|||
|
|
def test_sync_response():
|
|||
|
|
"""测试同步响应的序列化"""
|
|||
|
|
print("="*60)
|
|||
|
|
print("测试同步API响应序列化")
|
|||
|
|
print("="*60)
|
|||
|
|
|
|||
|
|
# 初始化数据库
|
|||
|
|
init_database()
|
|||
|
|
|
|||
|
|
# 获取数据库会话
|
|||
|
|
db = next(get_db())
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 查询所有激活用户
|
|||
|
|
users = db.query(User).filter(User.is_active == True).all()
|
|||
|
|
print(f"\n查询到 {len(users)} 个激活用户")
|
|||
|
|
|
|||
|
|
# 转换为字典列表
|
|||
|
|
user_list = []
|
|||
|
|
for i, user in enumerate(users):
|
|||
|
|
print(f"\n处理用户 {i+1}: {user.name}")
|
|||
|
|
user_dict = user.to_dict_with_feature()
|
|||
|
|
|
|||
|
|
# 打印字典的键
|
|||
|
|
print(f" 字典键: {list(user_dict.keys())}")
|
|||
|
|
|
|||
|
|
# 检查特征向量
|
|||
|
|
if 'feature_vector' in user_dict:
|
|||
|
|
feature = user_dict['feature_vector']
|
|||
|
|
print(f" 特征向量类型: {type(feature)}")
|
|||
|
|
print(f" 特征向量长度: {len(feature) if isinstance(feature, list) else 'N/A'}")
|
|||
|
|
if isinstance(feature, list) and len(feature) > 0:
|
|||
|
|
print(f" 第一个元素类型: {type(feature[0])}")
|
|||
|
|
|
|||
|
|
user_list.append(user_dict)
|
|||
|
|
|
|||
|
|
# 尝试构建Pydantic响应
|
|||
|
|
print("\n" + "="*60)
|
|||
|
|
print("尝试构建SyncUsersResponse对象...")
|
|||
|
|
print("="*60)
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
response = SyncUsersResponse(
|
|||
|
|
success=True,
|
|||
|
|
message="同步成功",
|
|||
|
|
total=len(user_list),
|
|||
|
|
users=user_list
|
|||
|
|
)
|
|||
|
|
print("✅ Pydantic模型验证成功!")
|
|||
|
|
|
|||
|
|
# 尝试序列化为JSON
|
|||
|
|
print("\n尝试序列化为JSON...")
|
|||
|
|
json_str = response.model_dump_json(indent=2)
|
|||
|
|
print("✅ JSON序列化成功!")
|
|||
|
|
print(f"\nJSON长度: {len(json_str)} 字符")
|
|||
|
|
|
|||
|
|
# 显示前500个字符
|
|||
|
|
if len(json_str) > 500:
|
|||
|
|
print(f"\n前500个字符:\n{json_str[:500]}...")
|
|||
|
|
else:
|
|||
|
|
print(f"\n完整JSON:\n{json_str}")
|
|||
|
|
|
|||
|
|
except ValidationError as e:
|
|||
|
|
print("❌ Pydantic验证失败!")
|
|||
|
|
print(f"错误详情:\n{e}")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"❌ JSON序列化失败: {e}")
|
|||
|
|
import traceback
|
|||
|
|
traceback.print_exc()
|
|||
|
|
|
|||
|
|
finally:
|
|||
|
|
db.close()
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
test_sync_response()
|