#!/usr/bin/env python3 """ 验证 API v3.0 重构是否完整 检查代码中是否还有旧的逻辑残留 """ import os import re from pathlib import Path def print_header(title): print(f"\n{'='*60}") print(f" {title}") print('='*60) def search_in_file(filepath, pattern, description): """在文件中搜索模式""" try: with open(filepath, 'r', encoding='utf-8') as f: content = f.read() matches = re.findall(pattern, content, re.MULTILINE) return matches except Exception as e: return None def check_removed_code(): """检查已移除的代码""" print_header("检查已移除的代码") checks = [ { "file": "search/es_query_builder.py", "pattern": r"if field == ['\"]price_ranges['\"]", "description": "硬编码的 price_ranges 逻辑", "should_exist": False }, { "file": "search/es_query_builder.py", "pattern": r"def add_dynamic_aggregations", "description": "add_dynamic_aggregations 方法", "should_exist": False }, { "file": "api/models.py", "pattern": r"aggregations.*Optional\[Dict", "description": "aggregations 参数（在 SearchRequest 中）", "should_exist": False }, { "file": "frontend/static/js/app.js", "pattern": r"price_ranges", "description": "前端硬编码的 price_ranges", "should_exist": False }, { "file": "frontend/static/js/app.js", "pattern": r"displayAggregations", "description": "旧的 displayAggregations 函数", "should_exist": False } ] all_passed = True for check in checks: filepath = os.path.join("/home/tw/SearchEngine", check["file"]) matches = search_in_file(filepath, check["pattern"], check["description"]) if matches is None: print(f" ⚠️ 无法读取：{check['file']}") continue if check["should_exist"]: if matches: print(f" ✓ 存在：{check['description']}") else: print(f" ✗ 缺失：{check['description']}") all_passed = False else: if matches: print(f" ✗ 仍存在：{check['description']}") print(f" 匹配：{matches[:2]}") all_passed = False else: print(f" ✓ 已移除：{check['description']}") return all_passed def check_new_code(): """检查新增的代码""" print_header("检查新增的代码") checks = [ { "file": "api/models.py", "pattern": r"class RangeFilter", "description": "RangeFilter 模型", "should_exist": True }, { "file": "api/models.py", "pattern": r"class FacetConfig", "description": "FacetConfig 模型", "should_exist": True }, { "file": "api/models.py", "pattern": r"class FacetValue", "description": "FacetValue 模型", "should_exist": True }, { "file": "api/models.py", "pattern": r"class FacetResult", "description": "FacetResult 模型", "should_exist": True }, { "file": "api/models.py", "pattern": r"range_filters.*RangeFilter", "description": "range_filters 参数", "should_exist": True }, { "file": "api/models.py", "pattern": r"facets.*FacetConfig", "description": "facets 参数", "should_exist": True }, { "file": "search/es_query_builder.py", "pattern": r"def build_facets", "description": "build_facets 方法", "should_exist": True }, { "file": "search/searcher.py", "pattern": r"def _standardize_facets", "description": "_standardize_facets 方法", "should_exist": True }, { "file": "api/routes/search.py", "pattern": r"@router.get\(['\"]\/suggestions", "description": "/search/suggestions 端点", "should_exist": True }, { "file": "api/routes/search.py", "pattern": r"@router.get\(['\"]\/instant", "description": "/search/instant 端点", "should_exist": True }, { "file": "frontend/static/js/app.js", "pattern": r"function displayFacets", "description": "displayFacets 函数", "should_exist": True }, { "file": "frontend/static/js/app.js", "pattern": r"rangeFilters", "description": "rangeFilters 状态", "should_exist": True } ] all_passed = True for check in checks: filepath = os.path.join("/home/tw/SearchEngine", check["file"]) matches = search_in_file(filepath, check["pattern"], check["description"]) if matches is None: print(f" ⚠️ 无法读取：{check['file']}") continue if check["should_exist"]: if matches: print(f" ✓ 存在：{check['description']}") else: print(f" ✗ 缺失：{check['description']}") all_passed = False else: if matches: print(f" ✗ 仍存在：{check['description']}") all_passed = False else: print(f" ✓ 已移除：{check['description']}") return all_passed def check_documentation(): """检查文档""" print_header("检查文档") docs = [ "API_DOCUMENTATION.md", "API_EXAMPLES.md", "MIGRATION_GUIDE_V3.md", "CHANGES.md" ] all_exist = True for doc in docs: filepath = os.path.join("/home/tw/SearchEngine", doc) if os.path.exists(filepath): size_kb = os.path.getsize(filepath) / 1024 print(f" ✓ 存在：{doc} ({size_kb:.1f} KB)") else: print(f" ✗ 缺失：{doc}") all_exist = False return all_exist def check_imports(): """检查模块导入""" print_header("检查模块导入") import sys sys.path.insert(0, '/home/tw/SearchEngine') try: from api.models import ( RangeFilter, FacetConfig, FacetValue, FacetResult, SearchRequest, SearchResponse, ImageSearchRequest, SearchSuggestRequest, SearchSuggestResponse ) print(" ✓ API 模型导入成功") from search.es_query_builder import ESQueryBuilder print(" ✓ ESQueryBuilder 导入成功") from search.searcher import Searcher, SearchResult print(" ✓ Searcher 导入成功") # 检查方法 qb = ESQueryBuilder('test', ['field1']) if hasattr(qb, 'build_facets'): print(" ✓ build_facets 方法存在") else: print(" ✗ build_facets 方法不存在") return False if hasattr(qb, 'add_dynamic_aggregations'): print(" ✗ add_dynamic_aggregations 方法仍存在（应该已删除）") return False else: print(" ✓ add_dynamic_aggregations 方法已删除") # 检查 SearchResult sr = SearchResult(hits=[], total=0, max_score=0, took_ms=10, facets=[]) if hasattr(sr, 'facets'): print(" ✓ SearchResult.facets 属性存在") else: print(" ✗ SearchResult.facets 属性不存在") return False if hasattr(sr, 'aggregations'): print(" ✗ SearchResult.aggregations 属性仍存在（应该已删除）") return False else: print(" ✓ SearchResult.aggregations 属性已删除") return True except Exception as e: print(f" ✗ 导入失败：{e}") return False def main(): """主函数""" print("\n" + "🔍 开始验证 API v3.0 重构") print(f"项目路径：/home/tw/SearchEngine\n") # 运行检查 check1 = check_removed_code() check2 = check_new_code() check3 = check_documentation() check4 = check_imports() # 总结 print_header("验证总结") results = { "已移除的代码": check1, "新增的代码": check2, "文档完整性": check3, "模块导入": check4 } all_passed = all(results.values()) for name, passed in results.items(): status = "✓ 通过" if passed else "✗ 失败" print(f" {status}: {name}") if all_passed: print(f"\n 🎉 所有检查通过！API v3.0 重构完成。") else: print(f"\n ⚠️ 部分检查失败，请检查上述详情。") print("\n" + "="*60 + "\n") return 0 if all_passed else 1 if __name__ == "__main__": exit(main())