JiuContinent
4d09d8e2e3
feat: 添加仪表板路由和响应数据脱敏功能 - 添加了仪表板模块路由并集成到主路由器中 - 实现了敏感数据响应脱敏配置和功能 - 增加了 Feishu 审批卡片操作处理功能 - 支持通过任务队列异步推送日常简报和项目周报 - 添加了风险事件生成的任务队列支持 - 在 smoke 测试中增加了相关功能验证 refactor: 格式化模型注册模块导入列表 - 将单行导入列表改为多行格式以提高可读性 ```
61 lines
1.7 KiB
Python
61 lines
1.7 KiB
Python
from typing import Any
|
|
|
|
from celery import Celery
|
|
|
|
from app.core.config import get_settings
|
|
from app.core.constants import ActorValue
|
|
from app.core.database import SessionLocal
|
|
from app.modules.feishu.constants import FeishuReceiveIdType
|
|
|
|
settings = get_settings()
|
|
|
|
celery_app = Celery(
|
|
"company_ai_platform",
|
|
broker=settings.redis_url,
|
|
backend=settings.celery_result_backend_url or settings.redis_url,
|
|
)
|
|
celery_app.conf.task_always_eager = settings.task_queue_always_eager
|
|
|
|
|
|
@celery_app.task(name="reports.push_daily_brief")
|
|
def push_daily_brief(
|
|
receive_id: str | None = None,
|
|
receive_id_type: str = FeishuReceiveIdType.CHAT_ID,
|
|
actor: str = ActorValue.SCHEDULER,
|
|
) -> dict[str, Any]:
|
|
from app.modules.reports.service import ReportService
|
|
|
|
db = SessionLocal()
|
|
try:
|
|
report = ReportService(db).daily_brief()
|
|
return ReportService(db).push_report(report, receive_id, receive_id_type, actor)
|
|
finally:
|
|
db.close()
|
|
|
|
|
|
@celery_app.task(name="reports.push_project_weekly")
|
|
def push_project_weekly(
|
|
receive_id: str | None = None,
|
|
receive_id_type: str = FeishuReceiveIdType.CHAT_ID,
|
|
actor: str = ActorValue.SCHEDULER,
|
|
) -> dict[str, Any]:
|
|
from app.modules.reports.service import ReportService
|
|
|
|
db = SessionLocal()
|
|
try:
|
|
report = ReportService(db).project_weekly()
|
|
return ReportService(db).push_report(report, receive_id, receive_id_type, actor)
|
|
finally:
|
|
db.close()
|
|
|
|
|
|
@celery_app.task(name="risks.generate_events")
|
|
def generate_risk_events(actor: str = ActorValue.SCHEDULER) -> dict[str, Any]:
|
|
from app.modules.risk.service import RiskService
|
|
|
|
db = SessionLocal()
|
|
try:
|
|
return RiskService(db).generate_events(actor=actor)
|
|
finally:
|
|
db.close()
|