feat(staff-friends): 员工客户列表独立页——对齐云客员工详情,好友数/微信群数下钻

员工列表点「好友数」原先跳全局 /customers?owner_account_id=x,布局与全局列表无异且计数/分页绑全局声明口径(同日 dfa4fdd 修的显示 bug),与云客「员工名下客户列表」体验割裂。新页 /staff-friends/:ownerId/:tab: - 微信客户/微信群两 tab(对齐云客截图);标题=员工显示名(入口直带, 直链经 /api/accounts 解析),返回箭头回员工列表 - 搜索(昵称/微信号/备注名/手机号,服务端 q)+ 查询/导出 CSV - 发送/接收条数=Customer.send_num/receive_num 累计值(CustomerOut 补两列,同步自云客 friend 记录,与截图同源) - 勾选批量「导出聊天记录」:新端点 GET /api/export/chat-html-zip,复用 chat-html 渲染器每会话一个 HTML 打 zip(文件名=备注>昵称>friend_id, 非法字符清洗;单会话 5000 行截断告警;一次上限 50 会话防误选打爆) - 行内:查看聊天记录→staff-chat;导出HTML聊天记录(时间范围弹窗) - WA 行保持原入口(该页暂仅微信) 后端 605 测试绿(zip 新 4 例);本地浏览器端到端验证:过滤 9 条/勾选激活/群 tab/返回箭头;zip 产物 application/zip 双会话双气泡。 Co-Authored-By: Claude Code <noreply@anthropic.com>
2026-09-08 16:14:44 +08:00
parent dfa4fdd420
commit 2e526cd056
7 changed files with 812 additions and 6 deletions
@@ -1,16 +1,19 @@
-"""Tests for GET /api/export/chat-html — 导出 1:1/群聊天为自包含 HTML。
+"""Tests for GET /api/export/chat-html(+zip) — 导出 1:1/群聊天为自包含 HTML。

 对齐云客微信客户/群列表「导出HTML聊天记录」：按 (channel, owner, talker_id|room_id)
 拉消息（可选 start/end 时间范围），渲染为左右气泡 HTML 作为 attachment 返回下载。
+zip 版:员工客户页勾选多个客户批量「导出聊天记录」(2026-09-08)。
 """

 from __future__ import annotations

 import functools
+import io
+import zipfile
 from sqlalchemy.ext.asyncio import AsyncSession

-from app.api.routers.export import EXPORT_MAX_ROWS
-from app.db.models import Message
+from app.api.routers.export import EXPORT_MAX_ROWS, ZIP_MAX_CONVS
+from app.db.models import Customer, Message


 async def _seed(session: AsyncSession) -> None:
@@ -160,3 +163,91 @@ async def test_export_group_ignores_owner_mismatch(make_client):
    assert r.status_code == 200
    assert "群消息1" in r.text
    assert "群消息2" in r.text
+
+
+# ---------------------------------------------------------------------------
+# GET /api/export/chat-html-zip — 批量导出(2026-09-08 员工客户页)
+# ---------------------------------------------------------------------------
+
+async def _seed_zip(session: AsyncSession) -> None:
+    """两个客户各几条 DM + 一条同 talker 的群消息(不应混入 DM zip)。"""
+    session.add_all([
+        Customer(channel="wechat", owner_account_id="wx_owner", friend_id="f1",
+                 friend_nickname="小蜜", friend_remark="蜜姐"),
+        Customer(channel="wechat", owner_account_id="wx_owner", friend_id="f2",
+                 friend_nickname="老王"),
+        Customer(channel="wechat", owner_account_id="wx_owner", friend_id="f3",
+                 friend_nickname='坏/名字<>:"|'),
+        Message(msg_id="z1", channel="wechat", owner_account_id="wx_owner",
+                talker_id="f1", room_id=None, text="f1你好", timestamp=1000,
+                is_send=False, type="text", type_raw=1),
+        Message(msg_id="z2", channel="wechat", owner_account_id="wx_owner",
+                talker_id="f2", room_id=None, text="f2在吗", timestamp=2000,
+                is_send=True, type="text", type_raw=1),
+        Message(msg_id="z3", channel="wechat", owner_account_id="wx_owner",
+                talker_id="f2", room_id=None, text="f2更早", timestamp=500,
+                is_send=True, type="text", type_raw=1),
+        # f1 名下的群消息:room_id 有值,zip 的 DM 查询必须排除
+        Message(msg_id="z4", channel="wechat", owner_account_id="wx_owner",
+                talker_id="f1", room_id="g@chatroom", text="群内消息不进zip",
+                timestamp=3000, is_send=True, type="text", type_raw=1),
+    ])
+    await session.commit()
+
+
+def _zip_entries(r) -> dict[str, str]:
+    assert r.headers["content-type"].startswith("application/zip")
+    with zipfile.ZipFile(io.BytesIO(r.content)) as zf:
+        return {name: zf.read(name).decode("utf-8") for name in zf.namelist()}
+
+
+async def test_zip_two_conversations_with_display_names(make_client):
+    """zip 含两个会话 HTML;文件名取备注>昵称;群消息不混入。"""
+    client = await make_client(_seed_zip)
+    r = await client.get("/api/export/chat-html-zip", params={
+        "channel": "wechat", "owner_account_id": "wx_owner",
+        "talker_ids": "f1,f2"})
+    assert r.status_code == 200
+    assert "attachment" in r.headers["content-disposition"]
+    entries = _zip_entries(r)
+    assert sorted(entries) == ["01_蜜姐.html", "02_老王.html"]
+    assert "f1你好" in entries["01_蜜姐.html"]
+    assert "f2在吗" in entries["02_老王.html"]
+    # DM 查询排除 room 消息
+    assert "群内消息不进zip" not in "".join(entries.values())
+
+
+async def test_zip_time_range_filter(make_client):
+    """start/end 同样作用于 zip 内每个会话。"""
+    client = await make_client(_seed_zip)
+    r = await client.get("/api/export/chat-html-zip", params={
+        "channel": "wechat", "owner_account_id": "wx_owner",
+        "talker_ids": "f2", "start": 1000})
+    assert r.status_code == 200
+    entries = _zip_entries(r)
+    assert "f2在吗" in entries["01_老王.html"]
+    assert "f2更早" not in entries["01_老王.html"]
+
+
+async def test_zip_filename_sanitized_and_fallback(make_client):
+    """非法字符客户名清洗;无 customer 行的 talker 回退 friend_id。"""
+    client = await make_client(_seed_zip)
+    r = await client.get("/api/export/chat-html-zip", params={
+        "channel": "wechat", "owner_account_id": "wx_owner",
+        "talker_ids": "f3,f_unknown"})
+    assert r.status_code == 200
+    names = sorted(_zip_entries(r))
+    assert names[0].startswith("01_") and "/" not in names[0] and "<" not in names[0]
+    assert names[1] == "02_f_unknown.html"
+
+
+async def test_zip_rejects_empty_and_over_limit(make_client):
+    """空列表/超上限 422,防误传。"""
+    client = await make_client(_seed_zip)
+    r_empty = await client.get("/api/export/chat-html-zip", params={
+        "channel": "wechat", "owner_account_id": "wx_owner", "talker_ids": " , "})
+    assert r_empty.status_code == 422
+    many = ",".join(f"t{i}" for i in range(ZIP_MAX_CONVS + 1))
+    r_many = await client.get("/api/export/chat-html-zip", params={
+        "channel": "wechat", "owner_account_id": "wx_owner", "talker_ids": many})
+    assert r_many.status_code == 422