import socket
import psutil
import subprocess

def kill_chrome_processes():
    """关闭所有Chrome进程"""
    for proc in psutil.process_iter(['name']):
        if 'chrome' in proc.info['name'].lower():
            try:
                proc.kill()
            except:
                pass
    print("已关闭所有Chrome进程")

def start_chrome_debug():
    """启动调试模式的Chrome"""
    chrome_path = r"C:\Program Files\Google\Chrome\Application\chrome.exe"
    debug_dir = r"C:\ChromeDebug"
    
    cmd = f'"{chrome_path}" --remote-debugging-port=9222 --user-data-dir="{debug_dir}"'
    
    try:
        subprocess.Popen(cmd)
        print("Chrome已启动")
    except Exception as e:
        print(f"启动Chrome失败: {e}")

def check_port(port=9222):
    """检查端口是否开启"""
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    result = sock.connect_ex(('127.0.0.1', port))
    sock.close()
    return result == 0

def setup_chrome_debug():
    """设置Chrome调试环境"""
    print("正在检查Chrome调试环境...")
    
    if check_port():
        print("调试端口已经开启")
        return True
        
    print("正在重启Chrome...")
    kill_chrome_processes()
    start_chrome_debug()
    
    # 等待端口开启
    import time
    for i in range(5):
        if check_port():
            print("调试端口已成功开启")
            return True
        print(f"等待端口开启...({i+1}/5)")
        time.sleep(2)
    
    print("无法开启调试端口")
    return False

# 运行设置
if __name__ == "__main__":
    if setup_chrome_debug():
        print("Chrome调试环境设置成功，可以开始运行爬虫代码")
    else:
        print("设置失败，请检查Chrome安装路径或尝试手动启动")