124 lines
4.2 KiB
Python
124 lines
4.2 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
修复 Chrome 调试连接问题
|
||
自动检测并处理 Chrome 调试模式连接
|
||
"""
|
||
|
||
import socket
|
||
import subprocess
|
||
import time
|
||
from loguru import logger
|
||
|
||
def check_chrome_debug_port(host="127.0.0.1", port=5003):
|
||
"""检查 Chrome 调试端口是否可用"""
|
||
try:
|
||
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||
sock.settimeout(2)
|
||
result = sock.connect_ex((host, port))
|
||
sock.close()
|
||
return result == 0
|
||
except Exception as e:
|
||
logger.error(f"检查端口 {host}:{port} 失败: {e}")
|
||
return False
|
||
|
||
def start_chrome_debug_mode(port=5003, user_data_dir="C:\\temp\\chrome_debug"):
|
||
"""启动 Chrome 调试模式"""
|
||
try:
|
||
logger.info(f"正在启动 Chrome 调试模式，端口: {port}")
|
||
|
||
# Chrome 可执行文件路径（Windows）
|
||
chrome_paths = [
|
||
r"C:\Program Files\Google\Chrome\Application\chrome.exe",
|
||
r"C:\Program Files (x86)\Google\Chrome\Application\chrome.exe"
|
||
]
|
||
|
||
chrome_exe = None
|
||
for path in chrome_paths:
|
||
try:
|
||
subprocess.run([path, "--version"], capture_output=True, check=True)
|
||
chrome_exe = path
|
||
logger.info(f"找到 Chrome 可执行文件: {path}")
|
||
break
|
||
except:
|
||
continue
|
||
|
||
if not chrome_exe:
|
||
logger.error("未找到 Chrome 可执行文件")
|
||
return False
|
||
|
||
# 启动 Chrome 调试模式
|
||
cmd = [
|
||
chrome_exe,
|
||
f"--remote-debugging-port={port}",
|
||
f"--user-data-dir={user_data_dir}",
|
||
"--no-first-run",
|
||
"--no-default-browser-check"
|
||
]
|
||
|
||
logger.info(f"启动命令: {' '.join(cmd)}")
|
||
process = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
||
|
||
# 等待 Chrome 启动
|
||
time.sleep(5)
|
||
|
||
# 检查端口是否可用
|
||
if check_chrome_debug_port(port=port):
|
||
logger.success("Chrome 调试模式启动成功")
|
||
return True
|
||
else:
|
||
logger.error("Chrome 调试模式启动失败")
|
||
process.terminate()
|
||
return False
|
||
|
||
except Exception as e:
|
||
logger.error(f"启动 Chrome 调试模式失败: {e}")
|
||
return False
|
||
|
||
def main():
|
||
"""主函数"""
|
||
logger.info("=== Chrome 调试模式检查与修复 ===")
|
||
|
||
# 检查当前端口状态
|
||
if check_chrome_debug_port():
|
||
logger.success("Chrome 调试端口 127.0.0.1:5003 已可用")
|
||
logger.info("您现在可以运行: python product/new_data.py")
|
||
return True
|
||
else:
|
||
logger.warning("Chrome 调试端口 127.0.0.1:5003 不可用")
|
||
|
||
# 询问用户是否要启动 Chrome 调试模式
|
||
logger.info("检测到 Chrome 调试模式未启动")
|
||
logger.info("建议使用以下替代方案:")
|
||
logger.info("1. 使用 Playwright 版本 (推荐): python product/new_data_stealth.py")
|
||
logger.info("2. 手动启动 Chrome 调试模式")
|
||
logger.info("3. 自动启动 Chrome 调试模式")
|
||
|
||
choice = input("请选择方案 (1/2/3, 默认1): ").strip() or "1"
|
||
|
||
if choice == "1":
|
||
logger.info("使用 Playwright 版本...")
|
||
logger.info("运行: python product/new_data_stealth.py")
|
||
return True
|
||
elif choice == "2":
|
||
logger.info("请手动运行以下命令:")
|
||
logger.info('chrome.exe --remote-debugging-port=5003 --user-data-dir="C:\\temp\\chrome_debug"')
|
||
return False
|
||
elif choice == "3":
|
||
if start_chrome_debug_mode():
|
||
logger.success("Chrome 调试模式启动成功")
|
||
logger.info("您现在可以运行: python product/new_data.py")
|
||
return True
|
||
else:
|
||
logger.error("自动启动失败，请尝试手动启动")
|
||
return False
|
||
else:
|
||
logger.error("无效选择")
|
||
return False
|
||
|
||
if __name__ == "__main__":
|
||
success = main()
|
||
if success:
|
||
logger.info("修复完成")
|
||
else:
|
||
logger.error("修复失败，请检查上述信息") |