JiuContinent
d7db84571d
feat: 添加飞书用户模块和订阅功能支持 - 新增feishu_users模块用于处理飞书用户身份验证和权限管理 - 新增subscriptions模块用于处理订阅相关功能 - 新增personalization模块用于个性化服务 - 在alembic迁移配置中注册新的模型模块 - 在API路由器中添加feishu_users和subscriptions路由 - 实现事件调度服务的改进，包括错误处理和状态更新优化 - 添加飞书命令处理的权限检查机制 - 实现飞书应用票据事件处理 - 改进审计日志记录功能 ```
91 lines
3.0 KiB
Python
91 lines
3.0 KiB
Python
from typing import Any
|
|
|
|
from sqlalchemy import select
|
|
|
|
from app.core.constants import ActorValue
|
|
from app.modules.audit.constants import (
|
|
AuditAction,
|
|
AuditRiskLevel,
|
|
AuditSource,
|
|
AuditTargetType,
|
|
)
|
|
from app.modules.audit.schemas import AuditLogCreate
|
|
from app.modules.audit.service import AuditService
|
|
from app.modules.business.constants import (
|
|
CLOSED_RISK_STATUSES,
|
|
)
|
|
from app.modules.business.models import (
|
|
RiskEvent,
|
|
)
|
|
from app.modules.business.service import serialize_model
|
|
from app.modules.risk.constants import (
|
|
RiskGenerationAction,
|
|
RiskGenerationResultKey,
|
|
RiskEventPayloadKey,
|
|
)
|
|
|
|
|
|
class RiskGenerationMixin:
|
|
def generate_events(self, actor: str = ActorValue.API) -> dict[str, Any]:
|
|
"""Generate or refresh risk-event ledger entries from current signals."""
|
|
|
|
payloads = self._build_event_payloads()
|
|
created = 0
|
|
updated = 0
|
|
skipped = 0
|
|
items: list[dict[str, Any]] = []
|
|
|
|
for payload in payloads:
|
|
record = self.db.execute(
|
|
select(RiskEvent).where(RiskEvent.code == payload[RiskEventPayloadKey.CODE])
|
|
).scalar_one_or_none()
|
|
if record is None:
|
|
record = RiskEvent(**payload)
|
|
self.db.add(record)
|
|
self.db.flush()
|
|
created += 1
|
|
action = RiskGenerationAction.CREATED
|
|
elif record.status in CLOSED_RISK_STATUSES:
|
|
skipped += 1
|
|
items.append(
|
|
{
|
|
RiskGenerationResultKey.ACTION: RiskGenerationAction.SKIPPED,
|
|
RiskGenerationResultKey.RISK_EVENT: serialize_model(record),
|
|
}
|
|
)
|
|
continue
|
|
else:
|
|
for key, value in payload.items():
|
|
if key != RiskEventPayloadKey.CODE:
|
|
setattr(record, key, value)
|
|
updated += 1
|
|
action = RiskGenerationAction.UPDATED
|
|
items.append(
|
|
{
|
|
RiskGenerationResultKey.ACTION: action,
|
|
RiskGenerationResultKey.RISK_EVENT: serialize_model(record),
|
|
}
|
|
)
|
|
|
|
AuditService(self.db).record(
|
|
AuditLogCreate(
|
|
actor=actor,
|
|
source=AuditSource.RISK,
|
|
action=AuditAction.GENERATE_EVENTS,
|
|
target_type=AuditTargetType.RISK_EVENTS,
|
|
risk_level=AuditRiskLevel.MEDIUM,
|
|
response_payload={
|
|
RiskGenerationResultKey.CREATED: created,
|
|
RiskGenerationResultKey.UPDATED: updated,
|
|
RiskGenerationResultKey.SKIPPED: skipped,
|
|
},
|
|
)
|
|
)
|
|
self.db.commit()
|
|
return {
|
|
RiskGenerationResultKey.CREATED: created,
|
|
RiskGenerationResultKey.UPDATED: updated,
|
|
RiskGenerationResultKey.SKIPPED: skipped,
|
|
RiskGenerationResultKey.ITEMS: items,
|
|
}
|