135 lines
5.3 KiB
Python
135 lines
5.3 KiB
Python
import asyncio
|
||
import time
|
||
import traceback
|
||
|
||
from typing import Optional, List, Dict, Tuple
|
||
from rich.traceback import install
|
||
|
||
from src.common.logger import get_logger
|
||
from src.config.config import global_config
|
||
from src.chat.message_receive.message import MessageRecv
|
||
from src.chat.message_receive.chat_stream import get_chat_manager
|
||
from src.chat.focus_chat.heartFC_chat import HeartFChatting
|
||
from src.chat.utils.utils import get_chat_type_and_target_info
|
||
|
||
logger = get_logger("sub_heartflow")
|
||
|
||
install(extra_lines=3)
|
||
|
||
|
||
class SubHeartflow:
|
||
def __init__(
|
||
self,
|
||
subheartflow_id,
|
||
):
|
||
"""子心流初始化函数
|
||
|
||
Args:
|
||
subheartflow_id: 子心流唯一标识符
|
||
"""
|
||
# 基础属性，两个值是一样的
|
||
self.subheartflow_id = subheartflow_id
|
||
self.chat_id = subheartflow_id
|
||
|
||
self.is_group_chat, self.chat_target_info = get_chat_type_and_target_info(self.chat_id)
|
||
self.log_prefix = get_chat_manager().get_stream_name(self.subheartflow_id) or self.subheartflow_id
|
||
|
||
# focus模式退出冷却时间管理
|
||
self.last_focus_exit_time: float = 0 # 上次退出focus模式的时间
|
||
|
||
# 随便水群 normal_chat 和 认真水群 focus_chat 实例
|
||
# CHAT模式激活 随便水群 FOCUS模式激活 认真水群
|
||
self.heart_fc_instance: Optional[HeartFChatting] = HeartFChatting(
|
||
chat_id=self.subheartflow_id,
|
||
) # 该sub_heartflow的HeartFChatting实例
|
||
|
||
async def initialize(self):
|
||
"""异步初始化方法，创建兴趣流并确定聊天类型"""
|
||
await self.heart_fc_instance.start()
|
||
|
||
async def _stop_heart_fc_chat(self):
|
||
"""停止并清理 HeartFChatting 实例"""
|
||
if self.heart_fc_instance.running:
|
||
logger.info(f"{self.log_prefix} 结束专注聊天...")
|
||
try:
|
||
await self.heart_fc_instance.shutdown()
|
||
except Exception as e:
|
||
logger.error(f"{self.log_prefix} 关闭 HeartFChatting 实例时出错: {e}")
|
||
logger.error(traceback.format_exc())
|
||
else:
|
||
logger.info(f"{self.log_prefix} 没有专注聊天实例，无需停止专注聊天")
|
||
|
||
async def _start_heart_fc_chat(self) -> bool:
|
||
"""启动 HeartFChatting 实例，确保 NormalChat 已停止"""
|
||
try:
|
||
# 如果任务已完成或不存在，则尝试重新启动
|
||
if self.heart_fc_instance._loop_task is None or self.heart_fc_instance._loop_task.done():
|
||
logger.info(f"{self.log_prefix} HeartFChatting 实例存在但循环未运行，尝试启动...")
|
||
try:
|
||
# 添加超时保护
|
||
await asyncio.wait_for(self.heart_fc_instance.start(), timeout=15.0)
|
||
logger.info(f"{self.log_prefix} HeartFChatting 循环已启动。")
|
||
return True
|
||
except Exception as e:
|
||
logger.error(f"{self.log_prefix} 尝试启动现有 HeartFChatting 循环时出错: {e}")
|
||
logger.error(traceback.format_exc())
|
||
# 出错时清理实例，准备重新创建
|
||
self.heart_fc_instance = None
|
||
else:
|
||
# 任务正在运行
|
||
logger.debug(f"{self.log_prefix} HeartFChatting 已在运行中。")
|
||
return True # 已经在运行
|
||
|
||
except Exception as e:
|
||
logger.error(f"{self.log_prefix} _start_heart_fc_chat 执行时出错: {e}")
|
||
logger.error(traceback.format_exc())
|
||
return False
|
||
|
||
def is_in_focus_cooldown(self) -> bool:
|
||
"""检查是否在focus模式的冷却期内
|
||
|
||
Returns:
|
||
bool: 如果在冷却期内返回True，否则返回False
|
||
"""
|
||
if self.last_focus_exit_time == 0:
|
||
return False
|
||
|
||
# 基础冷却时间10分钟，受auto_focus_threshold调控
|
||
base_cooldown = 10 * 60 # 10分钟转换为秒
|
||
cooldown_duration = base_cooldown / global_config.chat.auto_focus_threshold
|
||
|
||
current_time = time.time()
|
||
elapsed_since_exit = current_time - self.last_focus_exit_time
|
||
|
||
is_cooling = elapsed_since_exit < cooldown_duration
|
||
|
||
if is_cooling:
|
||
remaining_time = cooldown_duration - elapsed_since_exit
|
||
remaining_minutes = remaining_time / 60
|
||
logger.debug(
|
||
f"[{self.log_prefix}] focus冷却中，剩余时间: {remaining_minutes:.1f}分钟 (阈值: {global_config.chat.auto_focus_threshold})"
|
||
)
|
||
|
||
return is_cooling
|
||
|
||
def get_cooldown_progress(self) -> float:
|
||
"""获取冷却进度，返回0-1之间的值
|
||
|
||
Returns:
|
||
float: 0表示刚开始冷却，1表示冷却完成
|
||
"""
|
||
if self.last_focus_exit_time == 0:
|
||
return 1.0 # 没有冷却，返回1表示完全恢复
|
||
|
||
# 基础冷却时间10分钟，受auto_focus_threshold调控
|
||
base_cooldown = 10 * 60 # 10分钟转换为秒
|
||
cooldown_duration = base_cooldown / global_config.chat.auto_focus_threshold
|
||
|
||
current_time = time.time()
|
||
elapsed_since_exit = current_time - self.last_focus_exit_time
|
||
|
||
if elapsed_since_exit >= cooldown_duration:
|
||
return 1.0 # 冷却完成
|
||
|
||
return elapsed_since_exit / cooldown_duration
|