Compare commits
2 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| f7f14d7114 | |||
| ef4d26f11b |
62
app.py
62
app.py
@@ -364,6 +364,59 @@ async def capture_with_playwright(
|
|||||||
return {"success": False, "error": str(e)}
|
return {"success": False, "error": str(e)}
|
||||||
|
|
||||||
|
|
||||||
|
async def capture_with_cdp(
|
||||||
|
url_hint: str = "",
|
||||||
|
action: str = "screenshot",
|
||||||
|
cdp_port: int = 9222
|
||||||
|
):
|
||||||
|
"""
|
||||||
|
使用 CDP 连接到已打开的 Chrome 浏览器
|
||||||
|
用于方案三:手动验证后自动截图
|
||||||
|
"""
|
||||||
|
if not PLAYWRIGHT_AVAILABLE:
|
||||||
|
return {"success": False, "error": "Playwright not installed"}
|
||||||
|
|
||||||
|
try:
|
||||||
|
async with async_playwright() as p:
|
||||||
|
# 连接到已运行的 Chrome
|
||||||
|
browser = await p.chromium.connect_over_cdp(f"http://localhost:{cdp_port}")
|
||||||
|
|
||||||
|
# 获取所有页面
|
||||||
|
contexts = browser.contexts
|
||||||
|
result_data = []
|
||||||
|
|
||||||
|
for context in contexts:
|
||||||
|
for page in context.pages:
|
||||||
|
current_url = page.url
|
||||||
|
title = await page.title()
|
||||||
|
|
||||||
|
# 如果 URL 包含提示词或者是唯一页面
|
||||||
|
if url_hint in current_url or not url_hint:
|
||||||
|
if action == "screenshot":
|
||||||
|
temp_file = CAPTURE_DIR / f"cdp_capture_{uuid.uuid4()[:8]}.png"
|
||||||
|
await page.screenshot(path=str(temp_file), full_page=True)
|
||||||
|
result_data.append({
|
||||||
|
"url": current_url,
|
||||||
|
"title": title,
|
||||||
|
"file_path": str(temp_file)
|
||||||
|
})
|
||||||
|
elif action == "html":
|
||||||
|
html = await page.content()
|
||||||
|
result_data.append({
|
||||||
|
"url": current_url,
|
||||||
|
"title": title,
|
||||||
|
"html": html
|
||||||
|
})
|
||||||
|
|
||||||
|
if result_data:
|
||||||
|
return {"success": True, "pages": result_data}
|
||||||
|
else:
|
||||||
|
return {"success": False, "error": "No matching page found"}
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
return {"success": False, "error": f"CDP connection failed: {str(e)}"}
|
||||||
|
|
||||||
|
|
||||||
def capture_webpage(
|
def capture_webpage(
|
||||||
url: str,
|
url: str,
|
||||||
action: str = "screenshot",
|
action: str = "screenshot",
|
||||||
@@ -467,7 +520,8 @@ def capture():
|
|||||||
"full_page": false,
|
"full_page": false,
|
||||||
"viewport": {"width": 1920, "height": 1080},
|
"viewport": {"width": 1920, "height": 1080},
|
||||||
"wait_time": 2000,
|
"wait_time": 2000,
|
||||||
"backend": "auto" | "agent-browser" | "playwright"
|
"backend": "auto" | "agent-browser" | "playwright" | "chrome-cdp",
|
||||||
|
"cdp_port": 9222 // 用于 chrome-cdp 后端,连接到已打开的 Chrome
|
||||||
}
|
}
|
||||||
"""
|
"""
|
||||||
data = request.get_json()
|
data = request.get_json()
|
||||||
@@ -490,6 +544,8 @@ def capture():
|
|||||||
viewport = data.get("viewport")
|
viewport = data.get("viewport")
|
||||||
wait_time = int(data.get("wait_time", 2000))
|
wait_time = int(data.get("wait_time", 2000))
|
||||||
backend = data.get("backend", "auto")
|
backend = data.get("backend", "auto")
|
||||||
|
cdp_port = int(data.get("cdp_port", 9222))
|
||||||
|
url_hint = data.get("url_hint", "")
|
||||||
|
|
||||||
result = capture_webpage(
|
result = capture_webpage(
|
||||||
url=url,
|
url=url,
|
||||||
@@ -499,7 +555,9 @@ def capture():
|
|||||||
full_page=full_page,
|
full_page=full_page,
|
||||||
viewport=viewport,
|
viewport=viewport,
|
||||||
wait_time=wait_time,
|
wait_time=wait_time,
|
||||||
backend=backend
|
backend=backend,
|
||||||
|
|
||||||
|
|
||||||
)
|
)
|
||||||
|
|
||||||
if not result["success"]:
|
if not result["success"]:
|
||||||
|
|||||||
69
manual_capture_guide.md
Normal file
69
manual_capture_guide.md
Normal file
@@ -0,0 +1,69 @@
|
|||||||
|
# 手动 Chrome 捕获指南
|
||||||
|
|
||||||
|
## 方案三完整流程
|
||||||
|
|
||||||
|
### 1️⃣ 启动真实 Chrome(带调试端口)
|
||||||
|
|
||||||
|
在您的本地机器或服务器上运行:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# Linux
|
||||||
|
google-chrome --remote-debugging-port=9222 https://www.techpowerup.com/gpu-specs/rtx-pro-6000-blackwell.c4272
|
||||||
|
|
||||||
|
# 或者使用已有的 Chrome 进程(需要先关闭所有 Chrome)
|
||||||
|
google-chrome-stable --remote-debugging-port=9222 --user-data-dir=/tmp/chrome-debug
|
||||||
|
```
|
||||||
|
|
||||||
|
### 2️⃣ 手动验证(如 Cloudflare)
|
||||||
|
|
||||||
|
- 在浏览器中手动完成验证过程
|
||||||
|
- 等待页面完全加载
|
||||||
|
- 确认可以看到正常内容
|
||||||
|
|
||||||
|
### 3️⃣ 连接到 Chrome 并截图
|
||||||
|
|
||||||
|
验证完成后,运行脚本:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 连接到 Chrome 并截图
|
||||||
|
python3.12 /tmp/connect_chrome.py techpowerup
|
||||||
|
|
||||||
|
# 或者捕获当前所有页面
|
||||||
|
python3.12 /tmp/connect_chrome.py
|
||||||
|
```
|
||||||
|
|
||||||
|
### 4️⃣ 查看结果
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 查看截图
|
||||||
|
ls -lh /tmp/chrome_capture_*.png
|
||||||
|
|
||||||
|
# 查看HTML
|
||||||
|
head -100 /tmp/chrome_html_0.html
|
||||||
|
```
|
||||||
|
|
||||||
|
## API 集成方案
|
||||||
|
|
||||||
|
如果需要通过 API 获取内容,可以:
|
||||||
|
|
||||||
|
1. 在服务代码中添加 CDP 连接选项
|
||||||
|
2. 用户先手动验证
|
||||||
|
3. API 连接到已验证的浏览器实例
|
||||||
|
|
||||||
|
## 注意事项
|
||||||
|
|
||||||
|
⚠️ **重要**:
|
||||||
|
- Chrome 必须带 `--remote-debugging-port=9222` 启动
|
||||||
|
- 所有 Chrome 进程必须先关闭(或使用新的 user-data-dir)
|
||||||
|
- 验证完成后才能运行脚本
|
||||||
|
- 截图后浏览器保持打开,可以继续使用
|
||||||
|
|
||||||
|
## 一键脚本
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 启动 Chrome 并等待用户验证
|
||||||
|
google-chrome --remote-debugging-port=9222 "https://www.techpowerup.com/gpu-specs/rtx-pro-6000-blackwell.c4272" &
|
||||||
|
echo "请在浏览器中完成验证,然后按 Enter 继续截图..."
|
||||||
|
read
|
||||||
|
python3.12 /tmp/connect_chrome.py techpowerup
|
||||||
|
```
|
||||||
Reference in New Issue
Block a user