55 lines
2.1 KiB
Python
55 lines
2.1 KiB
Python
import logging
|
|
from typing import Generator, Optional
|
|
from src.config import Config
|
|
from src.email_client import fetch_unseen_emails, mark_as_seen, Email
|
|
from src.ai_client import summarize_email
|
|
from src.tg_bot import send_summary, format_summary
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def poll_accounts(cfg: Config) -> Generator[tuple[int, Email], None, None]:
|
|
for idx, acct in enumerate(cfg.email_accounts):
|
|
try:
|
|
logger.info(f"检查邮箱: {acct.username}")
|
|
emails = fetch_unseen_emails(acct)
|
|
if emails:
|
|
logger.info(f" 发现 {len(emails)} 封新邮件")
|
|
for mail in emails:
|
|
yield idx, mail
|
|
except Exception as e:
|
|
logger.error(f"轮询 {acct.username} 失败: {e}", exc_info=True)
|
|
|
|
|
|
def ai_process(cfg: Config, acct_idx: int, mail: Email) -> Optional[dict]:
|
|
logger.info(f" 正在摘要: {mail.subject}")
|
|
summary = summarize_email(cfg.ai, mail.recipient, mail.subject, mail.sender, mail.body, mail.account_email)
|
|
summary["recipient"] = mail.recipient
|
|
if "@" not in summary.get("sender", ""):
|
|
summary["sender"] = mail.sender
|
|
text = format_summary(summary, mail.account_email)
|
|
return {
|
|
"text": text,
|
|
"data": summary,
|
|
"original_body": mail.body,
|
|
"original_sender": mail.sender,
|
|
"original_recipient": mail.recipient,
|
|
"original_reply_to": mail.reply_to,
|
|
"acct_idx": acct_idx,
|
|
"account_email": mail.account_email,
|
|
"uid": mail.uid,
|
|
}
|
|
|
|
|
|
def tg_send_and_mark(cfg: Config, info: dict):
|
|
acct = cfg.email_accounts[info["acct_idx"]]
|
|
send_summary(cfg.telegram, cfg.telegram.chat_id,
|
|
info["text"], info["data"],
|
|
info["original_body"], info["acct_idx"],
|
|
original_sender=info.get("original_sender", ""),
|
|
original_recipient=info.get("original_recipient", ""),
|
|
original_reply_to=info.get("original_reply_to", ""),
|
|
account_email=info.get("account_email", ""))
|
|
mark_as_seen(acct, [info["uid"]])
|
|
logger.info(f" 已发送并标记已读")
|