63 lines
2.1 KiB
Python
63 lines
2.1 KiB
Python
"""fin_risk_alert 仓储：按状态查预警 + 处置条件更新（防并发，不 commit）。"""
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
|
|
from sqlalchemy import select, update
|
|
|
|
from model.fin_risk_alert import FinRiskAlert
|
|
from repositories.base import BaseRepository
|
|
|
|
|
|
class FinRiskAlertRepo(BaseRepository):
|
|
model = FinRiskAlert
|
|
|
|
async def list_by_status(
|
|
self, status: str | None = None, customer_id: int | None = None
|
|
) -> list[FinRiskAlert]:
|
|
stmt = select(FinRiskAlert)
|
|
if status is not None:
|
|
stmt = stmt.where(FinRiskAlert.status == status)
|
|
if customer_id is not None:
|
|
stmt = stmt.where(FinRiskAlert.customer_id == customer_id)
|
|
stmt = stmt.order_by(FinRiskAlert.id.desc())
|
|
return list((await self.db.scalars(stmt)).all())
|
|
|
|
async def conditional_handle(
|
|
self,
|
|
alert_id: int,
|
|
*,
|
|
handler_id: int,
|
|
new_status: str,
|
|
handle_result: str,
|
|
handle_time: datetime,
|
|
) -> bool:
|
|
"""处置预警：条件更新 WHERE id=? AND status='未处理'。
|
|
|
|
不 commit，由 service 层事务统一提交。rowcount==0 表示已被他人处置，
|
|
返回 False 供上层提示「已被处理」，避免重复落账/工单/通知。
|
|
"""
|
|
result = await self.db.execute(
|
|
update(FinRiskAlert)
|
|
.where(
|
|
FinRiskAlert.id == alert_id,
|
|
FinRiskAlert.status == "未处理",
|
|
)
|
|
.values(
|
|
status=new_status,
|
|
handler_id=handler_id,
|
|
handle_result=handle_result,
|
|
handle_time=handle_time,
|
|
)
|
|
)
|
|
return result.rowcount > 0
|
|
|
|
async def update_status(self, alert_id: int, *, status: str, **fields) -> int:
|
|
"""无条件更新预警状态（放行失败回写「已确认」等场景）。不 commit。"""
|
|
result = await self.db.execute(
|
|
update(FinRiskAlert)
|
|
.where(FinRiskAlert.id == alert_id)
|
|
.values(status=status, **fields)
|
|
)
|
|
return result.rowcount
|