"""个人信息API模块 提供个人信息查询功能,用于插件获取用户相关信息 使用方式: from src.plugin_system.apis import person_api person_id = person_api.get_person_id("qq", 123456) value = await person_api.get_person_value(person_id, "nickname") """ from typing import Any from src.common.logger import get_logger from src.person_info.person_info import Person logger = get_logger("person_api") # ============================================================================= # 个人信息API函数 # ============================================================================= def get_person_id(platform: str, user_id: int | str) -> str: """根据平台和用户ID获取person_id Args: platform: 平台名称,如 "qq", "telegram" 等 user_id: 用户ID Returns: str: 唯一的person_id(MD5哈希值) 示例: person_id = person_api.get_person_id("qq", 123456) """ try: return Person(platform=platform, user_id=str(user_id)).person_id except Exception as e: logger.error(f"[PersonAPI] 获取person_id失败: platform={platform}, user_id={user_id}, error={e}") return "" async def get_person_value(person_id: str, field_name: str, default: Any = None) -> Any: """根据person_id和字段名获取某个值 Args: person_id: 用户的唯一标识ID field_name: 要获取的字段名,如 "nickname", "impression" 等 default: 当字段不存在或获取失败时返回的默认值 Returns: Any: 字段值或默认值 示例: nickname = await person_api.get_person_value(person_id, "nickname", "未知用户") impression = await person_api.get_person_value(person_id, "impression") """ try: person = Person(person_id=person_id) value = getattr(person, field_name) return value if value is not None else default except Exception as e: logger.error(f"[PersonAPI] 获取用户信息失败: person_id={person_id}, field={field_name}, error={e}") return default def get_person_id_by_name(person_name: str) -> str: """根据用户名获取person_id Args: person_name: 用户名 Returns: str: person_id,如果未找到返回空字符串 示例: person_id = person_api.get_person_id_by_name("张三") """ try: person = Person(person_name=person_name) return person.person_id except Exception as e: logger.error(f"[PersonAPI] 根据用户名获取person_id失败: person_name={person_name}, error={e}") return ""