47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
from typing import Optional
|
||
from utils.revoke.message_revoke_manager import MessageRevokeManager
|
||
from wechat_ipad import WechatAPIClient
|
||
|
||
|
||
class MessageAutoRevoke:
|
||
"""消息自动撤回工具类"""
|
||
|
||
def __init__(self, client: WechatAPIClient, default_delay: float = 90.0):
|
||
"""
|
||
初始化消息自动撤回工具
|
||
|
||
Args:
|
||
client: WechatAPIClient实例
|
||
default_delay: 默认撤回延时（秒）
|
||
"""
|
||
self.client = client
|
||
self.revoke_manager = MessageRevokeManager(default_delay)
|
||
# 设置撤回回调函数
|
||
self.revoke_manager.set_revoke_callback(self.client.revoke_message)
|
||
|
||
def add_message_to_revoke(self, wxid: str, client_msg_id: int, create_time: int, new_msg_id: int,
|
||
delay: Optional[float] = None) -> None:
|
||
"""
|
||
添加消息到撤回队列
|
||
|
||
Args:
|
||
wxid: 接收人wxid
|
||
client_msg_id: 消息ID
|
||
create_time: 消息创建时间
|
||
new_msg_id: 新消息ID
|
||
delay: 延迟撤回时间（秒），如果为None则使用默认值
|
||
"""
|
||
self.revoke_manager.add_message_to_revoke(wxid, client_msg_id, create_time, new_msg_id, delay)
|
||
|
||
def set_default_delay(self, delay: float) -> None:
|
||
"""
|
||
设置默认撤回延时
|
||
|
||
Args:
|
||
delay: 默认撤回延时（秒）
|
||
"""
|
||
self.revoke_manager.set_default_delay(delay)
|
||
|
||
def stop(self) -> None:
|
||
"""停止消息撤回任务"""
|
||
self.revoke_manager.stop() |