feat:物料存入数据库

This commit is contained in:
2026-09-30 09:00:44 -07:00
parent 3e88b339a0
commit f6302e89b5
5 changed files with 131 additions and 26 deletions
+37 -22
View File
@@ -1,4 +1,4 @@
"""读取前端组件目录"""
"""读取前端组件目录:启动时载入进程内存缓存,未载入时回退读取本地 JSON 文件"""
from pathlib import Path
@@ -8,38 +8,53 @@ from app.schemas.component_resource import (
ComponentResult,
)
DEFAULT_CATALOG_PATH = Path(__file__).resolve().parent / "catalog" / "components.json"
# 进程内共享的组件目录,由启动流程载入;agent 与前端接口读取同一份
_cached_catalog: ComponentCatalog | None = None
def set_catalog_cache(catalog: ComponentCatalog | None) -> None:
global _cached_catalog
_cached_catalog = catalog
def get_catalog_cache() -> ComponentCatalog | None:
return _cached_catalog
def load_catalog_file(path: str | Path | None = None) -> ComponentCatalog | None:
# 如果传入的是相对路径,以项目根目录为基准,避免受启动时工作目录影响
target = Path(path) if path is not None else DEFAULT_CATALOG_PATH
if not target.is_absolute():
target = Path(__file__).resolve().parents[2] / target
try:
# utf-8-sig 同时兼容带 BOM 和不带 BOM 的 UTF-8 文件
content = target.read_text(encoding="utf-8-sig")
except FileNotFoundError:
return None
# 文件损坏或定义不合法直接报错,不能伪装成查无结果
return ComponentCatalog.model_validate_json(content)
class ComponentProvider:
def __init__(self, catalog_path: str | Path | None = None):
# 保存可选的目录文件路径,未指定时由查询方法使用项目内的默认目录
# 指定路径时始终读取该文件;未指定时优先使用启动缓存,缓存为空再读默认文件
self.catalog_path = catalog_path
def query(self, query: ComponentQuery) -> ComponentResult:
# TODO 以后迁移数据库,保持组件查询接口不变
# 优先使用调用方指定的路径,否则读取当前模块旁的 catalog/components.json
path = (
Path(self.catalog_path)
if self.catalog_path is not None
else Path(__file__).resolve().parent / "catalog" / "components.json"
catalog = (
_cached_catalog
if self.catalog_path is None and _cached_catalog is not None
else load_catalog_file(self.catalog_path)
)
# 如果传入的是相对路径,以项目根目录为基准,避免受启动时工作目录影响
if not path.is_absolute():
path = Path(__file__).resolve().parents[2] / path
try:
# 每次读取当前文件,避免更新目录后仍使用旧缓存
# utf-8-sig 同时兼容带 BOM 和不带 BOM 的 UTF-8 文件
content = path.read_text(encoding="utf-8-sig")
except FileNotFoundError:
if catalog is None:
# 如果目录文件不存在,返回未支持状态和原因,由调用方决定失败处理方式
return ComponentResult(
status="unsupported",
message="尚未配置前端组件目录,请先导出真实组件定义",
)
# 文件损坏或定义不合法直接报错,不能伪装成查无结果
catalog = ComponentCatalog.model_validate_json(content)
# 如果提供关键词,去除首尾空白并统一大小写,未提供时不启用关键词筛选
keyword = (
query.keyword.strip().casefold() if query.keyword is not None else None
@@ -67,8 +82,8 @@ class ComponentProvider:
# 如果设置了关键词,但检索文本中不包含该关键词,则跳过当前组件
if keyword is not None and keyword not in searchable:
continue
# 所有筛选条件通过,保留完整组件定义供后续规划和动作生成使用
matches.append(component)
# 深拷贝后返回,调用方修改结果不会污染共享缓存
matches.append(component.model_copy(deep=True))
# 根据是否有匹配项区分查询成功和未找到,同时保留目录版本、来源及结果说明
return ComponentResult(