""" 表情API模块 提供表情包相关功能，采用标准Python包设计模式 使用方式： from src.plugin_system.apis import emoji_api result = await emoji_api.get_by_description("开心") count = emoji_api.get_count() """ import random from typing import Optional, Tuple, List from src.common.logger import get_logger from src.chat.emoji_system.emoji_manager import get_emoji_manager from src.chat.utils.utils_image import image_path_to_base64 logger = get_logger("emoji_api") # ============================================================================= # 表情包获取API函数 # ============================================================================= async def get_by_description(description: str) -> Optional[Tuple[str, str, str]]: """根据描述选择表情包 Args: description: 表情包的描述文本，例如"开心"、"难过"、"愤怒"等 Returns: Optional[Tuple[str, str, str]]: (base64编码, 表情包描述, 匹配的情感标签) 或 None Raises: ValueError: 如果描述为空字符串 TypeError: 如果描述不是字符串类型 """ if not description: raise ValueError("描述不能为空") if not isinstance(description, str): raise TypeError("描述必须是字符串类型") try: logger.debug(f"[EmojiAPI] 根据描述获取表情包: {description}") emoji_manager = get_emoji_manager() emoji_result = await emoji_manager.get_emoji_for_text(description) if not emoji_result: logger.warning(f"[EmojiAPI] 未找到匹配描述 '{description}' 的表情包") return None emoji_path, emoji_description, matched_emotion = emoji_result emoji_base64 = image_path_to_base64(emoji_path) if not emoji_base64: logger.error(f"[EmojiAPI] 无法将表情包文件转换为base64: {emoji_path}") return None logger.debug(f"[EmojiAPI] 成功获取表情包: {emoji_description}, 匹配情感: {matched_emotion}") return emoji_base64, emoji_description, matched_emotion except Exception as e: logger.error(f"[EmojiAPI] 获取表情包失败: {e}") return None async def get_random(count: Optional[int] = 1) -> List[Tuple[str, str, str]]: """随机获取指定数量的表情包 Args: count: 要获取的表情包数量，默认为1 Returns: List[Tuple[str, str, str]]: 包含(base64编码, 表情包描述, 随机情感标签)的元组列表，失败则返回空列表 Raises: TypeError: 如果count不是整数类型 ValueError: 如果count为负数 """ if not isinstance(count, int): raise TypeError("count 必须是整数类型") if count < 0: raise ValueError("count 不能为负数") if count == 0: logger.warning("[EmojiAPI] count 为0，返回空列表") return [] try: logger.info(f"[EmojiAPI] 随机获取 {count} 个表情包") emoji_manager = get_emoji_manager() all_emojis = emoji_manager.emoji_objects if not all_emojis: logger.warning("[EmojiAPI] 没有可用的表情包") return [] # 过滤有效表情包 valid_emojis = [emoji for emoji in all_emojis if not emoji.is_deleted] if not valid_emojis: logger.warning("[EmojiAPI] 没有有效的表情包") return [] if len(valid_emojis) < count: logger.warning( f"[EmojiAPI] 有效表情包数量 ({len(valid_emojis)}) 少于请求的数量 ({count})，将返回所有有效表情包" ) count = len(valid_emojis) # 随机选择 selected_emojis = random.sample(valid_emojis, count) results = [] for selected_emoji in selected_emojis: emoji_base64 = image_path_to_base64(selected_emoji.full_path) if not emoji_base64: logger.error(f"[EmojiAPI] 无法转换表情包为base64: {selected_emoji.full_path}") continue matched_emotion = random.choice(selected_emoji.emotion) if selected_emoji.emotion else "随机表情" # 记录使用次数 await emoji_manager.record_usage(selected_emoji.hash) results.append((emoji_base64, selected_emoji.description, matched_emotion)) if not results and count > 0: logger.warning("[EmojiAPI] 随机获取表情包失败，没有一个可以成功处理") return [] logger.info(f"[EmojiAPI] 成功获取 {len(results)} 个随机表情包") return results except Exception as e: logger.error(f"[EmojiAPI] 获取随机表情包失败: {e}") return [] async def get_by_emotion(emotion: str) -> Optional[Tuple[str, str, str]]: """根据情感标签获取表情包 Args: emotion: 情感标签，如"happy"、"sad"、"angry"等 Returns: Optional[Tuple[str, str, str]]: (base64编码, 表情包描述, 匹配的情感标签) 或 None Raises: ValueError: 如果情感标签为空字符串 TypeError: 如果情感标签不是字符串类型 """ if not emotion: raise ValueError("情感标签不能为空") if not isinstance(emotion, str): raise TypeError("情感标签必须是字符串类型") try: logger.info(f"[EmojiAPI] 根据情感获取表情包: {emotion}") emoji_manager = get_emoji_manager() all_emojis = emoji_manager.emoji_objects # 筛选匹配情感的表情包 matching_emojis = [] matching_emojis.extend( emoji_obj for emoji_obj in all_emojis if not emoji_obj.is_deleted and emotion.lower() in [e.lower() for e in emoji_obj.emotion] ) if not matching_emojis: logger.warning(f"[EmojiAPI] 未找到匹配情感 '{emotion}' 的表情包") return None # 随机选择匹配的表情包 selected_emoji = random.choice(matching_emojis) emoji_base64 = image_path_to_base64(selected_emoji.full_path) if not emoji_base64: logger.error(f"[EmojiAPI] 无法转换表情包为base64: {selected_emoji.full_path}") return None # 记录使用次数 await emoji_manager.record_usage(selected_emoji.hash) logger.info(f"[EmojiAPI] 成功获取情感表情包: {selected_emoji.description}") return emoji_base64, selected_emoji.description, emotion except Exception as e: logger.error(f"[EmojiAPI] 根据情感获取表情包失败: {e}") return None # ============================================================================= # 表情包信息查询API函数 # ============================================================================= def get_count() -> int: """获取表情包数量 Returns: int: 当前可用的表情包数量 """ try: emoji_manager = get_emoji_manager() return emoji_manager.emoji_num except Exception as e: logger.error(f"[EmojiAPI] 获取表情包数量失败: {e}") return 0 def get_info(): """获取表情包系统信息 Returns: dict: 包含表情包数量、最大数量、可用数量信息 """ try: emoji_manager = get_emoji_manager() return { "current_count": emoji_manager.emoji_num, "max_count": emoji_manager.emoji_num_max, "available_emojis": len([e for e in emoji_manager.emoji_objects if not e.is_deleted]), } except Exception as e: logger.error(f"[EmojiAPI] 获取表情包信息失败: {e}") return {"current_count": 0, "max_count": 0, "available_emojis": 0} def get_emotions() -> List[str]: """获取所有可用的情感标签 Returns: list: 所有表情包的情感标签列表（去重） """ try: emoji_manager = get_emoji_manager() emotions = set() for emoji_obj in emoji_manager.emoji_objects: if not emoji_obj.is_deleted and emoji_obj.emotion: emotions.update(emoji_obj.emotion) return sorted(list(emotions)) except Exception as e: logger.error(f"[EmojiAPI] 获取情感标签失败: {e}") return [] def get_descriptions() -> List[str]: """获取所有表情包描述 Returns: list: 所有可用表情包的描述列表 """ try: emoji_manager = get_emoji_manager() descriptions = [] descriptions.extend( emoji_obj.description for emoji_obj in emoji_manager.emoji_objects if not emoji_obj.is_deleted and emoji_obj.description ) return descriptions except Exception as e: logger.error(f"[EmojiAPI] 获取表情包描述失败: {e}") return []