{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "e3d91216-d411-4039-a914-3d5e23cb5335",
   "metadata": {
    "ExecutionIndicator": {
     "show": true
    },
    "execution": {
     "iopub.execute_input": "2024-08-16T09:45:06.206727Z",
     "iopub.status.busy": "2024-08-16T09:45:06.206308Z",
     "iopub.status.idle": "2024-08-16T09:45:06.266958Z",
     "shell.execute_reply": "2024-08-16T09:45:06.266529Z",
     "shell.execute_reply.started": "2024-08-16T09:45:06.206705Z"
    },
    "tags": []
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "['/mnt/workspace/stable-diffusion-webui/modules/ui_postprocessing.py', '/mnt/workspace/stable-diffusion-webui/modules/.ipynb_checkpoints/ui_postprocessing-checkpoint.py']\n"
     ]
    }
   ],
   "source": [
    "import os\n",
    "\n",
    "def find_files_with_string(directory, search_string, file_extension=None):\n",
    "    matching_files = []\n",
    "    # 遍历目录下的所有文件和子目录\n",
    "    for root, dirs, files in os.walk(directory):\n",
    "        for file in files:\n",
    "            # 如果指定了文件格式，检查文件是否匹配\n",
    "            if file_extension and not file.endswith(file_extension):\n",
    "                continue\n",
    "            \n",
    "            file_path = os.path.join(root, file)\n",
    "            # 逐个文件打开并检查是否包含特定字符\n",
    "            try:\n",
    "                with open(file_path, 'r', encoding='utf-8') as f:\n",
    "                    if search_string in f.read():\n",
    "                        matching_files.append(file_path)\n",
    "            except (UnicodeDecodeError, FileNotFoundError):\n",
    "                # 如果文件不能被读取或者文件不存在，跳过\n",
    "                continue\n",
    "    return matching_files\n",
    "\n",
    "# 示例用法\n",
    "directory = '/mnt/workspace/stable-diffusion-webui'\n",
    "search_string = 'Batch from Directory'\n",
    "file_extension = '.py'  # 仅搜索特定格式的文件\n",
    "matching_files = find_files_with_string(directory, search_string, file_extension)\n",
    "print(matching_files)"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.10.12"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}