mirror of
https://github.com/cdryzun/tg_bot_collections.git
synced 2025-11-04 16:56:43 +08:00
0b60ae2fbe
* feat: add summary and search commands Signed-off-by: Frost Ming <me@frostming.com> * fix formats Signed-off-by: Frost Ming <me@frostming.com> * fix: clean up Signed-off-by: Frost Ming <me@frostming.com>
49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
import re
|
|
import zoneinfo
|
|
from datetime import datetime, timedelta
|
|
|
|
from telebot import TeleBot
|
|
from telebot.types import Message
|
|
|
|
PROMPT = """\
|
|
请将下面的聊天记录进行总结，包含讨论了哪些话题，有哪些亮点发言和主要观点。
|
|
引用用户名请加粗。直接返回内容即可，不要包含引导词和标题。
|
|
--- Messages Start ---
|
|
{messages}
|
|
--- Messages End ---
|
|
"""
|
|
|
|
|
|
def filter_message(message: Message, bot: TeleBot) -> bool:
|
|
"""过滤消息，排除非文本消息和命令消息"""
|
|
if not message.text:
|
|
return False
|
|
if not message.from_user:
|
|
return False
|
|
if message.from_user.id == bot.get_me().id:
|
|
return False
|
|
if message.text.startswith("/"):
|
|
return False
|
|
return True
|
|
|
|
|
|
date_regex = re.compile(r"^(\d+)([dhm])$")
|
|
|
|
|
|
def parse_date(date_str: str, locale: str) -> tuple[datetime, datetime]:
|
|
date_str = date_str.strip().lower()
|
|
now = datetime.now(tz=zoneinfo.ZoneInfo(locale))
|
|
if date_str == "today":
|
|
return now.replace(hour=0, minute=0, second=0, microsecond=0), now
|
|
elif m := date_regex.match(date_str):
|
|
number = int(m.group(1))
|
|
unit = m.group(2)
|
|
match unit:
|
|
case "d":
|
|
return now - timedelta(days=number), now
|
|
case "h":
|
|
return now - timedelta(hours=number), now
|
|
case "m":
|
|
return now - timedelta(minutes=number), now
|
|
raise ValueError(f"Unsupported date format: {date_str}")
|