Simon
88 lines
2.8 KiB
Python
88 lines
2.8 KiB
Python
# =============================================================================
|
||
# 企微IT智能服务台 — 火绒集成配置管理
|
||
# =============================================================================
|
||
# 说明：从系统配置表(system_configs)读取火绒 AccessKey/Secret/BaseUrl，
|
||
# 构建火绒API客户端实例
|
||
# =============================================================================
|
||
|
||
import logging
|
||
from typing import Optional
|
||
|
||
from sqlalchemy import select
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
||
from app.models.system_config import SystemConfig
|
||
from .client import HuorongClient
|
||
from .exceptions import HuorongConfigError
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
# 火绒配置在 system_configs 表中的 key 前缀
|
||
HUORONG_CONFIG_PREFIX = "integration_huorong_"
|
||
|
||
|
||
async def get_huorong_client(db: AsyncSession) -> HuorongClient:
|
||
"""从系统配置表构建火绒API客户端。
|
||
|
||
读取 integration_huorong_ 前缀的配置项，构建 HuorongClient 实例。
|
||
如果任何必填配置缺失，抛出 HuorongConfigError。
|
||
|
||
Args:
|
||
db: 数据库会话
|
||
|
||
Returns:
|
||
HuorongClient: 已配置的火绒API客户端实例
|
||
|
||
Raises:
|
||
HuorongConfigError: AccessKey ID/Secret/Base URL 任一缺失
|
||
"""
|
||
# 读取三个必填配置
|
||
result = await db.execute(
|
||
select(SystemConfig).where(
|
||
SystemConfig.config_key.startswith(HUORONG_CONFIG_PREFIX)
|
||
)
|
||
)
|
||
configs = list(result.scalars().all())
|
||
|
||
# 构建 key→value 映射
|
||
config_map = {cfg.config_key: cfg.config_value for cfg in configs}
|
||
|
||
access_key_id = config_map.get(f"{HUORONG_CONFIG_PREFIX}access_key_id", "")
|
||
access_key_secret = config_map.get(f"{HUORONG_CONFIG_PREFIX}access_key_secret", "")
|
||
base_url = config_map.get(f"{HUORONG_CONFIG_PREFIX}base_url", "")
|
||
|
||
# 校验必填项
|
||
if not access_key_id or not access_key_secret or not base_url:
|
||
missing = []
|
||
if not access_key_id:
|
||
missing.append("AccessKey ID")
|
||
if not access_key_secret:
|
||
missing.append("AccessKey Secret")
|
||
if not base_url:
|
||
missing.append("Base URL")
|
||
raise HuorongConfigError(
|
||
f"火绒集成配置不完整，缺失: {', '.join(missing)}，请先在管理后台完成配置"
|
||
)
|
||
|
||
return HuorongClient(
|
||
access_key_id=access_key_id,
|
||
access_key_secret=access_key_secret,
|
||
base_url=base_url,
|
||
)
|
||
|
||
|
||
async def is_huorong_configured(db: AsyncSession) -> bool:
|
||
"""检查火绒集成是否已完整配置。
|
||
|
||
Args:
|
||
db: 数据库会话
|
||
|
||
Returns:
|
||
bool: 三项配置均存在且非空时返回 True
|
||
"""
|
||
try:
|
||
client = await get_huorong_client(db)
|
||
return bool(client.access_key_id and client.access_key_secret and client.base_url)
|
||
except HuorongConfigError:
|
||
return False
|