148 lines
4.8 KiB
Python
Executable File
148 lines
4.8 KiB
Python
Executable File
# -*- coding: utf-8 -*-
|
||
"""
|
||
客服对话推送到企业微信群 - 客户消息与AI回复成对发送，保持上下文
|
||
"""
|
||
import asyncio
|
||
import os
|
||
from datetime import datetime
|
||
|
||
import httpx
|
||
from dotenv import load_dotenv
|
||
|
||
load_dotenv()
|
||
|
||
_last_push: dict[tuple[str, str], tuple[str, str, float]] = {}
|
||
|
||
def _get_webhook() -> str:
|
||
"""优先从 config 读取，与健康检查/日报保持一致"""
|
||
try:
|
||
from config.config import WECHAT_WEBHOOK
|
||
return WECHAT_WEBHOOK or os.getenv("WECHAT_WEBHOOK", "")
|
||
except Exception:
|
||
return os.getenv("WECHAT_WEBHOOK", "")
|
||
|
||
|
||
def _truncate(text: str, max_len: int = 200) -> str:
|
||
"""截断过长内容"""
|
||
if not text:
|
||
return ""
|
||
text = str(text).strip()
|
||
if len(text) > max_len:
|
||
return text[:max_len] + "..."
|
||
return text
|
||
|
||
|
||
def _get_recent_conversation(customer_id: str, acc_id: str, last_n: int = 8) -> list:
|
||
"""获取近期对话（同店铺），保持连贯上下文"""
|
||
try:
|
||
from db.chat_log_db import get_recent_conversation
|
||
return get_recent_conversation(customer_id, acc_id, limit=last_n)
|
||
except Exception:
|
||
return []
|
||
|
||
|
||
async def push_chat_to_wechat(
|
||
customer_name: str,
|
||
customer_id: str,
|
||
acc_id: str,
|
||
customer_msg: str,
|
||
reply_msg: str,
|
||
goods_name: str = "",
|
||
):
|
||
"""
|
||
将客户消息与AI回复推送到企业微信群，附带近期对话保持连贯。
|
||
"""
|
||
webhook = _get_webhook()
|
||
if not webhook:
|
||
return
|
||
# 去重：同一客户+店铺，若客户消息与回复完全相同且在窗口期内，则跳过
|
||
try:
|
||
import time
|
||
key = (customer_id or "", acc_id or "")
|
||
now = time.time()
|
||
last = _last_push.get(key)
|
||
if last:
|
||
last_customer_msg, last_reply_msg, last_ts = last
|
||
if (last_customer_msg or "") == (customer_msg or "") and (last_reply_msg or "") == (reply_msg or ""):
|
||
if now - last_ts < 30:
|
||
return
|
||
_last_push[key] = ((customer_msg or ""), (reply_msg or ""), now)
|
||
except Exception:
|
||
pass
|
||
reply_msg = _truncate(reply_msg, 300)
|
||
ts = datetime.now().strftime("%H:%M")
|
||
shop = acc_id or "未知店铺"
|
||
name = (customer_name or customer_id or "客户")[:12]
|
||
|
||
lines = [f"**📩 {ts} | {shop}**"]
|
||
if goods_name:
|
||
lines.append(f"**商品** {_truncate(goods_name, 80)}")
|
||
if customer_id:
|
||
lines.append(f"**客户ID** {customer_id}")
|
||
lines.append("")
|
||
|
||
# 附带近期对话，保持连贯
|
||
recent = _get_recent_conversation(customer_id, acc_id, last_n=8)
|
||
last_line = None
|
||
for m in recent:
|
||
role = customer_id if m.get("direction") == "in" else "客服"
|
||
msg = _truncate((m.get("message") or "").strip(), 120)
|
||
if msg:
|
||
line = f"{role}：{msg}"
|
||
# 防止日志中的重复记录在企微里连续刷屏
|
||
if line == last_line:
|
||
continue
|
||
lines.append(line)
|
||
last_line = line
|
||
# 当前回复（可能已在 recent 中有客户消息，客服回复是新的）
|
||
lines.append(f"客服：{reply_msg or '(无回复)'}")
|
||
|
||
content = "\n".join(lines)
|
||
enc = content.encode("utf-8")
|
||
if len(enc) > 3800:
|
||
content = enc[:3750].decode("utf-8", errors="ignore") + "\n...(略)"
|
||
try:
|
||
async with httpx.AsyncClient(timeout=8) as client:
|
||
resp = await client.post(
|
||
webhook,
|
||
json={"msgtype": "markdown", "markdown": {"content": content}},
|
||
)
|
||
data = resp.json()
|
||
if data.get("errcode") == 0:
|
||
pass # 成功静默
|
||
else:
|
||
print(f"[WechatChatLog] 推送失败: {data}")
|
||
except Exception as e:
|
||
print(f"[WechatChatLog] 推送异常: {e}")
|
||
|
||
|
||
async def send_morning_startup():
|
||
"""每天早上8点发送客服启动消息到企微群"""
|
||
webhook = _get_webhook()
|
||
if not webhook:
|
||
return
|
||
ts = datetime.now().strftime("%Y-%m-%d %H:%M")
|
||
content = f"**☀️ 客服已启动**\n{ts}"
|
||
try:
|
||
async with httpx.AsyncClient(timeout=8) as client:
|
||
await client.post(
|
||
webhook,
|
||
json={"msgtype": "markdown", "markdown": {"content": content}},
|
||
)
|
||
print(f"[WechatChatLog] 早8点启动消息已发送")
|
||
except Exception as e:
|
||
print(f"[WechatChatLog] 启动消息发送失败: {e}")
|
||
|
||
|
||
async def morning_startup_scheduler():
|
||
"""每天 8:00 发送启动消息"""
|
||
print("[WechatChatLog] 早8点启动消息定时任务已启动")
|
||
sent_today = None
|
||
while True:
|
||
now = datetime.now()
|
||
today = now.strftime("%Y-%m-%d")
|
||
if now.hour == 8 and now.minute == 0 and sent_today != today:
|
||
sent_today = today
|
||
await send_morning_startup()
|
||
await asyncio.sleep(30)
|