JiuContinent
92f490b97e
feat(core): 添加多API密钥支持和配置字段 添加了api_keys、audit_api_keys、approval_api_keys等字段用于支持多个服务密钥， 新增masked_response_fields用于配置响应掩码字段，以及legacy相关配置项。 feat(core): 增强响应数据掩码功能 扩展mask_configured函数支持域名参数，实现更精确的敏感字段掩码控制， 添加自定义掩码字段配置验证器。 feat(scheduler): 添加遗留系统同步调度任务 集成遗留项目和任务同步到定时调度器中，支持通过配置启用或禁用同步功能， 并可设置不同的执行时间计划。 feat(security): 实现多服务密钥认证机制 重构API密钥验证逻辑，支持单个主密钥和多个配置密钥的混合验证模式， 增加服务密钥启用状态检查和角色映射功能。 feat(task_queue): 扩展现有队列任务处理 为日常简报和周报推送任务添加Celery异步处理支持，新增遗留项目和任务同步任务， 统一任务分发接口。 feat(business): 扩展业务模型字段 为工作任务模型添加外部系统标识和外部ID字段，为风险事件模型增加分配、解决、关闭 等相关字段，并创建风险事件操作记录表。 feat(legacy_mysql): 实现遗留任务同步功能 添加遗留任务查询和同步路由，支持从旧MySQL数据库同步任务数据到内部系统， 包括同步结果统计和运行记录。 refactor(dashboard): 更新仪表板统计数据 增加未分配风险和失败推送运行统计，在概览中显示最新的推送和同步运行记录， 完善数据序列化展示。 fix(feishu): 修复审批事件重复处理 实现审批卡片操作事件的唯一性检查，防止重复审批操作，添加事件审计日志记录。 ```
69 lines
2.0 KiB
Python
69 lines
2.0 KiB
Python
from typing import Any
|
|
|
|
from app.core.config import get_settings
|
|
|
|
|
|
MASKED_VALUE = "[MASKED]"
|
|
SENSITIVE_RESPONSE_KEYS = frozenset(
|
|
{
|
|
"account_number",
|
|
"api_key",
|
|
"bank_account",
|
|
"card_no",
|
|
"direct_llm_api_key",
|
|
"email",
|
|
"feishu_app_secret",
|
|
"feishu_verification_token",
|
|
"hermes_api_key",
|
|
"id_card",
|
|
"mobile",
|
|
"openclaw_gateway_token",
|
|
"password",
|
|
"payment_account",
|
|
"phone",
|
|
"secret",
|
|
"tenant_access_token",
|
|
"token",
|
|
}
|
|
)
|
|
|
|
|
|
def mask_configured(value: Any, domain: str | None = None) -> Any:
|
|
"""Mask sensitive response fields when response masking is enabled."""
|
|
|
|
settings = get_settings()
|
|
if not settings.mask_sensitive_responses:
|
|
return value
|
|
configured_fields = {item.lower() for item in settings.masked_response_fields}
|
|
return mask_sensitive(value, domain=domain, configured_fields=configured_fields)
|
|
|
|
|
|
def mask_sensitive(
|
|
value: Any,
|
|
domain: str | None = None,
|
|
configured_fields: set[str] | None = None,
|
|
) -> Any:
|
|
if isinstance(value, dict):
|
|
masked: dict[str, Any] = {}
|
|
for key, item in value.items():
|
|
key_text = str(key)
|
|
if _should_mask(key_text, domain, configured_fields or set()):
|
|
masked[key_text] = MASKED_VALUE
|
|
else:
|
|
masked[key_text] = mask_sensitive(item, domain, configured_fields)
|
|
return masked
|
|
if isinstance(value, list):
|
|
return [mask_sensitive(item, domain, configured_fields) for item in value]
|
|
if isinstance(value, tuple):
|
|
return [mask_sensitive(item, domain, configured_fields) for item in value]
|
|
return value
|
|
|
|
|
|
def _should_mask(key: str, domain: str | None, configured_fields: set[str]) -> bool:
|
|
field = key.lower()
|
|
if field in SENSITIVE_RESPONSE_KEYS or field in configured_fields:
|
|
return True
|
|
if f"*.{field}" in configured_fields:
|
|
return True
|
|
return bool(domain and f"{domain.lower()}.{field}" in configured_fields)
|