JiuContinent
9cf7c44393
feat: 添加生命周期报告和AI规则管理功能 - 在Dockerfile中添加pillow依赖包用于图像处理 - 实现生命周期报告调度任务，支持日报和周报两种类型 - 新增TASK_RUN_LIFECYCLE任务常量和相关配置选项 - 扩展AI Agent服务以支持用户规则，并在分析时应用规则 - 添加AI用户规则创建、更新和查询接口 - 增加项目生命周期和财务需求分析技能 - 扩展现有模型以支持更完整的业务数据字段 - 实现飞书图片上传功能用于报告展示 ```
108 lines
3.3 KiB
Python
108 lines
3.3 KiB
Python
from typing import Any
|
|
|
|
from fastapi import HTTPException, status
|
|
from sqlalchemy import select
|
|
|
|
from app.core.http.pagination import bounded_limit
|
|
from app.core.utils.time import utc_now
|
|
from app.modules.business.models import (
|
|
ReportPushRun,
|
|
)
|
|
from app.modules.business.service import serialize_model
|
|
from app.modules.reports.constants import (
|
|
ReportErrorDetail,
|
|
ReportPushStatus,
|
|
)
|
|
|
|
from app.modules.reports.services.common import _json_safe, _next_code
|
|
|
|
|
|
class ReportPushRunMixin:
|
|
def create_push_run(
|
|
self,
|
|
report_type: str,
|
|
title: str | None,
|
|
receive_id: str | None,
|
|
receive_id_type: str,
|
|
actor: str,
|
|
status: str = ReportPushStatus.PENDING,
|
|
idempotency_key: str | None = None,
|
|
) -> ReportPushRun:
|
|
if idempotency_key:
|
|
existing = self.db.execute(
|
|
select(ReportPushRun).where(
|
|
ReportPushRun.idempotency_key == idempotency_key
|
|
)
|
|
).scalar_one_or_none()
|
|
if existing is not None:
|
|
return existing
|
|
record = ReportPushRun(
|
|
code=_next_code("PUSH"),
|
|
report_type=report_type,
|
|
title=title,
|
|
receive_id=receive_id,
|
|
receive_id_type=receive_id_type,
|
|
status=status,
|
|
actor=actor,
|
|
queued_at=utc_now(),
|
|
idempotency_key=idempotency_key,
|
|
)
|
|
self.db.add(record)
|
|
self.db.commit()
|
|
self.db.refresh(record)
|
|
return record
|
|
|
|
def update_push_run(
|
|
self,
|
|
code: str,
|
|
status: str,
|
|
task_id: str | None = None,
|
|
provider_response: dict[str, Any] | None = None,
|
|
error_message: str | None = None,
|
|
sent: bool = False,
|
|
) -> ReportPushRun:
|
|
record = self._get_push_run(code)
|
|
record.status = status
|
|
if task_id is not None:
|
|
record.task_id = task_id
|
|
if provider_response is not None:
|
|
record.provider_response = _json_safe(provider_response)
|
|
record.error_message = error_message
|
|
if sent:
|
|
record.sent_at = utc_now()
|
|
self.db.commit()
|
|
self.db.refresh(record)
|
|
return record
|
|
|
|
def list_push_runs(
|
|
self,
|
|
status_filter: str | None = None,
|
|
limit: int = 100,
|
|
) -> list[dict[str, Any]]:
|
|
stmt = select(ReportPushRun).order_by(ReportPushRun.id.desc()).limit(
|
|
bounded_limit(limit)
|
|
)
|
|
if status_filter:
|
|
stmt = (
|
|
select(ReportPushRun)
|
|
.where(ReportPushRun.status == status_filter)
|
|
.order_by(ReportPushRun.id.desc())
|
|
.limit(bounded_limit(limit))
|
|
)
|
|
return [serialize_model(item) for item in self.db.execute(stmt).scalars()]
|
|
|
|
def get_push_run(self, code: str) -> dict[str, Any]:
|
|
return serialize_model(self._get_push_run(code))
|
|
|
|
def _get_push_run(self, code: str) -> ReportPushRun:
|
|
|
|
record = self.db.execute(
|
|
select(ReportPushRun).where(ReportPushRun.code == code)
|
|
).scalar_one_or_none()
|
|
if record is None:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_404_NOT_FOUND,
|
|
detail=ReportErrorDetail.PUSH_RUN_NOT_FOUND,
|
|
)
|
|
return record
|