Selenium with antidetect browser profiles

"""Launch one XDLogin profile and attach Selenium to it through ChromeDriver.

    XDLOGIN_PROFILE_ID=<id> python python/selenium_attach.py [url] [--stop]

XDLOGIN_BASE_URL  API address (default http://127.0.0.1:31337)
XDLOGIN_API_KEY   only if you set a key in Settings > API & Auto
CHROMEDRIVER      path to a chromedriver matching the profile's Chrome; without it
                  Selenium Manager picks one for the Chrome installed on this machine
--stop            stop the profile afterwards instead of leaving it running
"""
import os
import sys
import time
from urllib.parse import quote

import requests
from selenium import webdriver
from selenium.common.exceptions import SessionNotCreatedException
from selenium.webdriver.chrome.service import Service

BASE = os.environ.get("XDLOGIN_BASE_URL", "http://127.0.0.1:31337").rstrip("/")
KEY = os.environ.get("XDLOGIN_API_KEY")
HEADERS = {"Authorization": f"Bearer {KEY}"} if KEY else {}


def api(method, path, timeout=90):
    try:
        r = requests.request(method, BASE + path, headers=HEADERS, timeout=timeout)
    except requests.ConnectionError:
        sys.exit(f"Can't reach the XDLogin API at {BASE}. In XDLogin: Settings > API & Auto > Start API.")
    if r.status_code == 401:
        sys.exit("401 Unauthorized: set XDLOGIN_API_KEY to the key in Settings > API & Auto, or unset it if no key is set.")
    try:
        body = r.json()
    except ValueError:
        body = {}
    return r.status_code, body


def connect_info(profile_id, timeout=30):
    # 409 until Chrome is ready, and also for a profile that is not running,
    # does not exist or uses the built-in engine: give up with the reason.
    deadline = time.monotonic() + timeout
    while True:
        code, body = api("GET", f"/api/profiles/{quote(profile_id)}/connect", timeout=10)
        if code == 200:
            return body["data"]
        if code != 409 or time.monotonic() > deadline:
            sys.exit(f"No DevTools endpoint: {body.get('error') or f'HTTP {code}'}")
        time.sleep(0.5)


def main():
    profile = os.environ.get("XDLOGIN_PROFILE_ID", "").strip()
    if not profile:
        sys.exit(f"Set XDLOGIN_PROFILE_ID to a profile id. List them with: curl {BASE}/api/profiles")
    args = sys.argv[1:]
    stop = "--stop" in args
    target = next((a for a in args if not a.startswith("--")), "https://example.com")

    code, body = api("POST", f"/api/profiles/{quote(profile)}/launch")
    if code != 200:
        sys.exit(f"Launch failed: {body.get('error') or f'HTTP {code}'}")
    info = connect_info(profile)
    print(f"Profile runs {info['browser']} at {info['debuggerAddress']}")

    options = webdriver.ChromeOptions()
    options.debugger_address = info["debuggerAddress"]  # attach instead of launching a new Chrome
    driver_path = os.environ.get("CHROMEDRIVER")
    service = Service(executable_path=driver_path) if driver_path else Service()
    try:
        driver = webdriver.Chrome(options=options, service=service)
    except SessionNotCreatedException as e:
        sys.exit(f"ChromeDriver refused to attach. Its major version must match the profile's browser "
                 f"({info['browser']}); point CHROMEDRIVER at a matching chromedriver.\n{e.msg}")

    try:
        driver.switch_to.new_window("tab")  # a new tab in the profile, with its cookies
        driver.get(target)
        print("Title:", driver.title)
        print("The page sees:", driver.execute_script(
            "return {userAgent: navigator.userAgent,"
            " timezone: Intl.DateTimeFormat().resolvedOptions().timeZone}"))
        driver.close()  # closes that tab only
    finally:
        driver.quit()  # ends the ChromeDriver session; the profile's Chrome keeps running

    if stop:
        code, body = api("POST", f"/api/profiles/{quote(profile)}/stop")
        print("Profile stopped." if code == 200 else f"Stop failed: {body.get('error') or f'HTTP {code}'}")


if __name__ == "__main__":
    main()