freedak/wxsales
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
5926c73ff12cc1e50cfad9dbb893ca7ecf8022ae
wxsales/demo/agent/mock_sync.py
T
freedakgmail 5926c73ff1 微信营销管理系统 - 益童宝销售管理平台 
- 数据库: PostgreSQL schema + 3名销售/21客户/341消息/6成交
- 采集代理: mock_sync.py 模拟微信聊天同步
- AI分析: analyze.py 规则模式 + 千问LLM模式
- 后端: FastAPI 11个API接口
- 前端: 仪表盘/销售列表/客户列表/成交记录/交流分析/录入成交
- 交流分析: 全部客户对话概览 + LLM标准范式对话生成
- 部署: systemd + nginx, 已部署至 sale.all8ai.top
2026-07-14 22:31:29 +08:00

741 lines
32 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters
This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.
#!/usr/bin/env python3
"""模拟销售设备采集代理 — 生成模拟微信数据并同步到中央 PostgreSQL。
用法:
python mock_sync.py --salesperson-id 1
python mock_sync.py --salesperson-id 1 --dsn "host=127.0.0.1 port=5434 dbname=wxchat_sales"
"""
from __future__ import annotations
import argparse
import datetime as dt
import hashlib
import json
import pathlib
import random
import sys
import psycopg2
from psycopg2.extras import execute_values
TZ = dt.timezone(dt.timedelta(hours=8))
# ---------------------------------------------------------------------------
# 数据库连接
# ---------------------------------------------------------------------------
DEFAULT_SOCKET = str(pathlib.Path(__file__).resolve().parent.parent / "db" / "socket")
DEFAULT_DSN = f"host={DEFAULT_SOCKET} port=5434 dbname=wxchat_sales"
# ---------------------------------------------------------------------------
# 销售人员信息
# ---------------------------------------------------------------------------
SALESPERSONS = {
1: {"name": "张伟", "wx_account": "wxid_zhangwei", "device_id": "macbook-zw-001"},
2: {"name": "李娜", "wx_account": "wxid_lina", "device_id": "macbook-ln-002"},
3: {"name": "王强", "wx_account": "wxid_wangqiang", "device_id": "macbook-wq-003"},
}
# ---------------------------------------------------------------------------
# 场景模板 — 真实客户对话
# ---------------------------------------------------------------------------
SCENARIO_TEMPLATES = [
{
"name": "湿疹宝宝咨询后成交",
"stage": "成交",
"contact": {
"nickname": "辰辰妈妈",
"remark": "辰辰妈-湿疹-2岁",
"industry": "宝妈",
"intent_level": "high",
},
"messages": [
("customer", "text", "你好，我家宝宝2岁，湿疹反复好几个月了，朋友推荐你这边益生菌"),
("sales", "text", "辰辰妈您好！宝宝湿疹确实让人心疼，请问现在湿疹主要在哪些部位？有用过什么药吗？"),
("customer", "text", "脸上和手臂都有，医生开了激素药膏，但停了就复发"),
("sales", "text", "理解，激素药膏只能暂时压制。益生菌是从肠道调节免疫，从根本上降低过敏反应。我们用的是丹麦进口的鼠李糖乳杆菌，有专门针对儿童湿疹的临床验证"),
("sales", "image", "[图片：临床验证报告截图]"),
("customer", "text", "这个是进口的？安全吗？2岁能吃吗？"),
("sales", "text", "是的，丹麦进口菌株，0岁以上就能用，无敏配方，不含牛奶蛋白和麸质。很多宝妈反馈坚持吃2-3个月湿疹明显好转"),
("customer", "text", "多少钱？怎么卖的？"),
("sales", "text", "单盒298元30袋，建议先吃3盒一个周期，3盒套餐798元算下来每天不到9块钱"),
("sales", "image", "[图片：产品包装图]"),
("customer", "text", "3盒798是吧，效果不好怎么办？"),
("sales", "text", "我们有售后指导，期间有任何问题随时找我。另外我发您几个同情况宝妈的反馈看看"),
("sales", "link", "[链接：宝妈真实反馈合集]"),
("customer", "text", "好的，那先来3盒试试"),
("sales", "text", "好的辰辰妈！3盒套餐798元，您方便现在付款吗？我这边给您安排发货"),
("customer", "text", "已经转过去了，麻烦尽快发货"),
("sales", "text", "收到！今天就给您发顺丰，预计后天到"),
("customer", "emoji", "[表情：谢谢]"),
],
},
{
"name": "过敏性鼻炎咨询后犹豫",
"stage": "异议处理",
"contact": {
"nickname": "乐乐妈",
"remark": "乐乐妈-鼻炎-5岁",
"industry": "宝妈",
"intent_level": "medium",
},
"messages": [
("customer", "text", "你好，听说益生菌对过敏性鼻炎有帮助？"),
("sales", "text", "乐乐妈您好！是的，益生菌可以调节肠道免疫，对过敏性鼻炎有改善作用。请问宝宝现在鼻炎什么症状？"),
("customer", "text", "早上起来一直打喷嚏流鼻涕，医生说是过敏性鼻炎"),
("sales", "text", "益生菌从肠道调节免疫系统，坚持吃可以逐渐降低过敏反应。我们的菌株是丹麦进口的，有临床验证"),
("customer", "text", "有没有副作用？安全吗？"),
("sales", "text", "很安全的，无敏配方，不含牛奶蛋白和麸质，5岁完全没问题"),
("customer", "text", "多少钱？"),
("sales", "text", "单盒298，3盒套餐798"),
("customer", "text", "有点贵啊..."),
("sales", "text", "算下来每天不到9块钱，比吃药划算多了"),
("customer", "text", "我再想想，和老公商量一下"),
("sales", "text", "好的，不着急。您可以先看看这个科普文章"),
("sales", "link", "[链接：益生菌与儿童过敏科普]"),
("customer", "text", "好的，我看看"),
],
},
{
"name": "朋友推荐来咨询",
"stage": "需求发现",
"contact": {
"nickname": "果果妈妈",
"remark": "果果妈-推荐来的",
"industry": "宝妈",
"intent_level": "medium",
},
"messages": [
("customer", "text", "你好，是小林推荐我加你的，她说你家益生菌不错"),
("sales", "text", "果果妈您好！谢谢小林推荐，请问宝宝多大了？有什么情况吗？"),
("customer", "text", "1岁半，最近总是拉肚子，小林说吃益生菌调理一下"),
("sales", "text", "拉肚子多久了？有去看过医生吗？"),
("customer", "text", "断断续续一个多月了，医生说没什么大问题，让注意饮食"),
("sales", "text", "那益生菌确实可以帮到宝宝。我们的菌株是丹麦进口的，专门针对儿童肠道和免疫调节。1岁半完全可以吃"),
("customer", "text", "成分安全吗？"),
("sales", "text", "无敏配方，不含牛奶蛋白、麸质和大豆，很多敏宝妈妈都在用"),
("customer", "text", "好的，我先了解一下"),
("sales", "text", "没问题，我给您发个产品详情，您先看看"),
("sales", "link", "[链接：产品详情页]"),
],
},
{
"name": "长期跟进后复购",
"stage": "成交",
"contact": {
"nickname": "糖糖妈妈",
"remark": "糖糖妈-复购-湿疹已好转",
"industry": "宝妈",
"intent_level": "high",
},
"messages": [
("customer", "text", "你好，我想了解一下益生菌"),
("sales", "text", "糖糖妈您好！请问宝宝什么情况？"),
("customer", "text", "3岁，湿疹，反反复复的"),
("sales", "text", "湿疹确实需要从内调理。我们的益生菌丹麦进口菌株，专门针对儿童过敏"),
("customer", "text", "多少钱？"),
("sales", "text", "单盒298，3盒套餐798"),
("customer", "text", "先了解一下，回头再说"),
("sales", "text", "好的，不着急"),
("sales", "text", "糖糖妈，宝宝最近湿疹怎么样了？"),
("customer", "text", "还是老样子，断不了根"),
("sales", "text", "湿疹确实需要坚持调理，益生菌一般吃2-3个月能看到明显改善"),
("customer", "text", "我再考虑考虑"),
("sales", "text", "糖糖妈，今天有个宝妈反馈说吃了2个月湿疹好多了，发您看看"),
("sales", "image", "[图片：宝妈反馈截图]"),
("customer", "text", "真的吗？那我也试试吧"),
("sales", "text", "好的！建议先拿3盒一个周期，798元"),
("customer", "text", "好的，付款吧"),
("sales", "text", "好的！我马上安排发货"),
("sales", "text", "糖糖妈，宝宝吃了这段时间效果怎么样？"),
("customer", "text", "比之前好多了，湿疹没那么频繁了"),
("sales", "text", "太好了！建议继续吃巩固一下，现在有6盒套餐1499更划算"),
("customer", "text", "好，那来6盒"),
("sales", "text", "好的！马上安排"),
],
},
{
"name": "对比竞品后选择",
"stage": "方案匹配",
"contact": {
"nickname": "壮壮妈妈",
"remark": "壮壮妈-对比合生元",
"industry": "宝妈",
"intent_level": "medium",
},
"messages": [
("customer", "text", "你好，我想给孩子买益生菌，但在对比你们和合生元"),
("sales", "text", "壮壮妈您好！对比是正常的。请问宝宝主要什么情况？"),
("customer", "text", "经常感冒，免疫力不太好"),
("sales", "text", "我们的菌株是丹麦进口的鼠李糖乳杆菌GG，有专门针对儿童免疫力的临床验证。合生元用的是国产菌株"),
("customer", "text", "但你们比合生元贵啊"),
("sales", "text", "菌株不一样，进口菌株的临床数据更充分，效果更有保障。算下来每天不到9块钱"),
("customer", "text", "有什么区别吗？"),
("sales", "image", "[图片：菌株对比表]"),
("sales", "text", "主要是菌株来源和临床验证数量不同。我们的LGG是全球研究最多的益生菌菌株之一"),
("customer", "text", "我再看看吧"),
("sales", "text", "好的，有问题随时问我"),
],
},
]
# ---------------------------------------------------------------------------
# 场景变体 — 给不同销售使用
# ---------------------------------------------------------------------------
SCENARIO_VARIANTS = [
{
"name": "食物过敏咨询后成交",
"stage": "成交",
"contact": {
"nickname": "依依妈",
"remark": "依依妈-食物过敏-3岁",
"industry": "宝妈",
"intent_level": "high",
},
"messages": [
("customer", "text", "你好，我家宝宝对牛奶蛋白过敏，听说益生菌能帮助脱敏？"),
("sales", "text", "依依妈您好！是的，益生菌可以帮助调节肠道免疫，对食物过敏有辅助改善作用。请问宝宝现在多大了？"),
("customer", "text", "3岁，查出来对牛奶蛋白和鸡蛋过敏"),
("sales", "text", "理解，食物过敏的宝宝确实需要从肠道调理。我们的菌株是丹麦进口LGG，有临床数据支持对食物过敏的改善"),
("customer", "text", "无敏配方吗？我家对牛奶蛋白过敏"),
("sales", "text", "是的，完全无敏配方，不含牛奶蛋白、麸质和大豆，过敏宝宝可以放心吃"),
("sales", "image", "[图片：成分表]"),
("customer", "text", "看着成分确实干净，多少钱？"),
("sales", "text", "单盒298，3盒套餐798，建议先吃一个周期"),
("customer", "text", "好，来3盒吧"),
("sales", "text", "好的依依妈！3盒798元，给您安排发货"),
("customer", "text", "付款了，麻烦尽快发"),
("sales", "text", "收到！今天就发顺丰"),
],
},
{
"name": "腹泻咨询后成交",
"stage": "成交",
"contact": {
"nickname": "阳阳妈",
"remark": "阳阳妈-腹泻-1岁",
"industry": "宝妈",
"intent_level": "high",
},
"messages": [
("customer", "text", "你好，宝宝1岁，最近老拉肚子，朋友说益生菌有用"),
("sales", "text", "阳阳妈您好！益生菌对宝宝腹泻确实有帮助。请问拉肚子多久了？"),
("customer", "text", "一个多星期了，吃了药好点，停了又拉"),
("sales", "text", "这种情况益生菌可以帮到宝宝调节肠道菌群。我们的丹麦进口菌株专门针对儿童肠道健康"),
("customer", "text", "1岁能吃吗？"),
("sales", "text", "可以的，0岁以上就能用，无敏配方很安全"),
("customer", "text", "多少钱？"),
("sales", "text", "单盒298，3盒套餐798"),
("customer", "text", "先来3盒试试"),
("sales", "text", "好的！3盒798元，马上安排发货"),
("customer", "text", "好的，已付款"),
],
},
{
"name": "价格异议后放弃",
"stage": "异议处理",
"contact": {
"nickname": "可可妈妈",
"remark": "可可妈-价格异议",
"industry": "宝妈",
"intent_level": "low",
},
"messages": [
("customer", "text", "益生菌多少钱？"),
("sales", "text", "可可妈您好！单盒298，3盒套餐798"),
("customer", "text", "这么贵？网上益生菌才几十块"),
("sales", "text", "我们的是丹麦进口菌株，有临床验证，和网上几十块的不一样"),
("customer", "text", "但是也太贵了"),
("sales", "text", "算下来每天不到9块钱，效果有保障"),
("customer", "text", "我再看看吧"),
("sales", "text", "好的，有需要随时找我"),
],
},
{
"name": "安全顾虑后犹豫",
"stage": "异议处理",
"contact": {
"nickname": "贝贝妈妈",
"remark": "贝贝妈-安全顾虑",
"industry": "宝妈",
"intent_level": "medium",
},
"messages": [
("customer", "text", "你好，宝宝6个月，能吃益生菌吗？"),
("sales", "text", "贝贝妈您好！6个月以上的宝宝可以吃的。请问宝宝什么情况？"),
("customer", "text", "老是胀气，睡眠也不好"),
("sales", "text", "益生菌可以帮宝宝调节肠道，改善胀气。我们的丹麦进口菌株0岁以上可用"),
("customer", "text", "这么小的宝宝吃安全吗？有没有副作用？"),
("sales", "text", "很安全的，无敏配方，不含牛奶蛋白和麸质"),
("customer", "text", "我还是有点担心，毕竟宝宝才6个月"),
("sales", "text", "理解您的顾虑，很多宝妈也有同样的担心。其实益生菌是很成熟的品类，我们的菌株有大量临床数据支持"),
("customer", "text", "我再想想吧"),
("sales", "text", "好的，不着急。有问题随时问我"),
],
},
{
"name": "过早报价后被拒",
"stage": "异议处理",
"contact": {
"nickname": "欢欢妈妈",
"remark": "欢欢妈-过早报价",
"industry": "宝妈",
"intent_level": "low",
},
"messages": [
("customer", "text", "益生菌多少钱？"),
("sales", "text", "您好！单盒298，3盒798，6盒1499"),
("customer", "text", "哦，这么贵"),
("sales", "text", "丹麦进口菌株，效果很好的"),
("customer", "text", "我先看看吧"),
("sales", "text", "好的，需要随时联系"),
("customer", "text", "嗯"),
],
},
]
# ---------------------------------------------------------------------------
# 非客户联系人模板
# ---------------------------------------------------------------------------
NON_CUSTOMER_TEMPLATES = [
{
"nickname": "韩国代购-小美",
"remark": "",
"is_group": False,
"messages": [
("other", "text", "【韩国直邮】儿童维生素团购开始啦，DHA+益生菌组合装只要199！"),
("other", "text", "需要的姐妹私我～"),
],
},
{
"nickname": "母婴用品批发",
"remark": "",
"is_group": False,
"messages": [
("other", "text", "姐，我们新款益生菌做活动，比你现在用的便宜一半，要不要看看？"),
("other", "text", "保证正品，支持验货"),
("other", "emoji", "[表情：微笑]"),
],
},
{
"nickname": "朵朵阿姨",
"remark": "",
"is_group": False,
"messages": [
("other", "text", "在吗？"),
("other", "text", "你朋友圈那个是什么产品？"),
("other", "text", "好的知道了"),
],
},
{
"nickname": "老婆 ❤️",
"remark": "",
"is_group": False,
"messages": [
("other", "text", "今晚回来吃饭吗？"),
("other", "text", "孩子放学了"),
("other", "text", "周末去哪玩？"),
("other", "text", "好的"),
],
},
{
"nickname": "快递小哥",
"remark": "",
"is_group": False,
"messages": [
("other", "text", "你的快递到了，放门口了"),
("other", "text", "签收一下"),
],
},
{
"nickname": "大学同学",
"remark": "",
"is_group": False,
"messages": [
("other", "text", "好久不见啊"),
("other", "text", "最近怎么样"),
("other", "text", "改天聚聚"),
],
},
{
"nickname": "同行-小张",
"remark": "",
"is_group": False,
"messages": [
("other", "text", "张哥，最近怎么样？"),
("other", "text", "你们那边转化率怎么样？"),
("other", "text", "交流一下经验"),
],
},
{
"nickname": "健身教练",
"remark": "",
"is_group": False,
"messages": [
("other", "text", "明天还来训练吗？"),
("other", "text", "记得带毛巾"),
],
},
{
"nickname": "房产中介小王",
"remark": "",
"is_group": False,
"messages": [
("other", "text", "哥，最近有考虑换房吗？"),
("other", "text", "新盘有优惠"),
],
},
{
"nickname": "表妹",
"remark": "",
"is_group": False,
"messages": [
("other", "text", "哥，你那边益生菌我朋友想了解一下"),
("other", "text", "我推她微信给你"),
("other", "text", "好的"),
],
},
]
# ---------------------------------------------------------------------------
# 群聊模板
# ---------------------------------------------------------------------------
GROUP_TEMPLATES = [
{
"nickname": "宝妈育儿交流群",
"is_group": True,
"messages": [
("other", "text", "[群公告] 本群禁发广告，违者移出"),
("other", "text", "有没有宝妈推荐好的儿童面霜？"),
("other", "text", "我家用的那个丝塔芙还不错"),
("other", "text", "谢谢推荐"),
("other", "text", "有没有宝妈推荐好的儿童霜？"),
("other", "text", "大家宝宝都几岁上幼儿园的？"),
("other", "text", "我家3岁送的"),
("other", "text", "这么早吗？会不会太小了"),
("other", "text", "还好，适应期过了就好了"),
],
},
{
"nickname": "过敏宝宝互助群",
"is_group": True,
"messages": [
("other", "text", "我家宝宝湿疹终于好了"),
("other", "text", "怎么好的？用的什么？"),
("other", "text", "吃了益生菌加上注意饮食"),
("other", "text", "你们都用的什么益生菌？"),
("other", "text", "益生菌真的有用吗？"),
("other", "text", "我觉得有用的，坚持吃了2个月"),
("other", "text", "我也想试试"),
("other", "text", "可以试试，从肠道调理确实有道理"),
("other", "text", "有没有副作用？"),
("other", "text", "我们吃的没什么副作用"),
("other", "text", "好的，谢谢分享"),
],
},
{
"nickname": "育儿干货分享群",
"is_group": True,
"messages": [
("other", "text", "今天分享一篇关于儿童免疫力的文章"),
("other", "link", "[链接：儿童免疫力科普]"),
("other", "text", "写得好，收藏了"),
("other", "text", "谢谢分享"),
("other", "text", "有没有关于过敏的科普？"),
("other", "text", "有的，我找找"),
],
},
]
# ---------------------------------------------------------------------------
# 每个销售的联系人分配方案
# ---------------------------------------------------------------------------
# 格式: (scenario_index_or_variant_name, contact_nickname, contact_remark)
# scenario_index 指向 SCENARIO_TEMPLATES, variant_name 指向 SCENARIO_VARIANTS
SALESPERSON_ASSIGNMENTS = {
1: [ # 张伟 — 资深
("scenario_0", None, None), # 湿疹成交
("scenario_2", None, None), # 朋友推荐
("variant_食物过敏咨询后成交", None, None),
("variant_腹泻咨询后成交", None, None),
("scenario_3", None, None), # 长期跟进复购
# 潜在客户
("scenario_2", "小宇妈", "小宇妈-咨询"),
("scenario_2", "安安妈妈", "安安妈-咨询"),
# 非客户
("non_customer", "韩国代购-小美", ""),
("non_customer", "母婴用品批发", ""),
("non_customer", "朵朵阿姨", ""),
("non_customer", "老婆 ❤️", ""),
# 群聊
("group", "宝妈育儿交流群", ""),
("group", "过敏宝宝互助群", ""),
],
2: [ # 李娜 — 新人
("scenario_1", None, None), # 鼻炎犹豫
("variant_价格异议后放弃", None, None),
("scenario_4", None, None), # 竞品对比
("variant_过早报价后被拒", None, None),
("scenario_2", "甜甜妈", "甜甜妈-推荐咨询"),
# 潜在客户
("scenario_2", "球球妈妈", "球球妈-咨询"),
("scenario_2", "米米妈", "米米妈-咨询"),
# 非客户
("non_customer", "同行-小张", ""),
("non_customer", "快递小哥", ""),
("non_customer", "大学同学", ""),
# 群聊
("group", "宝妈育儿交流群", ""),
("group", "育儿干货分享群", ""),
],
3: [ # 王强 — 中等
("scenario_4", None, None), # 竞品对比
("scenario_3", None, None), # 长期跟进复购
("variant_腹泻咨询后成交", None, None),
("variant_安全顾虑后犹豫", None, None),
("scenario_2", "星星妈", "星星妈-推荐咨询"),
# 潜在客户
("scenario_2", "晨晨妈妈", "晨晨妈-咨询"),
("scenario_2", "露露妈", "露露妈-咨询"),
# 非客户
("non_customer", "健身教练", ""),
("non_customer", "房产中介小王", ""),
("non_customer", "表妹", ""),
# 群聊
("group", "过敏宝宝互助群", ""),
("group", "宝妈育儿交流群", ""),
],
}
# ---------------------------------------------------------------------------
# 生成与同步逻辑
# ---------------------------------------------------------------------------
def get_variant(name: str):
for v in SCENARIO_VARIANTS:
if v["name"] == name:
return v
raise KeyError(f"variant not found: {name}")
def generate_contacts_for_salesperson(sp_id: int):
"""返回 [(nickname, remark, is_group, wx_username, scenario_or_template)]"""
assignments = SALESPERSON_ASSIGNMENTS.get(sp_id, [])
result = []
for i, (kind, nick_override, remark_override) in enumerate(assignments):
if kind.startswith("scenario_"):
idx = int(kind.split("_")[1])
tpl = SCENARIO_TEMPLATES[idx]
nick = nick_override or tpl["contact"]["nickname"]
remark = remark_override or tpl["contact"].get("remark", "")
is_group = False
wx_username = f"{SALESPERSONS[sp_id]['wx_account']}_contact_{i+1}"
result.append((nick, remark, is_group, wx_username, tpl))
elif kind.startswith("variant_"):
tpl = get_variant(kind[8:])
nick = nick_override or tpl["contact"]["nickname"]
remark = remark_override or tpl["contact"].get("remark", "")
is_group = False
wx_username = f"{SALESPERSONS[sp_id]['wx_account']}_contact_{i+1}"
result.append((nick, remark, is_group, wx_username, tpl))
elif kind == "non_customer":
nick = nick_override
remark = remark_override or ""
# 找到对应模板
tpl = None
for t in NON_CUSTOMER_TEMPLATES:
if t["nickname"] == nick:
tpl = t
break
if tpl is None:
tpl = NON_CUSTOMER_TEMPLATES[i % len(NON_CUSTOMER_TEMPLATES)]
nick = tpl["nickname"]
is_group = False
wx_username = f"{SALESPERSONS[sp_id]['wx_account']}_contact_{i+1}"
result.append((nick, remark, is_group, wx_username, tpl))
elif kind == "group":
nick = nick_override
remark = remark_override or ""
tpl = None
for t in GROUP_TEMPLATES:
if t["nickname"] == nick:
tpl = t
break
if tpl is None:
tpl = GROUP_TEMPLATES[i % len(GROUP_TEMPLATES)]
nick = tpl["nickname"]
is_group = True
wx_username = f"{SALESPERSONS[sp_id]['wx_account']}_group_{i+1}@chatroom"
result.append((nick, remark, is_group, wx_username, tpl))
return result
def sync_to_central(pg, sp_id: int, contacts_data, run_counter: int = 1):
"""将联系人、会话、消息写入中央数据库。"""
sp_info = SALESPERSONS[sp_id]
source_shard = f"mock_{sp_id}"
base_time = dt.datetime(2026, 6, 15, 10, 0, 0, tzinfo=TZ)
# 每次运行追加的时间偏移
time_offset = dt.timedelta(days=run_counter * 7)
contact_ids = {}
conversation_ids = {}
with pg.cursor() as cur:
# 写入联系人
for nick, remark, is_group, wx_username, tpl in contacts_data:
display_name = remark or nick
cur.execute("""
INSERT INTO contact (salesperson_id, wx_username, nickname, remark, display_name, is_group)
VALUES (%s, %s, %s, %s, %s, %s)
ON CONFLICT (salesperson_id, wx_username) DO UPDATE SET
nickname=excluded.nickname, remark=excluded.remark,
display_name=excluded.display_name, is_group=excluded.is_group
RETURNING id
""", (sp_id, wx_username, nick, remark, display_name, is_group))
contact_ids[wx_username] = cur.fetchone()[0]
# 写入会话
for nick, remark, is_group, wx_username, tpl in contacts_data:
conv_type = "group" if is_group else "single"
cur.execute("""
INSERT INTO conversation (salesperson_id, contact_id, wx_identifier, conv_type, last_synced_at)
VALUES (%s, %s, %s, %s, now())
ON CONFLICT (salesperson_id, wx_identifier) DO UPDATE SET
contact_id=excluded.contact_id, last_synced_at=now()
RETURNING id
""", (sp_id, contact_ids[wx_username], wx_username, conv_type))
conversation_ids[wx_username] = cur.fetchone()[0]
# 写入消息
local_id_counter = (run_counter - 1) * 10000 # 每次运行从不同区间开始
msg_count = 0
batch = []
for nick, remark, is_group, wx_username, tpl in contacts_data:
conv_id = conversation_ids[wx_username]
contact_id = contact_ids[wx_username]
messages = tpl["messages"]
# 确定发送者
sp_wx = sp_info["wx_account"]
sp_name = sp_info["name"]
for j, (sender_role, msg_type, content) in enumerate(messages):
local_id_counter += 1
# 时间递增：每条消息间隔 2～30 分钟
msg_time = base_time + time_offset + dt.timedelta(minutes=j * random.randint(2, 30))
if sender_role == "customer":
sender_wx = wx_username if not is_group else f"wxid_customer_{j}"
sender_name = nick
elif sender_role == "sales":
sender_wx = sp_wx
sender_name = sp_name
else: # other
sender_wx = f"wxid_other_{j}" if is_group else wx_username
sender_name = nick
source_table = "Msg_" + hashlib.md5(wx_username.encode()).hexdigest()[:16]
batch.append((
sp_id, conv_id, sender_wx, sender_name, msg_type,
content, content, msg_time,
source_shard, source_table, local_id_counter,
))
msg_count += 1
if len(batch) >= 2000:
execute_values(cur, """
INSERT INTO message (salesperson_id, conversation_id, sender_wx_username,
sender_display_name, message_type, raw_content, normalized_content,
created_at, source_shard, source_table, source_local_id)
VALUES %s ON CONFLICT (source_shard, source_table, source_local_id) DO NOTHING
""", batch, page_size=2000)
pg.commit()
batch = []
if batch:
execute_values(cur, """
INSERT INTO message (salesperson_id, conversation_id, sender_wx_username,
sender_display_name, message_type, raw_content, normalized_content,
created_at, source_shard, source_table, source_local_id)
VALUES %s ON CONFLICT (source_shard, source_table, source_local_id) DO NOTHING
""", batch, page_size=2000)
pg.commit()
return len(contact_ids), msg_count
def get_run_counter(pg, sp_id: int) -> int:
"""根据已有消息数推断运行次数。"""
with pg.cursor() as cur:
cur.execute("""
SELECT count(DISTINCT source_local_id / 10000) FROM message
WHERE salesperson_id = %s AND source_shard = %s
""", (sp_id, f"mock_{sp_id}"))
result = cur.fetchone()[0]
return int(result or 0) + 1
def main():
ap = argparse.ArgumentParser(description="模拟采集代理")
ap.add_argument("--salesperson-id", type=int, required=True)
ap.add_argument("--dsn", default=DEFAULT_DSN)
args = ap.parse_args()
sp_id = args.salesperson_id
if sp_id not in SALESPERSONS:
print(f"[ERROR] 未知销售 ID: {sp_id}", file=sys.stderr)
return 1
sp_info = SALESPERSONS[sp_id]
print(f"[INFO] 模拟采集代理启动: {sp_info['name']} (id={sp_id})")
pg = psycopg2.connect(args.dsn)
pg.autocommit = False
try:
run_counter = get_run_counter(pg, sp_id)
print(f"[INFO] 第 {run_counter} 次同步")
contacts_data = generate_contacts_for_salesperson(sp_id)
print(f"[INFO] 生成 {len(contacts_data)} 个联系人")
contact_count, msg_count = sync_to_central(pg, sp_id, contacts_data, run_counter)
with pg.cursor() as cur:
cur.execute("SELECT count(*) FROM message WHERE salesperson_id=%s", (sp_id,))
total_msgs = cur.fetchone()[0]
print(json.dumps({
"salesperson_id": sp_id,
"salesperson_name": sp_info["name"],
"run": run_counter,
"contacts": contact_count,
"new_messages": msg_count,
"total_messages": total_msgs,
}, ensure_ascii=False, indent=2))
return 0
except Exception as exc:
pg.rollback()
print(f"[ERROR] {exc}", file=sys.stderr)
return 1
finally:
pg.close()
if __name__ == "__main__":
sys.exit(main())
Reference in New Issue View Git Blame Copy Permalink