Selenium 模块化项目架构

这是一个使用 Python 和 Selenium 的模块化项目架构，支持通过命令行参数调用不同的业务逻辑，并支持两种方式启动和控制 Google Chrome 浏览器。

文件结构

.
├── browser_manager.py          # 管理浏览器启动和 WebDriver 实例
├── config.py                   # 存放项目配置，如调试端口和用户目录
├── main.py                     # 项目主入口，负责解析参数和调用模块
├── requirements.txt            # 项目依赖库
├── modules/                    # 存放所有独立的业务逻辑模块
│   └── or_address_handler.py   # 处理 o-address.csv 的业务模块
├── data/                       # 存放数据文件
│   └── o-address.csv           # 示例数据文件
└── chrome_user_data/           # (自动生成) Chrome 用户数据目录

安装依赖

在开始之前，请确保你已经安装了 Python。然后通过以下命令安装所需的库：

pip install -r requirements.txt

如何运行

现在，所有操作都通过命令行来执行，使用 main.py 作为入口。

基本命令格式

python main.py --module <模块名称> [--debug]

--module <模块名称>: 必需参数。指定要运行的业务模块。当前可用的模块是 or-address。
--debug: 可选参数。如果带上这个参数，脚本会尝试连接到一个已存在的 Chrome 浏览器，而不是启动一个新浏览器。

示例 1: 启动新浏览器执行任务

这是最常见的用法。下面的命令会启动一个全新的浏览器，并执行 or-address 模块定义的任务。

python main.py --module or-address

示例 2: 连接到现有浏览器执行任务

如果你需要保持登录状态或利用已打开的浏览器会话，可以使用 --debug 模式。

手动启动 Chrome 浏览器：打开你的命令行工具 (CMD 或 PowerShell)，然后运行以下命令。这会启动一个特殊的 Chrome 实例，并开启一个调试端口。
```
chrome.exe --remote-debugging-port=9222 --user-data-dir="d:/project/zhiyun-scripy/chrome_user_data"
```
注意:
- --remote-debugging-port 的值必须与 config.py 中的 CHROME_DEBUGGER_PORT 一致。
- --user-data-dir 指定了一个用户配置文件的目录，请确保路径是绝对路径。
- 如果 chrome.exe 不在你的系统 PATH 中，你可能需要提供它的完整路径。
运行脚本并连接：在另一个终端中，运行以下命令。--debug 参数会告诉脚本去连接你刚才手动打开的浏览器。
```
python main.py --module or-address --debug
```

如何添加新的业务模块？

在 modules/ 目录下创建一个新的 Python 文件，例如 new_task_handler.py。

在这个新文件中，定义一个函数，该函数接收 driver 作为参数。例如：

# modules/new_task_handler.py
def process_new_task(driver):
    print("正在执行新任务...")
    driver.get("https://www.example.com")
    # ... 在这里编写你的新业务逻辑 ...

打开 main.py 文件，将你的新模块和处理函数添加到 MODULES 字典中：

# main.py
from modules.or_address_handler import process_addresses
from modules.new_task_handler import process_new_task # 导入新函数

MODULES = {
    "or-address": process_addresses,
    "new-task": process_new_task, # 添加新模块的映射
}

现在你就可以通过命令行调用你的新模块了：
```
python main.py --module new-task
```

其他说明

browser_manager.py: 一般情况下你不需要修改这个文件。
config.py: 如果你需要更改调试端口或用户数据目录的名称，可以修改这个文件。
data/: 所有需要被脚本读取的数据文件（如 CSV, Excel, TXT）都建议放在这个目录下。

4.0 KiB Raw Blame History Unescape Escape