376 lines
13 KiB
Python
376 lines
13 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
召回完成检测器（Recall Completion Detector）
|
||
|
||
ETL 数据更新后，扫描所有 MAIN 关系对的结算记录，
|
||
记录广义召回事件（recall_events），匹配活跃任务标记 completed，
|
||
对所有到店的 MAIN 关联客户生成回访任务（follow_up_visit）。
|
||
|
||
由 trigger_jobs 中的 recall_completion_check 配置驱动（event: etl_data_updated）。
|
||
|
||
CHANGE 2026-04-08 | Fix-13 改造：
|
||
- 扫描范围从"有 active 任务的客户"扩大为"所有 os_label='MAIN' 的关联客户"
|
||
- 新增 recall_events 事件表记录广义召回（按天去重）
|
||
- 无 active 任务的客户到店也生成 follow_up_visit
|
||
"""
|
||
|
||
import json
|
||
import logging
|
||
from datetime import timedelta
|
||
|
||
from app.trace.decorators import trace_service
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
def _get_connection():
|
||
"""延迟导入 get_connection，避免模块级导入失败。"""
|
||
from app.database import get_connection
|
||
|
||
return get_connection()
|
||
|
||
|
||
def _insert_history(
|
||
cur,
|
||
task_id: int,
|
||
action: str,
|
||
old_status: str | None = None,
|
||
new_status: str | None = None,
|
||
old_task_type: str | None = None,
|
||
new_task_type: str | None = None,
|
||
detail: dict | None = None,
|
||
) -> None:
|
||
"""在 coach_task_history 中记录变更。"""
|
||
cur.execute(
|
||
"""
|
||
INSERT INTO biz.coach_task_history
|
||
(task_id, action, old_status, new_status,
|
||
old_task_type, new_task_type, detail)
|
||
VALUES (%s, %s, %s, %s, %s, %s, %s)
|
||
""",
|
||
(
|
||
task_id,
|
||
action,
|
||
old_status,
|
||
new_status,
|
||
old_task_type,
|
||
new_task_type,
|
||
json.dumps(detail) if detail else None,
|
||
),
|
||
)
|
||
|
||
|
||
@trace_service(description_zh="执行维客检测", description_en="Run recall detection")
|
||
def run(payload: dict | None = None, job_id: int | None = None) -> dict:
|
||
"""
|
||
召回完成检测主流程。
|
||
|
||
CHANGE 2026-04-08 | Fix-13 改造：扫描所有 MAIN 关系对。
|
||
1. 从 biz.sites 获取所有活跃门店
|
||
2. 对每个 site_id，通过 _fdw_context 扫描 MAIN 关系对的结算记录
|
||
3. 有结算 → 写 recall_events + 完成任务（如有）+ 生成回访
|
||
"""
|
||
completed_count = 0
|
||
event_count = 0
|
||
|
||
conn = _get_connection()
|
||
try:
|
||
# ── 1. 从业务库获取所有活跃门店 ──
|
||
with conn.cursor() as cur:
|
||
cur.execute(
|
||
"SELECT site_id FROM biz.sites WHERE is_active = true"
|
||
)
|
||
site_ids = [r[0] for r in cur.fetchall()]
|
||
conn.commit()
|
||
|
||
# ── 2. 逐 site_id 处理 ──
|
||
for site_id in site_ids:
|
||
try:
|
||
result = _process_site(conn, site_id)
|
||
completed_count += result["completed"]
|
||
event_count += result["events"]
|
||
except Exception:
|
||
logger.exception(
|
||
"处理门店召回检测失败: site_id=%s", site_id
|
||
)
|
||
conn.rollback()
|
||
|
||
# ── 更新 last_run_at（兼容 trigger_scheduler 调度记录） ──
|
||
if job_id is not None:
|
||
from app.services.trigger_scheduler import update_job_last_run_at
|
||
|
||
with conn.cursor() as cur:
|
||
cur.execute("BEGIN")
|
||
update_job_last_run_at(cur, job_id)
|
||
conn.commit()
|
||
|
||
finally:
|
||
conn.close()
|
||
|
||
logger.info(
|
||
"召回完成检测完成: completed_count=%d, event_count=%d",
|
||
completed_count, event_count,
|
||
)
|
||
return {"completed_count": completed_count, "event_count": event_count}
|
||
|
||
|
||
def _process_site(conn, site_id: int) -> dict:
|
||
"""
|
||
处理单个门店的广义召回检测。
|
||
|
||
CHANGE 2026-04-08 | Fix-13 改造：
|
||
1. 从 ETL 查询所有 os_label='MAIN' 的 (assistant_id, member_id) 对
|
||
2. 批量查询这些客户的最新结算记录
|
||
3. 对每个有新结算的关系对：写 recall_events + 完成任务 + 生成回访
|
||
"""
|
||
completed = 0
|
||
events = 0
|
||
|
||
from app.services.fdw_queries import _fdw_context
|
||
|
||
# ── 1. 获取本门店所有 MAIN 关系对 ──
|
||
with _fdw_context(conn, site_id) as cur:
|
||
cur.execute(
|
||
"""
|
||
SELECT assistant_id, member_id
|
||
FROM app.v_dws_member_assistant_relation_index
|
||
WHERE os_label = 'MAIN'
|
||
"""
|
||
)
|
||
main_pairs = [(r[0], r[1]) for r in cur.fetchall()]
|
||
|
||
if not main_pairs:
|
||
return {"completed": 0, "events": 0}
|
||
|
||
# ── 2. 批量查询这些客户的最新结算时间 ──
|
||
member_ids = list({mid for _, mid in main_pairs})
|
||
settlement_map: dict[tuple[int, int], object] = {} # (assistant_id, member_id) → latest_pay_time
|
||
|
||
with _fdw_context(conn, site_id) as cur:
|
||
cur.execute(
|
||
"""
|
||
SELECT sl.site_assistant_id AS assistant_id,
|
||
sh.member_id,
|
||
MAX(sh.pay_time) AS latest_pay_time
|
||
FROM app.v_dwd_settlement_head sh
|
||
JOIN app.v_dwd_assistant_service_log sl
|
||
ON sl.order_settle_id = sh.order_settle_id
|
||
AND sl.is_delete = 0
|
||
WHERE sh.member_id = ANY(%s)
|
||
AND sh.settle_type IN (1, 3)
|
||
GROUP BY sl.site_assistant_id, sh.member_id
|
||
""",
|
||
(member_ids,),
|
||
)
|
||
for row in cur.fetchall():
|
||
settlement_map[(row[0], row[1])] = row[2]
|
||
|
||
# ── 3. 获取本门店所有 active 的召回/回访任务（用于匹配） ──
|
||
active_tasks_map: dict[tuple[int, int], list] = {} # (assistant_id, member_id) → [(id, task_type, created_at)]
|
||
with conn.cursor() as cur:
|
||
cur.execute(
|
||
"""
|
||
SELECT id, assistant_id, member_id, task_type, created_at
|
||
FROM biz.coach_tasks
|
||
WHERE site_id = %s
|
||
AND status = 'active'
|
||
AND task_type IN ('high_priority_recall', 'priority_recall', 'follow_up_visit')
|
||
""",
|
||
(site_id,),
|
||
)
|
||
for row in cur.fetchall():
|
||
key = (row[1], row[2])
|
||
active_tasks_map.setdefault(key, []).append(
|
||
{"id": row[0], "task_type": row[3], "created_at": row[4]}
|
||
)
|
||
conn.commit()
|
||
|
||
# ── 4. 逐关系对处理 ──
|
||
for assistant_id, member_id in main_pairs:
|
||
latest_pay = settlement_map.get((assistant_id, member_id))
|
||
if latest_pay is None:
|
||
continue
|
||
|
||
active_tasks = active_tasks_map.get((assistant_id, member_id), [])
|
||
|
||
try:
|
||
result = _process_pair(
|
||
conn, site_id, assistant_id, member_id,
|
||
latest_pay, active_tasks,
|
||
)
|
||
completed += result["completed"]
|
||
events += result["events"]
|
||
except Exception:
|
||
logger.exception(
|
||
"处理关系对失败: site_id=%s, assistant_id=%s, member_id=%s",
|
||
site_id, assistant_id, member_id,
|
||
)
|
||
conn.rollback()
|
||
|
||
return {"completed": completed, "events": events}
|
||
|
||
|
||
def _process_pair(
|
||
conn,
|
||
site_id: int,
|
||
assistant_id: int,
|
||
member_id: int,
|
||
latest_pay_time,
|
||
active_tasks: list[dict],
|
||
) -> dict:
|
||
"""
|
||
处理单个 MAIN 关系对的召回检测。
|
||
|
||
CHANGE 2026-04-08 | Fix-13 改造：
|
||
- 写 recall_events（ON CONFLICT DO NOTHING 按天去重）
|
||
- 有 active 召回任务且 pay_time > created_at → 完成任务
|
||
- 关闭旧回访 → 新建回访（48h）
|
||
- 无 active 任务也生成回访
|
||
|
||
返回: {"completed": int, "events": int}
|
||
"""
|
||
completed = 0
|
||
events = 0
|
||
|
||
with conn.cursor() as cur:
|
||
cur.execute("BEGIN")
|
||
|
||
# ── 1. 写 recall_events（按天去重） ──
|
||
# 先查是否有匹配的召回任务（用于填充 task_id/task_type）
|
||
recall_tasks = [
|
||
t for t in active_tasks
|
||
if t["task_type"] in ("high_priority_recall", "priority_recall")
|
||
and latest_pay_time > t["created_at"]
|
||
]
|
||
event_task_id = recall_tasks[0]["id"] if recall_tasks else None
|
||
event_task_type = recall_tasks[0]["task_type"] if recall_tasks else None
|
||
|
||
cur.execute(
|
||
"""
|
||
INSERT INTO biz.recall_events
|
||
(site_id, assistant_id, member_id, pay_time, task_id, task_type)
|
||
VALUES (%s, %s, %s, %s, %s, %s)
|
||
ON CONFLICT (site_id, assistant_id, member_id, (date_trunc('day', pay_time AT TIME ZONE 'Asia/Shanghai')))
|
||
DO NOTHING
|
||
RETURNING id
|
||
""",
|
||
(site_id, assistant_id, member_id, latest_pay_time,
|
||
event_task_id, event_task_type),
|
||
)
|
||
inserted = cur.fetchone()
|
||
if inserted is None:
|
||
# 今天已记录过，跳过后续处理（避免重复生成回访）
|
||
conn.commit()
|
||
return {"completed": 0, "events": 0}
|
||
|
||
events = 1
|
||
|
||
# ── 2. 完成匹配的召回任务 ──
|
||
has_active_recall = len(recall_tasks) > 0
|
||
|
||
for task in recall_tasks:
|
||
cur.execute(
|
||
"""
|
||
UPDATE biz.coach_tasks
|
||
SET status = 'completed',
|
||
completed_at = %s,
|
||
completed_task_type = %s,
|
||
completion_type = 'auto',
|
||
updated_at = NOW()
|
||
WHERE id = %s AND status = 'active'
|
||
""",
|
||
(latest_pay_time, task["task_type"], task["id"]),
|
||
)
|
||
_insert_history(
|
||
cur,
|
||
task["id"],
|
||
action="completed",
|
||
old_status="active",
|
||
new_status="completed",
|
||
old_task_type=task["task_type"],
|
||
new_task_type=task["task_type"],
|
||
detail={
|
||
"service_time": str(latest_pay_time),
|
||
"completed_task_type": task["task_type"],
|
||
},
|
||
)
|
||
completed += 1
|
||
|
||
# ── 3. 关闭已有的 active 回访任务 ──
|
||
cur.execute(
|
||
"""
|
||
SELECT id FROM biz.coach_tasks
|
||
WHERE site_id = %s AND assistant_id = %s AND member_id = %s
|
||
AND task_type = 'follow_up_visit' AND status = 'active'
|
||
""",
|
||
(site_id, assistant_id, member_id),
|
||
)
|
||
old_follow_ups = cur.fetchall()
|
||
for (old_id,) in old_follow_ups:
|
||
cur.execute(
|
||
"""
|
||
UPDATE biz.coach_tasks
|
||
SET status = 'inactive', updated_at = NOW()
|
||
WHERE id = %s
|
||
""",
|
||
(old_id,),
|
||
)
|
||
_insert_history(
|
||
cur, old_id,
|
||
action="superseded_by_new_visit",
|
||
old_status="active", new_status="inactive",
|
||
old_task_type="follow_up_visit", new_task_type="follow_up_visit",
|
||
detail={"reason": "new_service_record", "service_time": str(latest_pay_time)},
|
||
)
|
||
|
||
# ── 4. 创建新的回访任务（48h 过期） ──
|
||
expires_at = (
|
||
latest_pay_time + timedelta(hours=48)
|
||
if hasattr(latest_pay_time, '__add__') else None
|
||
)
|
||
cur.execute(
|
||
"""
|
||
INSERT INTO biz.coach_tasks
|
||
(site_id, assistant_id, member_id, task_type, status,
|
||
expires_at, created_at, updated_at)
|
||
VALUES (%s, %s, %s, 'follow_up_visit', 'active', %s, NOW(), NOW())
|
||
RETURNING id
|
||
""",
|
||
(site_id, assistant_id, member_id, expires_at),
|
||
)
|
||
new_follow_up_id = cur.fetchone()[0]
|
||
_insert_history(
|
||
cur, new_follow_up_id,
|
||
action="created",
|
||
old_status=None, new_status="active",
|
||
new_task_type="follow_up_visit",
|
||
detail={
|
||
"reason": "service_record_detected",
|
||
"service_time": str(latest_pay_time),
|
||
"had_recall": has_active_recall,
|
||
},
|
||
)
|
||
|
||
conn.commit()
|
||
|
||
# ── 5. 触发 recall_completed 事件（仅当有召回任务被完成时） ──
|
||
if has_active_recall:
|
||
try:
|
||
from app.services.trigger_scheduler import fire_event
|
||
fire_event(
|
||
"recall_completed",
|
||
{
|
||
"site_id": site_id,
|
||
"assistant_id": assistant_id,
|
||
"member_id": member_id,
|
||
"service_time": str(latest_pay_time),
|
||
},
|
||
)
|
||
except Exception:
|
||
logger.exception(
|
||
"触发 recall_completed 事件失败: site_id=%s, assistant_id=%s, member_id=%s",
|
||
site_id, assistant_id, member_id,
|
||
)
|
||
|
||
return {"completed": completed, "events": events}
|