""" 通知调度服务 告警创建后的异步处理流水线： 1. VLM 复核截图 → 写入 alarm_llm_analysis 2. 查表获取通知人员 (camera → area → person) 3. 推送企微通知： - 群聊：image + news + @text 组合消息 - 个人：button_interaction 模板卡片 全程异步执行，不阻塞告警接收主流程。 VLM 或企微不可用时自动降级，不影响系统运行。 """ from datetime import datetime from typing import Dict, List from app.models import ( get_session, AlarmEvent, AlarmLlmAnalysis, CameraAreaBinding, AreaPersonBinding, NotifyArea, ) from app.config import settings from app.services.vlm_service import get_vlm_service from app.services.wechat_service import get_wechat_service from app.utils.logger import logger from app.utils.timezone import beijing_now async def process_alarm_notification(alarm_data: Dict): """ 告警通知处理流水线（异步，fire-and-forget） Args: alarm_data: 告警字典，包含 alarm_id, alarm_type, device_id, snapshot_url, alarm_level, event_time 等字段 """ alarm_id = alarm_data.get("alarm_id", "") alarm_type = alarm_data.get("alarm_type", "") device_id = alarm_data.get("device_id", "") snapshot_url = alarm_data.get("snapshot_url", "") alarm_level = alarm_data.get("alarm_level", 2) event_time = alarm_data.get("event_time", "") logger.info(f"开始处理告警通知: {alarm_id}") try: # ========== 1. VLM 复核 ========== vlm_service = get_vlm_service() camera_name = alarm_data.get("camera_name", device_id) scene_id = alarm_data.get("scene_id", "") # 查找区域名称用于 VLM prompt（比 UUID 更有意义） area_name_for_vlm, _ = _get_notify_persons(device_id, alarm_level) roi_name = area_name_for_vlm if area_name_for_vlm != "未知区域" else scene_id # snapshot_url 可能是 COS object key，需转为可访问的预签名URL presigned_snapshot_url = _get_presigned_url(snapshot_url) vlm_result = await vlm_service.verify_alarm( snapshot_url=presigned_snapshot_url, alarm_type=alarm_type, camera_name=camera_name, roi_name=roi_name, ) # 写入 alarm_llm_analysis 表 _save_vlm_result(alarm_id, vlm_result) # VLM 明确判定为误报（非降级跳过）→ 更新告警状态，不通知 if not vlm_result.get("confirmed", True) and not vlm_result.get("skipped"): _mark_false_alarm(alarm_id) logger.info(f"VLM 判定误报，跳过通知: {alarm_id}") return # VLM 跳过（降级）且 confirmed=False → 不改告警状态，仅跳过通知 if not vlm_result.get("confirmed", True) and vlm_result.get("skipped"): logger.info(f"VLM 降级跳过，不推送但保留告警状态: {alarm_id}") return # ========== 2. 查表获取通知人员 ========== description = vlm_result.get("description", "") area_name = area_name_for_vlm _, persons = _get_notify_persons(device_id, alarm_level) # 演示模式：数据库无人员时，使用配置的测试 userid if not persons and settings.wechat.test_uids: test_uids = [uid.strip() for uid in settings.wechat.test_uids.split(",") if uid.strip()] if test_uids: persons = [{"person_name": "测试用户", "wechat_uid": uid, "role": "TEST"} for uid in test_uids] area_name = "演示区域" logger.info(f"演示模式: 使用测试用户 {test_uids}") if not persons: logger.warning(f"未找到通知人员: camera={device_id}, 跳过企微推送") return # ========== 3. 推送企微通知 ========== wechat_service = get_wechat_service() if not wechat_service.enabled: logger.info(f"企微未启用，跳过通知: {alarm_id}") return event_time_str = ( event_time.strftime("%Y-%m-%d %H:%M:%S") if isinstance(event_time, datetime) else str(event_time or "") ) user_ids = [p["wechat_uid"] for p in persons] group_chat_id = settings.wechat.group_chat_id # ---- 3a. 群聊组合消息 ---- if group_chat_id: await wechat_service.send_group_alarm_combo( chat_id=group_chat_id, alarm_id=alarm_id, alarm_type=alarm_type, area_name=area_name, camera_name=camera_name, description=description or f"{area_name} 检测到告警", event_time=event_time_str, alarm_level=alarm_level, snapshot_url=presigned_snapshot_url, mention_user_ids=user_ids, ) else: logger.debug("未配置群聊ID，跳过群聊推送") # ---- 3b. 个人按钮交互卡片 ---- sent = await wechat_service.send_alarm_card( user_ids=user_ids, alarm_id=alarm_id, alarm_type=alarm_type, area_name=area_name, camera_name=camera_name, description=description, event_time=event_time_str, alarm_level=alarm_level, ) if sent: logger.info(f"告警通知完成: {alarm_id}") else: logger.warning(f"个人卡片发送失败: {alarm_id}") except Exception as e: logger.error(f"告警通知处理失败: {alarm_id}, error={e}", exc_info=True) def _get_presigned_url(snapshot_url: str) -> str: """将 COS object key 转为预签名 URL""" if not snapshot_url: return "" if snapshot_url.startswith("http"): return snapshot_url try: from app.services.oss_storage import get_oss_storage return get_oss_storage().get_presigned_url(snapshot_url) except Exception as e: logger.warning(f"生成预签名URL失败: {e}") return snapshot_url def _save_vlm_result(alarm_id: str, vlm_result: Dict): """将 VLM 复核结果写入 alarm_llm_analysis 表""" db = get_session() try: analysis = AlarmLlmAnalysis( alarm_id=alarm_id, llm_model="qwen3-vl-flash", analysis_type="REVIEW", summary=vlm_result.get("description", ""), is_false_alarm=not vlm_result.get("confirmed", True), confidence_score=None if vlm_result.get("skipped") else 0.9, suggestion="VLM跳过" if vlm_result.get("skipped") else None, ) db.add(analysis) db.commit() except Exception as e: db.rollback() logger.error(f"保存 VLM 结果失败: {e}") finally: db.close() def _mark_false_alarm(alarm_id: str): """将告警标记为误报""" db = get_session() try: alarm = db.query(AlarmEvent).filter(AlarmEvent.alarm_id == alarm_id).first() if alarm: now = beijing_now() alarm.alarm_status = "FALSE" alarm.handle_status = "IGNORED" alarm.handle_remark = "VLM复核判定误报" alarm.handled_at = now if alarm.duration_ms is None and alarm.event_time: try: delta = now - alarm.event_time alarm.duration_ms = int(delta.total_seconds() * 1000) except Exception: pass if alarm.last_frame_time is None: alarm.last_frame_time = now db.commit() logger.info(f"告警已标记误报: {alarm_id}") except Exception as e: db.rollback() logger.error(f"标记误报失败: {e}") finally: db.close() def _get_notify_persons(camera_id: str, alarm_level: int) -> tuple: """ 根据摄像头ID查找通知人员 Returns: (area_name, [{"person_name": ..., "wechat_uid": ..., "role": ...}]) """ db = get_session() try: binding = db.query(CameraAreaBinding).filter( CameraAreaBinding.camera_id == camera_id ).first() if not binding: return ("未知区域", []) area = db.query(NotifyArea).filter( NotifyArea.area_id == binding.area_id, NotifyArea.enabled == 1, ).first() area_name = area.area_name if area else "未知区域" persons = db.query(AreaPersonBinding).filter( AreaPersonBinding.area_id == binding.area_id, AreaPersonBinding.enabled == 1, AreaPersonBinding.notify_level <= alarm_level, ).all() result = [ { "person_name": p.person_name, "wechat_uid": p.wechat_uid, "role": p.role, } for p in persons ] return (area_name, result) except Exception as e: logger.error(f"查询通知人员失败: {e}") return ("未知区域", []) finally: db.close()