第 1 章：AI 编程思维重塑

Step 1：原始需求

问题分析：这个需求没有指定搜索范围、匹配规则、输出格式、性能要求——AI 只能猜测，结果大概率不符合预期。

Step 2：优化后的提示词（应用三层模型）

你是一位 Python 系统工具开发者。

【战略层 - 目标】
构建一个命令行文件搜索工具，用于在大型代码仓库中快速定位文件。
目标用户：Linux 开发者，日常需要按文件名模式、内容关键字查找文件。

【战术层 - 约束】
- 语言：Python 3.10+，仅使用标准库
- 性能：支持搜索 10000+ 文件的目录，需要在 3 秒内返回结果
- 设计：使用生成器模式避免一次性加载所有结果到内存
- 错误处理：优雅处理权限不足、符号链接循环等边界情况

【操作层 - 输出】
请生成单个 Python 文件，包含：
1. 完整的类型注解和 docstring
2. argparse 命令行接口：--pattern（glob模式）、--content（正则）、--dir（搜索目录）
3. 彩色终端输出（使用 ANSI 转义码）
4. 搜索进度指示器
5. 所有公开函数附带使用示例的文档注释

Step 3：AI 输出结果（参考实现）

#!/usr/bin/env python3
"""文件搜索工具 — 支持文件名模式匹配和内容正则搜索。

使用示例:
    # 按文件名搜索
    python search.py --pattern "*.py" --dir ./src

    # 按内容搜索
    python search.py --content "TODO|FIXME" --dir ./src

    # 组合搜索
    python search.py --pattern "*.py" --content "class.*TestCase" --dir ./
"""

import argparse
import fnmatch
import os
import re
import sys
import time
from collections.abc import Generator
from dataclasses import dataclass
from typing import Optional


@dataclass
class SearchMatch:
    """搜索结果条目。"""
    path: str
    line_number: Optional[int]
    line_content: Optional[str]
    match_type: str  # 'filename' | 'content'


class FileSearcher:
    """文件搜索引擎，支持文件名和内容双维度搜索。

    使用生成器模式，避免一次性加载所有结果到内存。
    """

    def __init__(
        self,
        root_dir: str,
        pattern: Optional[str] = None,
        content_regex: Optional[str] = None,
        follow_symlinks: bool = False,
    ):
        self.root_dir = os.path.abspath(root_dir)
        self.pattern = pattern
        self.content_regex = re.compile(content_regex) if content_regex else None
        self.follow_symlinks = follow_symlinks
        self._files_scanned: int = 0
        self._matches_found: int = 0

    def search(self) -> Generator[SearchMatch, None, None]:
        """执行搜索，逐个产出匹配结果。"""
        for dirpath, dirnames, filenames in os.walk(
            self.root_dir, followlinks=self.follow_symlinks
        ):
            # 跳过隐藏目录
            dirnames[:] = [d for d in dirnames if not d.startswith('.')]

            for filename in filenames:
                self._files_scanned += 1
                filepath = os.path.join(dirpath, filename)
                relpath = os.path.relpath(filepath, self.root_dir)

                # --- 文件名匹配 ---
                if self.pattern and fnmatch.fnmatch(filename, self.pattern):
                    self._matches_found += 1
                    yield SearchMatch(
                        path=relpath,
                        line_number=None,
                        line_content=None,
                        match_type='filename',
                    )
                    continue  # 文件名已匹配，可选是否继续检查内容

                # --- 内容匹配 ---
                if self.content_regex:
                    try:
                        with open(filepath, 'r', encoding='utf-8', errors='ignore') as f:
                            for lineno, line in enumerate(f, 1):
                                if self.content_regex.search(line):
                                    self._matches_found += 1
                                    yield SearchMatch(
                                        path=relpath,
                                        line_number=lineno,
                                        line_content=line.rstrip('\n'),
                                        match_type='content',
                                    )
                    except (PermissionError, OSError):
                        continue  # 优雅跳过不可读文件

    @property
    def stats(self) -> dict[str, int]:
        return {'files_scanned': self._files_scanned, 'matches_found': self._matches_found}


# 终端颜色
class Colors:
    CYAN = '\033[96m'
    GREEN = '\033[92m'
    YELLOW = '\033[93m'
    RED = '\033[91m'
    RESET = '\033[0m'
    BOLD = '\033[1m'


def format_output(match: SearchMatch, base_dir: str) -> str:
    """格式化单个搜索结果，带语法高亮颜色。"""
    if match.match_type == 'filename':
        return f"{Colors.CYAN}📄 {match.path}{Colors.RESET}"
    else:
        return (
            f"{Colors.YELLOW}📝 {match.path}{Colors.RESET}:{match.line_number}:  "
            f"{match.line_content.strip()[:120]}"
        )


def main():
    parser = argparse.ArgumentParser(
        description='🔍 高性能文件搜索工具',
        formatter_class=argparse.RawDescriptionHelpFormatter,
    )
    parser.add_argument('--pattern', '-p', help='文件名 glob 模式（如 "*.py"）')
    parser.add_argument('--content', '-c', help='文件内容正则表达式')
    parser.add_argument('--dir', '-d', default='.', help='搜索根目录（默认当前目录）')
    parser.add_argument('--follow-symlinks', action='store_true', help='跟踪符号链接')
    args = parser.parse_args()

    if not args.pattern and not args.content:
        parser.error('至少需要指定 --pattern 或 --content 之一')

    start_time = time.perf_counter()
    searcher = FileSearcher(
        root_dir=args.dir,
        pattern=args.pattern,
        content_regex=args.content,
        follow_symlinks=args.follow_symlinks,
    )

    print(f"\n{Colors.BOLD}🔍 开始搜索...{Colors.RESET}\n")

    for match in searcher.search():
        print(format_output(match, args.dir))

    elapsed = time.perf_counter() - start_time
    s = searcher.stats
    print(f"\n{Colors.BOLD}─── 搜索统计 ───{Colors.RESET}")
    print(f"  扫描文件: {s['files_scanned']}")
    print(f"  匹配结果: {s['matches_found']}")
    print(f"  耗时: {elapsed:.2f}s\n")


if __name__ == '__main__':
    main()

Step 4：人工审查要点