init rep

modules/standard_selector.py

@@ -0,0 +1,167 @@
+#  @line_count 200
+"""测试规范选择器，实现规则+AI混合选择策略"""
+from typing import List, Dict, Any, Optional, TYPE_CHECKING
+from .test_standard_loader import TestStandardLoader
+import json
+import re
+
+if TYPE_CHECKING:
+    from .api_client import APIClient
+
+
+class StandardSelector:
+    """测试规范选择器，实现规则+AI混合选择策略"""
+    
+    def __init__(self, loader: TestStandardLoader, api_client: Optional['APIClient'] = None):
+        """
+        初始化规范选择器
+        
+        Args:
+            loader: 测试规范加载器
+            api_client: API客户端，用于AI辅助选择
+        """
+        self.loader = loader
+        self.api_client = api_client
+        self.mapping_rules = loader.get_requirement_mapping()
+        self.keyword_mapping = loader.get_keyword_mapping()
+    
+    def select_standards(self, requirement: Dict[str, Any], use_ai: bool = True) -> List[str]:
+        """
+        为需求选择适用的测试规范
+        
+        Args:
+            requirement: 需求字典，包含requirement_type、description等
+            use_ai: 是否使用AI辅助选择（当规则匹配不足时）
+            
+        Returns:
+            测试规范ID列表
+        """
+        # 第一步：规则匹配
+        rule_based = self._rule_based_selection(requirement)
+        
+        # 第二步：如果规则匹配不足，使用AI补充
+        if use_ai and len(rule_based) < 2 and self.api_client:
+            ai_based = self._ai_based_selection(requirement)
+            # 合并去重
+            all_standards = list(set(rule_based + ai_based))
+            return all_standards
+        
+        return rule_based
+    
+    def _rule_based_selection(self, requirement: Dict[str, Any]) -> List[str]:
+        """
+        基于规则的规范选择
+        
+        Args:
+            requirement: 需求字典
+            
+        Returns:
+            测试规范ID列表
+        """
+        standards = []
+        req_type = requirement.get('requirement_type', '')
+        req_desc = requirement.get('description', '')
+        
+        # 1. 需求类型映射
+        if req_type in self.mapping_rules:
+            mapping = self.mapping_rules[req_type]
+            # 主要测试类型
+            standards.extend(mapping.get('primary', []))
+            # 次要测试类型
+            standards.extend(mapping.get('secondary', []))
+            # 条件性测试类型
+            conditionals = mapping.get('conditional', {})
+            if '有界面' in conditionals and ('界面' in req_desc or '显示' in req_desc):
+                standards.extend(conditionals['有界面'])
+            if '有性能要求' in conditionals and ('性能' in req_desc or '速度' in req_desc or '时间' in req_desc):
+                standards.extend(conditionals['有性能要求'])
+            if '安全关键' in conditionals and ('安全' in req_desc or '危险' in req_desc or '报警' in req_desc):
+                standards.extend(conditionals['安全关键'])
+            if '多软件交互' in conditionals and ('交互' in req_desc or '协同' in req_desc):
+                standards.extend(conditionals['多软件交互'])
+            if '性能相关' in conditionals and ('性能' in req_desc):
+                standards.extend(conditionals['性能相关'])
+            if '安全相关' in conditionals and ('安全' in req_desc):
+                standards.extend(conditionals['安全相关'])
+        
+        # 2. 关键词匹配
+        for keyword, test_types in self.keyword_mapping.items():
+            if keyword in req_desc:
+                standards.extend(test_types)
+        
+        # 3. 接口信息检测
+        if requirement.get('interface_info'):
+            standards.append('外部接口测试')
+        
+        # 去重并返回
+        return list(set(standards))
+    
+    def _ai_based_selection(self, requirement: Dict[str, Any]) -> List[str]:
+        """
+        基于AI的规范选择（当规则匹配不足时使用）
+        
+        Args:
+            requirement: 需求字典
+            
+        Returns:
+            测试规范ID列表
+        """
+        if not self.api_client:
+            return []
+        
+        # 获取所有测试规范名称
+        all_standards = self.loader.get_all_standards()
+        standard_names = [f"{i+1}. {std.get('name', '')}" for i, std in enumerate(all_standards)]
+        
+        prompt = f"""根据以下需求，从14个测试类型中选择最适用的3-5个：
+
+{chr(10).join(standard_names)}
+
+需求类型：{requirement.get('requirement_type', '未知')}
+需求描述：{requirement.get('description', '')}
+
+请只返回测试类型的名称列表，格式为JSON数组，例如：["功能测试", "性能测试", "边界测试"]
+不要包含其他说明文字。"""
+        
+        try:
+            response = self.api_client.call_api(prompt)
+            # 尝试解析JSON
+            test_types = self._parse_ai_response(response)
+            return test_types
+        except Exception as e:
+            # 如果AI选择失败，返回空列表
+            print(f"AI选择失败: {str(e)}")
+            return []
+    
+    def _parse_ai_response(self, response: str) -> List[str]:
+        """
+        解析AI响应，提取测试类型列表
+        
+        Args:
+            response: AI响应文本
+            
+        Returns:
+            测试类型名称列表
+        """
+        # 尝试直接解析JSON
+        try:
+            # 尝试提取JSON数组
+            json_match = re.search(r'\[.*?\]', response, re.DOTALL)
+            if json_match:
+                json_str = json_match.group(0)
+                test_types = json.loads(json_str)
+                if isinstance(test_types, list):
+                    return test_types
+        except:
+            pass
+        
+        # 如果JSON解析失败，尝试文本匹配
+        all_standards = self.loader.get_all_standards()
+        standard_names = [std.get('name', '') for std in all_standards]
+        
+        found_types = []
+        for name in standard_names:
+            if name in response:
+                found_types.append(name)
+        
+        return found_types