54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
# @line_count 100
|
||
"""解析器工厂"""
|
||
from typing import Dict, Any, Type, List
|
||
from .base_adapter import BaseParserAdapter
|
||
from .section_array_adapter import SectionArrayAdapter
|
||
from .requirement_tree_adapter import RequirementTreeAdapter
|
||
|
||
|
||
class ParserFactory:
|
||
"""解析器工厂，自动检测格式并创建合适的适配器"""
|
||
|
||
# 注册所有适配器（按优先级排序）
|
||
_adapters: List[Type[BaseParserAdapter]] = [
|
||
RequirementTreeAdapter, # 新格式优先
|
||
SectionArrayAdapter, # 旧格式
|
||
# 未来可以在这里添加更多适配器
|
||
]
|
||
|
||
@classmethod
|
||
def create_adapter(cls, data: Dict[str, Any]) -> BaseParserAdapter:
|
||
"""
|
||
创建合适的适配器
|
||
|
||
Args:
|
||
data: JSON数据
|
||
|
||
Returns:
|
||
适配器实例
|
||
|
||
Raises:
|
||
ValueError: 如果无法识别格式
|
||
"""
|
||
for adapter_class in cls._adapters:
|
||
if adapter_class.can_parse(data):
|
||
return adapter_class(data)
|
||
|
||
raise ValueError(
|
||
"无法识别JSON格式。支持的格式：\n"
|
||
"- 需求树格式（包含'需求内容'和'文档元数据'）\n"
|
||
"- 章节数组格式（包含'sections'数组）"
|
||
)
|
||
|
||
@classmethod
|
||
def register_adapter(cls, adapter_class: Type[BaseParserAdapter],
|
||
priority: int = 0):
|
||
"""
|
||
注册新的适配器
|
||
|
||
Args:
|
||
adapter_class: 适配器类
|
||
priority: 优先级（越小越优先，0为最高优先级）
|
||
"""
|
||
cls._adapters.insert(priority, adapter_class)
|