Jarvis-Brain 0.1.5.5__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- jarvis_brain-0.1.5.5/.gitignore +4 -0
- jarvis_brain-0.1.5.5/PKG-INFO +10 -0
- jarvis_brain-0.1.5.5/README.md +246 -0
- jarvis_brain-0.1.5.5/mcp_tools/__init__.py +0 -0
- jarvis_brain-0.1.5.5/mcp_tools/dp_tools.py +166 -0
- jarvis_brain-0.1.5.5/mcp_tools/main.py +31 -0
- jarvis_brain-0.1.5.5/pyproject.toml +34 -0
- jarvis_brain-0.1.5.5/tools/__init__.py +0 -0
- jarvis_brain-0.1.5.5/tools/browser_manager.py +47 -0
- jarvis_brain-0.1.5.5/tools/tools.py +108 -0
- jarvis_brain-0.1.5.5/uv.lock +2132 -0
|
@@ -0,0 +1,246 @@
|
|
|
1
|
+
# Jarvis Brain MCP
|
|
2
|
+
|
|
3
|
+
<div align="center">
|
|
4
|
+
|
|
5
|
+
一个基于 FastMCP 和 DrissionPage 的浏览器自动化 MCP 服务器
|
|
6
|
+
|
|
7
|
+
[](https://www.python.org/downloads/)
|
|
8
|
+
[](https://github.com/yourusername/jarvis-mcp)
|
|
9
|
+
|
|
10
|
+
## 📖 简介
|
|
11
|
+
|
|
12
|
+
Jarvis Brain MCP 是一个强大的浏览器自动化工具,通过 Model Context Protocol (MCP) 提供服务。它集成了 DrissionPage 浏览器控制能力,支持多浏览器实例管理、智能 WAF 检测、HTML 获取与压缩等功能,特别适用于网页爬取、自动化测试和反爬虫分析场景。
|
|
13
|
+
|
|
14
|
+
## ✨ 核心特性
|
|
15
|
+
|
|
16
|
+
### 🌐 浏览器管理
|
|
17
|
+
- **多实例浏览器池**: 使用单例模式管理多个浏览器实例,支持并发操作
|
|
18
|
+
- **标签页控制**: 创建、切换、关闭标签页,灵活管理浏览器标签
|
|
19
|
+
- **智能端口分配**: 自动分配随机端口 (9223-9934),避免端口冲突
|
|
20
|
+
|
|
21
|
+
### 🛡️ WAF 检测
|
|
22
|
+
- **多维度检测**: 对比 requests、有头浏览器、无头浏览器三种方式获取的 HTML
|
|
23
|
+
- **Cookie 特征识别**: 自动识别瑞数、加速乐等常见 WAF 的 Cookie 特征
|
|
24
|
+
- **智能推荐**: 根据检测结果推荐最适合的采集方案(requests / headless / head)
|
|
25
|
+
|
|
26
|
+
### 📄 HTML 处理
|
|
27
|
+
- **智能压缩**: 自动移除 style、script、meta 标签及相关属性
|
|
28
|
+
- **压缩比计算**: 实时计算压缩率,用于 WAF 检测判断
|
|
29
|
+
- **本地保存**: 将获取的 HTML 保存到本地,便于后续分析
|
|
30
|
+
|
|
31
|
+
### 🔍 元素检测
|
|
32
|
+
- **CSS 选择器支持**: 检测页面中是否存在指定的 CSS 选择器元素
|
|
33
|
+
- **智能格式化**: 自动添加 `css:` 前缀,简化使用
|
|
34
|
+
|
|
35
|
+
## 🚀 快速开始
|
|
36
|
+
|
|
37
|
+
### 安装
|
|
38
|
+
|
|
39
|
+
```bash
|
|
40
|
+
pip install Jarvis_Brain
|
|
41
|
+
```
|
|
42
|
+
|
|
43
|
+
或从源码安装:
|
|
44
|
+
|
|
45
|
+
```bash
|
|
46
|
+
git clone https://github.com/yourusername/jarvis-mcp.git
|
|
47
|
+
cd jarvis-mcp
|
|
48
|
+
pip install -e .
|
|
49
|
+
```
|
|
50
|
+
|
|
51
|
+
## 🛠️ MCP 工具列表
|
|
52
|
+
|
|
53
|
+
### TeamNode-Dp 模块
|
|
54
|
+
|
|
55
|
+
#### 1. visit_url
|
|
56
|
+
打开指定 URL 并创建浏览器实例。
|
|
57
|
+
|
|
58
|
+
**参数:**
|
|
59
|
+
- `url` (str): 要访问的网页 URL
|
|
60
|
+
|
|
61
|
+
**返回:**
|
|
62
|
+
- `message`: 操作结果消息
|
|
63
|
+
- `tab_id`: 标签页 ID
|
|
64
|
+
- `browser_port`: 浏览器端口号
|
|
65
|
+
|
|
66
|
+
#### 2. get_html
|
|
67
|
+
获取指定标签页的 HTML 源码并保存到本地。
|
|
68
|
+
|
|
69
|
+
**参数:**
|
|
70
|
+
- `browser_port` (int): 浏览器端口号
|
|
71
|
+
- `tab_id` (str): 标签页 ID
|
|
72
|
+
|
|
73
|
+
**返回:**
|
|
74
|
+
- `message`: 操作结果消息
|
|
75
|
+
- `tab_id`: 标签页 ID
|
|
76
|
+
- `html_local_path`: HTML 文件保存路径
|
|
77
|
+
|
|
78
|
+
#### 3. get_new_tab
|
|
79
|
+
在指定浏览器中创建新标签页并打开 URL。
|
|
80
|
+
|
|
81
|
+
**参数:**
|
|
82
|
+
- `browser_port` (int): 浏览器端口号
|
|
83
|
+
- `url` (str): 要访问的 URL
|
|
84
|
+
|
|
85
|
+
**返回:**
|
|
86
|
+
- `message`: 操作结果消息
|
|
87
|
+
- `tab_id`: 新标签页 ID
|
|
88
|
+
|
|
89
|
+
#### 4. switch_tab
|
|
90
|
+
切换到指定的标签页。
|
|
91
|
+
|
|
92
|
+
**参数:**
|
|
93
|
+
- `browser_port` (int): 浏览器端口号
|
|
94
|
+
- `tab_id` (str): 要切换到的标签页 ID
|
|
95
|
+
|
|
96
|
+
**返回:**
|
|
97
|
+
- `message`: 操作结果消息
|
|
98
|
+
|
|
99
|
+
#### 5. close_tab
|
|
100
|
+
关闭指定的标签页。
|
|
101
|
+
|
|
102
|
+
**参数:**
|
|
103
|
+
- `browser_port` (int): 浏览器端口号
|
|
104
|
+
- `tab_id` (str): 要关闭的标签页 ID
|
|
105
|
+
|
|
106
|
+
**返回:**
|
|
107
|
+
- `message`: 操作结果消息
|
|
108
|
+
|
|
109
|
+
#### 6. check_selector
|
|
110
|
+
检查标签页中是否存在指定的 CSS 选择器元素。
|
|
111
|
+
|
|
112
|
+
**参数:**
|
|
113
|
+
- `browser_port` (int): 浏览器端口号
|
|
114
|
+
- `tab_id` (str): 标签页 ID
|
|
115
|
+
- `css_selector` (str): CSS 选择器
|
|
116
|
+
|
|
117
|
+
**返回:**
|
|
118
|
+
- `message`: 操作结果消息
|
|
119
|
+
- `tab_id`: 标签页 ID
|
|
120
|
+
- `selector`: 完整的选择器
|
|
121
|
+
- `selector_ele_exist` (bool): 元素是否存在
|
|
122
|
+
|
|
123
|
+
### JarvisNode 模块
|
|
124
|
+
|
|
125
|
+
#### 7. assert_waf
|
|
126
|
+
智能检测网页是否使用了 WAF 及页面渲染类型。
|
|
127
|
+
|
|
128
|
+
**检测原理:**
|
|
129
|
+
1. 通过 Cookie 特征识别已知 WAF(瑞数、加速乐等)
|
|
130
|
+
2. 对比 requests、无头浏览器、有头浏览器获取的 HTML 压缩比
|
|
131
|
+
3. 根据压缩比差异判断页面类型和推荐采集方案
|
|
132
|
+
|
|
133
|
+
**参数:**
|
|
134
|
+
- `browser_port` (int): 浏览器端口号
|
|
135
|
+
- `tab_id` (str): 标签页 ID
|
|
136
|
+
|
|
137
|
+
**返回:**
|
|
138
|
+
- `message`: 操作结果消息
|
|
139
|
+
- `tab_id`: 标签页 ID
|
|
140
|
+
- `recommend_team`: 推荐的采集方案
|
|
141
|
+
- `requests`: 静态页面,无防护
|
|
142
|
+
- `drissionpage_headless`: 动态页面或有 requests 防护
|
|
143
|
+
- `drissionpage_head`: 有无头检测或复杂 WAF
|
|
144
|
+
- `raw_head_rate_difference`: requests 与有头浏览器压缩比差异
|
|
145
|
+
- `raw_headless_rate_difference`: requests 与无头浏览器压缩比差异
|
|
146
|
+
- `head_headless_rate_difference`: 有头与无头浏览器压缩比差异
|
|
147
|
+
|
|
148
|
+
## 📊 WAF 检测逻辑
|
|
149
|
+
|
|
150
|
+
### 判定规则
|
|
151
|
+
|
|
152
|
+
| 场景 | requests vs 有头 | requests vs 无头 | 有头 vs 无头 | 推荐方案 | 说明 |
|
|
153
|
+
|------|------------------|------------------|--------------|----------|------|
|
|
154
|
+
| 静态页面无防护 | < 40% | < 40% | < 40% | `requests` | 三种方式结果一致 |
|
|
155
|
+
| 动态页面 / requests 防护 | > 40% | > 40% | < 30% | `drissionpage_headless` | requests 拿不到正确结果 |
|
|
156
|
+
| 无头检测 / 复杂 WAF | < 15% | > 40% | > 40% | `drissionpage_head` | 必须使用有头浏览器 |
|
|
157
|
+
| 已知 WAF (Cookie) | - | - | - | `drissionpage_head` | 检测到瑞数/加速乐等 Cookie |
|
|
158
|
+
| 状态码检测 | 412/521 | - | - | `drissionpage_head` | 瑞数(412)/加速乐(521) |
|
|
159
|
+
|
|
160
|
+
### Cookie 特征库
|
|
161
|
+
|
|
162
|
+
当前支持识别的 WAF:
|
|
163
|
+
- **瑞数**: Cookie name 长度为 13,value 长度为 88
|
|
164
|
+
- **加速乐**: Cookie name 包含 `_jsl`
|
|
165
|
+
|
|
166
|
+
## 🏗️ 项目结构
|
|
167
|
+
|
|
168
|
+
```
|
|
169
|
+
Jarvis-mcp/
|
|
170
|
+
├── mcp_tools/ # MCP 工具模块
|
|
171
|
+
│ ├── __init__.py
|
|
172
|
+
│ ├── main.py # 主入口,注册 MCP 工具
|
|
173
|
+
│ └── dp_tools.py # DrissionPage 工具函数
|
|
174
|
+
├── tools/ # 核心工具模块
|
|
175
|
+
│ ├── __init__.py
|
|
176
|
+
│ ├── browser_manager.py # 浏览器池管理(单例模式)
|
|
177
|
+
│ └── tools.py # HTML 处理、WAF 检测等工具函数
|
|
178
|
+
├── dist/ # 打包文件
|
|
179
|
+
├── pyproject.toml # 项目配置
|
|
180
|
+
└── README.md # 项目文档
|
|
181
|
+
```
|
|
182
|
+
|
|
183
|
+
## 🔧 技术栈
|
|
184
|
+
|
|
185
|
+
- **[FastMCP](https://github.com/jlowin/fastmcp)**: MCP 服务器框架
|
|
186
|
+
- **[DrissionPage](https://github.com/g1879/DrissionPage)**: 浏览器控制库
|
|
187
|
+
- **[htmlmin](https://github.com/mankyd/htmlmin)**: HTML 压缩
|
|
188
|
+
- **[BeautifulSoup4](https://www.crummy.com/software/BeautifulSoup/)**: HTML 解析
|
|
189
|
+
- **[curl_cffi](https://github.com/yifeikong/curl_cffi)**: HTTP 请求库
|
|
190
|
+
|
|
191
|
+
## 📝 使用方法
|
|
192
|
+
|
|
193
|
+
### teamNode mcp配置
|
|
194
|
+
|
|
195
|
+
```json
|
|
196
|
+
"JarvisNode": {
|
|
197
|
+
"command": "uvx",
|
|
198
|
+
"args": ["--python", "3.11", "--from", "Jarvis_Brain@latest", "jarvis-mcp"],
|
|
199
|
+
"env": {
|
|
200
|
+
"MCP_MODULES": "TeamNode-Dp",
|
|
201
|
+
"BASE_CWD": os.getcwd(),
|
|
202
|
+
}
|
|
203
|
+
},
|
|
204
|
+
```
|
|
205
|
+
|
|
206
|
+
### JarvisNode mcp配置
|
|
207
|
+
|
|
208
|
+
```json
|
|
209
|
+
"JarvisNode": {
|
|
210
|
+
"command": "uvx",
|
|
211
|
+
"args": ["--python", "3.11", "--from", "Jarvis_Brain@latest", "jarvis-mcp"],
|
|
212
|
+
"env": {
|
|
213
|
+
"MCP_MODULES": "TeamNode-Dp,JarvisNode",
|
|
214
|
+
"BASE_CWD": os.getcwd(),
|
|
215
|
+
}
|
|
216
|
+
},
|
|
217
|
+
```
|
|
218
|
+
|
|
219
|
+
## 🌟 应用场景
|
|
220
|
+
|
|
221
|
+
1. **网页爬虫**: 智能选择最优采集方案,提高爬取效率
|
|
222
|
+
2. **反爬虫分析**: 快速识别网站使用的 WAF 类型
|
|
223
|
+
3. **自动化测试**: 多浏览器实例并发测试
|
|
224
|
+
4. **数据采集**: 处理动态渲染、反爬虫网站
|
|
225
|
+
5. **安全研究**: 分析网站防护策略
|
|
226
|
+
|
|
227
|
+
## 📄 许可证
|
|
228
|
+
|
|
229
|
+
本项目采用 MIT 许可证。
|
|
230
|
+
|
|
231
|
+
## 🤝 贡献
|
|
232
|
+
|
|
233
|
+
欢迎提交 Issue 和 Pull Request!
|
|
234
|
+
|
|
235
|
+
## 📮 联系方式
|
|
236
|
+
|
|
237
|
+
如有问题或建议,请通过以下方式联系:
|
|
238
|
+
|
|
239
|
+
- 提交 [Issue](https://github.com/yourusername/jarvis-mcp/issues)
|
|
240
|
+
- 邮箱: your.email@example.com
|
|
241
|
+
|
|
242
|
+
---
|
|
243
|
+
|
|
244
|
+
<div align="center">
|
|
245
|
+
Made with ❤️ by Jarvis Team
|
|
246
|
+
</div>
|
|
File without changes
|
|
@@ -0,0 +1,166 @@
|
|
|
1
|
+
import json
|
|
2
|
+
import os
|
|
3
|
+
from typing import Any
|
|
4
|
+
|
|
5
|
+
from DrissionPage._elements.none_element import NoneElement
|
|
6
|
+
from fastmcp import FastMCP
|
|
7
|
+
|
|
8
|
+
from tools.tools import compress_html, requests_html, dp_headless_html, assert_waf_cookie, dp_mcp_message_pack
|
|
9
|
+
|
|
10
|
+
html_source_code_local_save_path = os.path.join(os.getcwd(), "html-source-code")
|
|
11
|
+
waf_status_code_dict = {
|
|
12
|
+
412: "瑞数",
|
|
13
|
+
521: "加速乐"
|
|
14
|
+
}
|
|
15
|
+
|
|
16
|
+
|
|
17
|
+
def register_visit_url(mcp: FastMCP, browser_manager):
|
|
18
|
+
@mcp.tool(name="visit_url", description="使用Drissionpage打开url访问某个网站")
|
|
19
|
+
async def visit_url(url: str) -> dict[str, Any]:
|
|
20
|
+
port, _browser = browser_manager.create_browser()
|
|
21
|
+
tab = _browser.get_tab()
|
|
22
|
+
tab.get(url)
|
|
23
|
+
tab_id = tab.tab_id
|
|
24
|
+
return dp_mcp_message_pack(
|
|
25
|
+
f"已在[{port}]端口创建浏览器对象,并已打开链接:{url}",
|
|
26
|
+
tab_id=tab_id,
|
|
27
|
+
browser_port=port
|
|
28
|
+
)
|
|
29
|
+
|
|
30
|
+
|
|
31
|
+
def register_get_html(mcp: FastMCP, browser_manager):
|
|
32
|
+
@mcp.tool(name="get_html", description="使用Drissionpage获取某一个tab页的html")
|
|
33
|
+
async def get_html(browser_port: int, tab_id: str) -> dict[str, Any]:
|
|
34
|
+
_browser = browser_manager.get_browser(browser_port)
|
|
35
|
+
tab = _browser.get_tab(tab_id)
|
|
36
|
+
file_name = tab.title + f"_{tab_id}.html"
|
|
37
|
+
if not os.path.exists(html_source_code_local_save_path):
|
|
38
|
+
os.makedirs(html_source_code_local_save_path)
|
|
39
|
+
abs_path = os.path.join(html_source_code_local_save_path, file_name)
|
|
40
|
+
with open(abs_path, "w", encoding="utf-8") as f:
|
|
41
|
+
min_html, compress_rate = compress_html(tab.html)
|
|
42
|
+
f.write(min_html)
|
|
43
|
+
return dp_mcp_message_pack(f"已保存tab页:【{tab_id}】的html源码", tab_id=tab_id, html_local_path=abs_path)
|
|
44
|
+
|
|
45
|
+
|
|
46
|
+
def register_get_new_tab(mcp: FastMCP, browser_manager):
|
|
47
|
+
@mcp.tool(name="get_new_tab", description="使用Drissionpage创建一个新的tab页,在新的tab页中打开url")
|
|
48
|
+
async def get_new_tab(browser_port: int, url: str) -> dict[str, Any]:
|
|
49
|
+
_browser = browser_manager.get_browser(browser_port)
|
|
50
|
+
tab = _browser.new_tab(url)
|
|
51
|
+
_browser.activate_tab(tab)
|
|
52
|
+
tab_id = tab.tab_id
|
|
53
|
+
return dp_mcp_message_pack(f"已创建新的tab页,并打开链接:{url}", tab_id=tab_id)
|
|
54
|
+
|
|
55
|
+
|
|
56
|
+
def register_switch_tab(mcp: FastMCP, browser_manager):
|
|
57
|
+
@mcp.tool(name="switch_tab", description="根据传入的tab_id切换到对应的tab页", )
|
|
58
|
+
async def switch_tab(browser_port: int, tab_id: str) -> dict[str, Any]:
|
|
59
|
+
_browser = browser_manager.get_browser(browser_port)
|
|
60
|
+
_browser.activate_tab(tab_id)
|
|
61
|
+
return dp_mcp_message_pack(f"已将tab页:【{tab_id}】切换至最前端")
|
|
62
|
+
|
|
63
|
+
|
|
64
|
+
def register_close_tab(mcp: FastMCP, browser_manager):
|
|
65
|
+
@mcp.tool(name="close_tab", description="根据传入的tab_id关闭tab页", )
|
|
66
|
+
async def close_tab(browser_port, tab_id) -> dict[str, Any]:
|
|
67
|
+
_browser = browser_manager.get_browser(browser_port)
|
|
68
|
+
_browser.close_tabs(tab_id)
|
|
69
|
+
return dp_mcp_message_pack(f"已将tab页:【{tab_id}】关闭")
|
|
70
|
+
|
|
71
|
+
|
|
72
|
+
def register_check_selector(mcp: FastMCP, browser_manager):
|
|
73
|
+
@mcp.tool(name="check_selector", description="查找tab页中是否包含元素,并返回元素attr_name所对应的值")
|
|
74
|
+
async def check_selector(browser_port: int, tab_id: str, css_selector: str, attr_name: str = "text") -> dict[str, Any]:
|
|
75
|
+
_browser = browser_manager.get_browser(browser_port)
|
|
76
|
+
target_tab = _browser.get_tab(tab_id)
|
|
77
|
+
css_selector = css_selector
|
|
78
|
+
if "css:" not in css_selector:
|
|
79
|
+
css_selector = "css:" + css_selector
|
|
80
|
+
target_eles = target_tab.eles(css_selector)
|
|
81
|
+
exist_flag = False
|
|
82
|
+
if len(target_eles) != 0:
|
|
83
|
+
exist_flag = True
|
|
84
|
+
if attr_name == "text":
|
|
85
|
+
ele_text_list = [i.text.replace("\n", "") for i in target_eles]
|
|
86
|
+
attr_output = "\n".join(ele_text_list)
|
|
87
|
+
else:
|
|
88
|
+
attr_output = json.dumps([i.attr(attr_name) for i in target_eles])
|
|
89
|
+
return dp_mcp_message_pack(
|
|
90
|
+
f"已完成tab页:【{tab_id}】对:【{css_selector}】的检查",
|
|
91
|
+
tab_id=tab_id,
|
|
92
|
+
selector=css_selector,
|
|
93
|
+
selector_ele_exist=exist_flag,
|
|
94
|
+
attr_output=attr_output
|
|
95
|
+
)
|
|
96
|
+
|
|
97
|
+
|
|
98
|
+
def register_quit_browser(mcp: FastMCP, browser_manager):
|
|
99
|
+
@mcp.tool(name="quit_browser", description="退出浏览器会话,关闭浏览器")
|
|
100
|
+
async def quit_browser(browser_port: int) -> dict[str, Any]:
|
|
101
|
+
flag, _browser = browser_manager.remove_page(browser_port)
|
|
102
|
+
if flag:
|
|
103
|
+
_browser.quit()
|
|
104
|
+
return dp_mcp_message_pack(
|
|
105
|
+
f"浏览器[{browser_port}],退出会话,关闭浏览器{'成功' if flag else '失败'}",
|
|
106
|
+
browser_port=browser_port,
|
|
107
|
+
quit_flag=flag
|
|
108
|
+
)
|
|
109
|
+
|
|
110
|
+
|
|
111
|
+
def register_assert_waf(mcp: FastMCP, browser_manager):
|
|
112
|
+
@mcp.tool(name="assert_waf",
|
|
113
|
+
description="通过对比requests、有头浏览器、无头浏览器获取到的html,判断网页是否使用了waf以及是否为动态渲染的网页")
|
|
114
|
+
async def assert_waf(browser_port: int, tab_id: str) -> dict[str, Any]:
|
|
115
|
+
_browser = browser_manager.get_browser(browser_port)
|
|
116
|
+
target_tab = _browser.get_tab(tab_id)
|
|
117
|
+
recommend_team = "drissionpage_head"
|
|
118
|
+
head_cookies = target_tab.cookies()
|
|
119
|
+
# 通过cookie判断是否有waf
|
|
120
|
+
waf_flag, waf_type = assert_waf_cookie(head_cookies)
|
|
121
|
+
head_html = target_tab.html
|
|
122
|
+
min_head_html, head_rate = compress_html(head_html, only_text=True)
|
|
123
|
+
raw_html, status_code = requests_html(target_tab.url)
|
|
124
|
+
min_raw_html, raw_rate = compress_html(raw_html, only_text=True)
|
|
125
|
+
r_h_rate_diff = abs(head_rate - raw_rate)
|
|
126
|
+
# 如果有已知的防火墙,则不浪费时间使用无头获取html和压缩比了
|
|
127
|
+
if waf_flag or status_code in waf_status_code_dict.keys():
|
|
128
|
+
return dp_mcp_message_pack(
|
|
129
|
+
f"已完成tab页:【{tab_id}】的分析,该tab页存在waf",
|
|
130
|
+
tab_id=tab_id,
|
|
131
|
+
recommend_team=recommend_team,
|
|
132
|
+
raw_head_rate_difference=r_h_rate_diff,
|
|
133
|
+
raw_headless_rate_difference=0,
|
|
134
|
+
head_headless_rate_difference=0
|
|
135
|
+
)
|
|
136
|
+
|
|
137
|
+
headless_html = dp_headless_html(target_tab.url)
|
|
138
|
+
min_headless_html, headless_rate = compress_html(headless_html, only_text=True)
|
|
139
|
+
r_hless_rate_diff = abs(raw_rate - headless_rate)
|
|
140
|
+
h_hless_rate_diff = abs(head_rate - headless_rate)
|
|
141
|
+
# 最优情况:requests,dp有头和无头拿到的结果基本一致,认定为没有防护的静态网页
|
|
142
|
+
if r_h_rate_diff < 40 and r_hless_rate_diff < 40 and h_hless_rate_diff < 40:
|
|
143
|
+
recommend_team = "requests"
|
|
144
|
+
# 最差情况:requests,dp有头和无头拿到的结果差距都很大,认定为有浏览器无头检测+动态网页
|
|
145
|
+
# if r_h_rate_diff < 40 and r_hless_rate_diff < 40 and h_hless_rate_diff < 40:
|
|
146
|
+
# 较差1:dp有头和无头差距很小,但是requests拿不到正确结果,认定为有requests防护 or 动态网页
|
|
147
|
+
elif h_hless_rate_diff < 30 and r_hless_rate_diff > 40:
|
|
148
|
+
recommend_team = "drissionpage_headless"
|
|
149
|
+
# 较差2:有头和无头差距很大,但是requests和有头拿到的结果基本一致,认定为要么有别的没有防护requests的waf,或者间歇性的瑞数【此时应该拿有头的cookie去判断其中是否有瑞数特征,上面已经做了】
|
|
150
|
+
# if r_h_rate_diff < 15 and h_hless_rate_diff > 40:
|
|
151
|
+
return dp_mcp_message_pack(
|
|
152
|
+
f"已完成tab页:【{tab_id}】的分析,该tab页存在waf",
|
|
153
|
+
tab_id=tab_id,
|
|
154
|
+
recommend_team=recommend_team,
|
|
155
|
+
raw_head_rate_difference=r_h_rate_diff,
|
|
156
|
+
raw_headless_rate_difference=h_hless_rate_diff,
|
|
157
|
+
head_headless_rate_difference=h_hless_rate_diff
|
|
158
|
+
)
|
|
159
|
+
|
|
160
|
+
# def register_highlight_element_captcha(mcp: FastMCP, browser_manager):
|
|
161
|
+
# @mcp.tool(name="highlight_element_captcha",
|
|
162
|
+
# description="将传入的Selector在页面上高亮,并截屏")
|
|
163
|
+
# async def highlight_element_captcha(browser_port: int, tab_id: str, selector: str) -> dict[str, Any]:
|
|
164
|
+
# _browser = browser_manager.get_browser(browser_port)
|
|
165
|
+
# tab = _browser.get_tab(tab_id)
|
|
166
|
+
# tab.ele
|
|
@@ -0,0 +1,31 @@
|
|
|
1
|
+
from fastmcp import FastMCP
|
|
2
|
+
|
|
3
|
+
from mcp_tools.dp_tools import *
|
|
4
|
+
from tools.browser_manager import browser_manager
|
|
5
|
+
|
|
6
|
+
mcp = FastMCP("Jarvis Brain Mcp Tools")
|
|
7
|
+
|
|
8
|
+
# 根据环境变量加载模块
|
|
9
|
+
enabled_modules = os.getenv("MCP_MODULES", "TeamNode-Dp").split(",")
|
|
10
|
+
base_cwd = os.getenv("BASE_CWD", os.path.expanduser('~'))
|
|
11
|
+
|
|
12
|
+
if "TeamNode-Dp" in enabled_modules:
|
|
13
|
+
# 页面管理
|
|
14
|
+
register_close_tab(mcp, browser_manager)
|
|
15
|
+
register_switch_tab(mcp, browser_manager)
|
|
16
|
+
register_get_new_tab(mcp, browser_manager)
|
|
17
|
+
# 功能
|
|
18
|
+
register_visit_url(mcp, browser_manager)
|
|
19
|
+
register_get_html(mcp, browser_manager)
|
|
20
|
+
register_check_selector(mcp, browser_manager)
|
|
21
|
+
|
|
22
|
+
if "JarvisNode" in enabled_modules:
|
|
23
|
+
register_assert_waf(mcp, browser_manager)
|
|
24
|
+
|
|
25
|
+
|
|
26
|
+
def main():
|
|
27
|
+
mcp.run(transport="stdio")
|
|
28
|
+
|
|
29
|
+
|
|
30
|
+
if __name__ == '__main__':
|
|
31
|
+
main()
|
|
@@ -0,0 +1,34 @@
|
|
|
1
|
+
[project]
|
|
2
|
+
name = "Jarvis_Brain" # 别人下载时用的名字,必须在 PyPI 上唯一
|
|
3
|
+
version = "0.1.5.5"
|
|
4
|
+
description = "Jarvis brain mcp"
|
|
5
|
+
dependencies = [
|
|
6
|
+
"fastmcp",
|
|
7
|
+
"DrissionPage",
|
|
8
|
+
"minify-html",
|
|
9
|
+
"beautifulsoup4",
|
|
10
|
+
"curl_cffi"
|
|
11
|
+
]
|
|
12
|
+
requires-python = ">=3.10"
|
|
13
|
+
|
|
14
|
+
[project.scripts]
|
|
15
|
+
jarvis-mcp = "mcp_tools.main:main"
|
|
16
|
+
|
|
17
|
+
[build-system]
|
|
18
|
+
requires = ["hatchling"]
|
|
19
|
+
build-backend = "hatchling.build"
|
|
20
|
+
|
|
21
|
+
[tool.hatch.build.targets.wheel]
|
|
22
|
+
packages = ["mcp_tools", "tools"]
|
|
23
|
+
|
|
24
|
+
[tool.hatch.build]
|
|
25
|
+
exclude = [
|
|
26
|
+
"venv",
|
|
27
|
+
".venv",
|
|
28
|
+
"env",
|
|
29
|
+
".env",
|
|
30
|
+
".idea",
|
|
31
|
+
"*.pyc",
|
|
32
|
+
"dist",
|
|
33
|
+
"call_test"
|
|
34
|
+
]
|
|
File without changes
|
|
@@ -0,0 +1,47 @@
|
|
|
1
|
+
"""浏览器池管理模块 - 单例模式确保状态共享"""
|
|
2
|
+
import random
|
|
3
|
+
from typing import Optional, Tuple
|
|
4
|
+
import os
|
|
5
|
+
from DrissionPage import ChromiumPage, ChromiumOptions
|
|
6
|
+
|
|
7
|
+
|
|
8
|
+
class BrowserManager:
|
|
9
|
+
"""浏览器池管理器 - 使用单例模式"""
|
|
10
|
+
_instance = None
|
|
11
|
+
|
|
12
|
+
def __new__(cls):
|
|
13
|
+
if cls._instance is None:
|
|
14
|
+
cls._instance = super().__new__(cls)
|
|
15
|
+
cls._instance.browser_pool = {}
|
|
16
|
+
return cls._instance
|
|
17
|
+
|
|
18
|
+
def create_browser(self) -> Tuple[int, ChromiumPage]:
|
|
19
|
+
"""创建新的浏览器实例"""
|
|
20
|
+
random_port = random.randint(9223, 9934)
|
|
21
|
+
while random_port in self.browser_pool:
|
|
22
|
+
random_port = random.randint(9223, 9934)
|
|
23
|
+
|
|
24
|
+
co = ChromiumOptions().set_local_port(random_port)
|
|
25
|
+
custom_data_dir = os.path.join(os.path.expanduser('~'), 'DrissionPage', "userData", f"{random_port}")
|
|
26
|
+
co.set_user_data_path(custom_data_dir) # 设置用户数据路径
|
|
27
|
+
# if not os.path.exists(custom_data_dir):
|
|
28
|
+
# os.makedirs(custom_data_dir)
|
|
29
|
+
self.browser_pool[random_port] = ChromiumPage(co)
|
|
30
|
+
return random_port, self.browser_pool[random_port]
|
|
31
|
+
|
|
32
|
+
def get_browser(self, port: int) -> Optional[ChromiumPage]:
|
|
33
|
+
"""根据端口获取浏览器实例"""
|
|
34
|
+
return self.browser_pool.get(port)
|
|
35
|
+
|
|
36
|
+
def remove_browser(self, port: int) -> Tuple[bool, Optional[ChromiumPage]]:
|
|
37
|
+
"""根据端口移除浏览器实例"""
|
|
38
|
+
browser = self.browser_pool.pop(port, None)
|
|
39
|
+
return browser is not None, browser
|
|
40
|
+
|
|
41
|
+
def list_browsers(self) -> list[int]:
|
|
42
|
+
"""列出所有活跃的浏览器端口"""
|
|
43
|
+
return list(self.browser_pool.keys())
|
|
44
|
+
|
|
45
|
+
|
|
46
|
+
# 创建全局单例实例
|
|
47
|
+
browser_manager = BrowserManager()
|
|
@@ -0,0 +1,108 @@
|
|
|
1
|
+
import time
|
|
2
|
+
import random
|
|
3
|
+
import os
|
|
4
|
+
import minify_html
|
|
5
|
+
from DrissionPage import ChromiumPage, ChromiumOptions
|
|
6
|
+
from bs4 import BeautifulSoup
|
|
7
|
+
from curl_cffi import requests
|
|
8
|
+
from lxml import html, etree
|
|
9
|
+
|
|
10
|
+
|
|
11
|
+
# 使用requests获取html,用于测试是否使用了瑞数和jsl
|
|
12
|
+
def requests_html(url):
|
|
13
|
+
headers = {
|
|
14
|
+
"User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/142.0.0.0 Safari/537.36",
|
|
15
|
+
}
|
|
16
|
+
response = requests.get(url, headers=headers, verify=False)
|
|
17
|
+
response.encoding = "utf-8"
|
|
18
|
+
return response.text, response.status_code
|
|
19
|
+
|
|
20
|
+
|
|
21
|
+
# 使用dp无头模式获取html,用于测试是否使用了其他waf,如移动waf
|
|
22
|
+
def dp_headless_html(url):
|
|
23
|
+
opt = ChromiumOptions().headless(True)
|
|
24
|
+
opt.set_argument('--no-sandbox')
|
|
25
|
+
"""创建新的浏览器实例"""
|
|
26
|
+
random_port = random.randint(9934, 10034)
|
|
27
|
+
custom_data_dir = os.path.join(os.path.expanduser('~'), 'DrissionPage', "userData", f"{random_port}")
|
|
28
|
+
opt.set_user_data_path(custom_data_dir) # 设置用户数据路径
|
|
29
|
+
opt.set_local_port(random_port)
|
|
30
|
+
page = ChromiumPage(opt)
|
|
31
|
+
tab = page.latest_tab
|
|
32
|
+
tab.get(url)
|
|
33
|
+
# todo: 目前没有更好的方式,为了数据渲染完全,只能硬等【受网速波动影响比较大】
|
|
34
|
+
time.sleep(10)
|
|
35
|
+
page_html = tab.html
|
|
36
|
+
# 无头浏览器在用完之后一定要记得再page级别进行quit
|
|
37
|
+
page.quit()
|
|
38
|
+
return page_html
|
|
39
|
+
|
|
40
|
+
|
|
41
|
+
# 压缩html
|
|
42
|
+
def compress_html(content, only_text=False):
|
|
43
|
+
doc = html.fromstring(content)
|
|
44
|
+
# 删除 style 和 script 标签
|
|
45
|
+
for element in doc.xpath('//style | //script'):
|
|
46
|
+
element.getparent().remove(element)
|
|
47
|
+
|
|
48
|
+
# 删除 link 标签
|
|
49
|
+
for link in doc.xpath('//link[@rel="stylesheet"]'):
|
|
50
|
+
link.getparent().remove(link)
|
|
51
|
+
|
|
52
|
+
# 删除 meta 标签(新增功能)
|
|
53
|
+
for meta in doc.xpath('//meta'):
|
|
54
|
+
meta.getparent().remove(meta)
|
|
55
|
+
|
|
56
|
+
# 删除 style 属性
|
|
57
|
+
for element in doc.xpath('//*[@style]'):
|
|
58
|
+
element.attrib.pop('style')
|
|
59
|
+
|
|
60
|
+
# 删除所有 on* 事件属性
|
|
61
|
+
for element in doc.xpath('//*'):
|
|
62
|
+
for attr in list(element.attrib.keys()):
|
|
63
|
+
if attr.startswith('on'):
|
|
64
|
+
element.attrib.pop(attr)
|
|
65
|
+
|
|
66
|
+
result = etree.tostring(doc, encoding='unicode')
|
|
67
|
+
result = minify_html.minify(result)
|
|
68
|
+
compress_rate = round(len(content) / len(result) * 100)
|
|
69
|
+
print(f"html压缩比=> {compress_rate}%")
|
|
70
|
+
if not only_text:
|
|
71
|
+
return result, compress_rate
|
|
72
|
+
soup = BeautifulSoup(result, 'html.parser')
|
|
73
|
+
result = soup.get_text(strip=True)
|
|
74
|
+
return result, compress_rate
|
|
75
|
+
|
|
76
|
+
|
|
77
|
+
# 通过cookie判断是否有waf,需要通过遇到的例子,不断的完善cookie判别函数
|
|
78
|
+
def assert_waf_cookie(cookies: list):
|
|
79
|
+
for cookie in cookies:
|
|
80
|
+
cookie_name = cookie['name']
|
|
81
|
+
cookie_value = cookie['value']
|
|
82
|
+
if len(cookie_name) == 13 and len(cookie_value) == 88:
|
|
83
|
+
return True, "瑞数"
|
|
84
|
+
if "_jsl" in cookie_name:
|
|
85
|
+
return True, "加速乐"
|
|
86
|
+
return False, "没有waf"
|
|
87
|
+
|
|
88
|
+
|
|
89
|
+
# 对dp_mcp的消息打包
|
|
90
|
+
def dp_mcp_message_pack(message: str, **kwargs):
|
|
91
|
+
text_obj = {key: value for key, value in kwargs.items()}
|
|
92
|
+
text_obj.update({"message": message})
|
|
93
|
+
return {
|
|
94
|
+
"content": [{
|
|
95
|
+
"type": "text",
|
|
96
|
+
# "text": json.dumps(text_obj, ensure_ascii=False)
|
|
97
|
+
"text": text_obj
|
|
98
|
+
}]
|
|
99
|
+
}
|
|
100
|
+
|
|
101
|
+
# todo: 大致盘一下各种判定的逻辑【以下的所有压缩比之间的差距均取“绝对值”】
|
|
102
|
+
# 1. 如果requests、无头、有头获取到的压缩比之间从差距都在15%以内,则认定该页面是静态页面,此时优先使用requests请求
|
|
103
|
+
# 2. 如果requests的status_code为特定的412,或者521,则判定是瑞数和jsl。[此时还有一个特点:requests的压缩比会与其他两种方式获取到的压缩比差距非常大(一两千的那种)]
|
|
104
|
+
# 3. 如果requests、无头、有头获取到的压缩比之间差距都在40%以上,则判定该页面只可以用有头采集
|
|
105
|
+
# 4. 如果无头和有头获取到的压缩比之间差距小于15%,但是requests和无头的差距大于40%,则认定该页面可以使用无头浏览器采集
|
|
106
|
+
# 5. 如果requests和有头获取到的压缩比之间差距小于15%,但是无头和有头的差距大于40%,则认定该页面优先使用有头浏览器采集
|
|
107
|
+
# 【此时可能是:1.使用了别的检测无头的waf。2.网站使用瑞数,但是这次请求没有拦截requests(不知道是不是瑞数那边故意设置的),
|
|
108
|
+
# 此时如果想进一步判定是否是瑞数,可以使用有头浏览器取一下cookies,如果cookies里面存在瑞数的cookie,那么就可以断定是瑞数】
|