mirror of https://github.com/Mai-with-u/MaiBot.git
122 lines
3.7 KiB
Python
122 lines
3.7 KiB
Python
from typing import Optional
|
||
from src.config.config import global_config
|
||
from src.chat.frequency_control.utils import parse_stream_config_to_chat_id
|
||
|
||
|
||
def get_config_base_focus_value(chat_id: Optional[str] = None) -> float:
|
||
"""
|
||
根据当前时间和聊天流获取对应的 focus_value
|
||
"""
|
||
if not global_config.chat.focus_value_adjust:
|
||
return global_config.chat.focus_value
|
||
|
||
if chat_id:
|
||
stream_focus_value = get_stream_specific_focus_value(chat_id)
|
||
if stream_focus_value is not None:
|
||
return stream_focus_value
|
||
|
||
global_focus_value = get_global_focus_value()
|
||
if global_focus_value is not None:
|
||
return global_focus_value
|
||
|
||
return global_config.chat.focus_value
|
||
|
||
|
||
def get_stream_specific_focus_value(chat_id: str) -> Optional[float]:
|
||
"""
|
||
获取特定聊天流在当前时间的专注度
|
||
|
||
Args:
|
||
chat_stream_id: 聊天流ID（哈希值）
|
||
|
||
Returns:
|
||
float: 专注度值，如果没有配置则返回 None
|
||
"""
|
||
# 查找匹配的聊天流配置
|
||
for config_item in global_config.chat.focus_value_adjust:
|
||
if not config_item or len(config_item) < 2:
|
||
continue
|
||
|
||
stream_config_str = config_item[0] # 例如 "qq:1026294844:group"
|
||
|
||
# 解析配置字符串并生成对应的 chat_id
|
||
config_chat_id = parse_stream_config_to_chat_id(stream_config_str)
|
||
if config_chat_id is None:
|
||
continue
|
||
|
||
# 比较生成的 chat_id
|
||
if config_chat_id != chat_id:
|
||
continue
|
||
|
||
# 使用通用的时间专注度解析方法
|
||
return get_time_based_focus_value(config_item[1:])
|
||
|
||
return None
|
||
|
||
|
||
def get_time_based_focus_value(time_focus_list: list[str]) -> Optional[float]:
|
||
"""
|
||
根据时间配置列表获取当前时段的专注度
|
||
|
||
Args:
|
||
time_focus_list: 时间专注度配置列表，格式为 ["HH:MM,focus_value", ...]
|
||
|
||
Returns:
|
||
float: 专注度值，如果没有配置则返回 None
|
||
"""
|
||
from datetime import datetime
|
||
|
||
current_time = datetime.now().strftime("%H:%M")
|
||
current_hour, current_minute = map(int, current_time.split(":"))
|
||
current_minutes = current_hour * 60 + current_minute
|
||
|
||
# 解析时间专注度配置
|
||
time_focus_pairs = []
|
||
for time_focus_str in time_focus_list:
|
||
try:
|
||
time_str, focus_str = time_focus_str.split(",")
|
||
hour, minute = map(int, time_str.split(":"))
|
||
focus_value = float(focus_str)
|
||
minutes = hour * 60 + minute
|
||
time_focus_pairs.append((minutes, focus_value))
|
||
except (ValueError, IndexError):
|
||
continue
|
||
|
||
if not time_focus_pairs:
|
||
return None
|
||
|
||
# 按时间排序
|
||
time_focus_pairs.sort(key=lambda x: x[0])
|
||
|
||
# 查找当前时间对应的专注度
|
||
current_focus_value = None
|
||
for minutes, focus_value in time_focus_pairs:
|
||
if current_minutes >= minutes:
|
||
current_focus_value = focus_value
|
||
else:
|
||
break
|
||
|
||
# 如果当前时间在所有配置时间之前，使用最后一个时间段的专注度（跨天逻辑）
|
||
if current_focus_value is None and time_focus_pairs:
|
||
current_focus_value = time_focus_pairs[-1][1]
|
||
|
||
return current_focus_value
|
||
|
||
|
||
def get_global_focus_value() -> Optional[float]:
|
||
"""
|
||
获取全局默认专注度配置
|
||
|
||
Returns:
|
||
float: 专注度值，如果没有配置则返回 None
|
||
"""
|
||
for config_item in global_config.chat.focus_value_adjust:
|
||
if not config_item or len(config_item) < 2:
|
||
continue
|
||
|
||
# 检查是否为全局默认配置（第一个元素为空字符串）
|
||
if config_item[0] == "":
|
||
return get_time_based_focus_value(config_item[1:])
|
||
|
||
return None
|