K2cr2O1
caf5b06097
feat(ws_pool): 新增 WebSocket 连接池实现 perf(json): 使用 orjson 替代标准 json 库提升性能 style: 清理未使用的导入和冗余代码 docs: 更新架构文档和开发规范 test: 添加 WebSocket 连接池测试用例 fix(plugins): 修复自动审批插件 API 调用参数格式
87 lines
3.3 KiB
Python
87 lines
3.3 KiB
Python
"""
|
||
好友与陌生人相关 API 模块
|
||
|
||
该模块定义了 `FriendAPI` Mixin 类，提供了所有与好友、陌生人信息
|
||
等相关的 OneBot v11 API 封装。
|
||
"""
|
||
import orjson
|
||
from typing import List, Dict, Any
|
||
from .base import BaseAPI
|
||
from models.objects import FriendInfo, StrangerInfo
|
||
from ..managers.redis_manager import redis_manager
|
||
|
||
|
||
class FriendAPI(BaseAPI):
|
||
"""
|
||
`FriendAPI` Mixin 类，提供了所有与好友、陌生人操作相关的 API 方法。
|
||
"""
|
||
|
||
async def send_like(self, user_id: int, times: int = 1) -> Dict[str, Any]:
|
||
"""
|
||
向指定用户发送 "戳一戳" (点赞)。
|
||
|
||
Args:
|
||
user_id (int): 目标用户的 QQ 号。
|
||
times (int, optional): 点赞次数，建议不超过 10 次。Defaults to 1.
|
||
|
||
Returns:
|
||
Dict[str, Any]: OneBot API 的响应数据。
|
||
"""
|
||
return await self.call_api("send_like", {"user_id": user_id, "times": times})
|
||
|
||
async def get_stranger_info(self, user_id: int, no_cache: bool = False) -> StrangerInfo:
|
||
"""
|
||
获取陌生人的信息。
|
||
|
||
Args:
|
||
user_id (int): 目标用户的 QQ 号。
|
||
no_cache (bool, optional): 是否不使用缓存，直接从服务器获取。Defaults to False.
|
||
|
||
Returns:
|
||
StrangerInfo: 包含陌生人信息的 `StrangerInfo` 数据对象。
|
||
"""
|
||
cache_key = f"neobot:cache:get_stranger_info:{user_id}"
|
||
if not no_cache:
|
||
cached_data = await redis_manager.redis.get(cache_key)
|
||
if cached_data:
|
||
return StrangerInfo(**orjson.loads(cached_data))
|
||
|
||
res = await self.call_api("get_stranger_info", {"user_id": user_id, "no_cache": no_cache})
|
||
await redis_manager.redis.set(cache_key, orjson.dumps(res), ex=3600) # 缓存 1 小时
|
||
return StrangerInfo(**res)
|
||
|
||
async def get_friend_list(self, no_cache: bool = False) -> List[FriendInfo]:
|
||
"""
|
||
获取机器人账号的好友列表。
|
||
|
||
Args:
|
||
no_cache (bool, optional): 是否不使用缓存，直接从服务器获取最新信息。Defaults to False.
|
||
|
||
Returns:
|
||
List[FriendInfo]: 包含所有好友信息的 `FriendInfo` 对象列表。
|
||
"""
|
||
cache_key = f"neobot:cache:get_friend_list:{self.self_id}"
|
||
if not no_cache:
|
||
cached_data = await redis_manager.redis.get(cache_key)
|
||
if cached_data:
|
||
return [FriendInfo(**item) for item in orjson.loads(cached_data)]
|
||
|
||
res = await self.call_api("get_friend_list")
|
||
await redis_manager.redis.set(cache_key, orjson.dumps(res), ex=3600) # 缓存 1 小时
|
||
return [FriendInfo(**item) for item in res]
|
||
|
||
async def set_friend_add_request(self, flag: str, approve: bool = True, remark: str = "") -> Dict[str, Any]:
|
||
"""
|
||
处理收到的加好友请求。
|
||
|
||
Args:
|
||
flag (str): 请求的标识，需要从 `request` 事件中获取。
|
||
approve (bool, optional): 是否同意该好友请求。Defaults to True.
|
||
remark (str, optional): 在同意请求时，为该好友设置的备注。Defaults to "".
|
||
|
||
Returns:
|
||
Dict[str, Any]: OneBot API 的响应数据。
|
||
"""
|
||
return await self.call_api("set_friend_add_request", {"flag": flag, "approve": approve, "remark": remark})
|
||
|