快捷導(dǎo)航

Python中多瀏覽器實例項目的隔離策略與實現(xiàn)

更新時間：2025年03月04日 09:10:35 作者：百錦再@新空間代碼工作室

在 Python 實現(xiàn)多瀏覽器實例的 JavaScript 注入時,要確保數(shù)據(jù)隔離,會話隔離,存儲隔離,下面是一些關(guān)鍵的隔離策略和代碼實現(xiàn),希望對大家有所幫助

在 Python 實現(xiàn)多瀏覽器實例的 JavaScript 注入時，要確保 數(shù)據(jù)隔離、會話隔離、存儲隔離，否則多個實例之間可能會共享 Cookie、LocalStorage 或者其他持久化數(shù)據(jù)，導(dǎo)致數(shù)據(jù)污染或沖突。以下是一些關(guān)鍵的隔離策略和代碼實現(xiàn)。

1. 數(shù)據(jù)隔離

問題：

默認(rèn)情況下，不同瀏覽器實例可能會共享同一個 Profile（如 Chrome 賬戶、緩存等）。
共享 LocalStorage 或 SessionStorage 可能導(dǎo)致數(shù)據(jù)沖突。

解決方案：

啟動 Chrome 時，使用 --user-data-dir 讓每個實例使用獨立的數(shù)據(jù)目錄。
禁用緩存 --disable-cache，避免數(shù)據(jù)復(fù)用。
運行時清理 LocalStorage 和 SessionStorage。

代碼示例（Pyppeteer）：

import asyncio
import os
from pyppeteer import launch

async def inject_js(browser_id, url, script):
    user_data_dir = f"./chrome_profiles/profile_{browser_id}"  # 每個實例獨立的用戶數(shù)據(jù)目錄
    os.makedirs(user_data_dir, exist_ok=True)
    
    browser = await launch(
        headless=False,
        args=[
            '--no-sandbox',
            '--disable-cache',
            f'--user-data-dir={user_data_dir}',  # 使用獨立的用戶數(shù)據(jù)目錄
        ]
    )
    
    page = await browser.newPage()
    await page.goto(url)
    
    # 執(zhí)行 JavaScript 代碼并清理本地存儲
    await page.evaluate(f"""
        {script}
        localStorage.clear();
        sessionStorage.clear();
    """)

    print(f"[{browser_id}] 執(zhí)行完畢")
    await browser.close()

async def main():
    url = "https://www.example.com"
    script = "document.body.style.backgroundColor = 'green';"
    
    tasks = [inject_js(i, url, script) for i in range(5)]  # 5 個瀏覽器實例
    await asyncio.gather(*tasks)

asyncio.run(main())

關(guān)鍵點：

獨立 user-data-dir：確保不同實例的數(shù)據(jù)目錄不同，防止緩存、Cookie、LocalStorage 共享。
清除本地存儲：每次執(zhí)行前，清除 localStorage 和 sessionStorage，避免數(shù)據(jù)污染。

2. 會話隔離

問題：

瀏覽器實例可能會共享 Cookie、SessionStorage，導(dǎo)致數(shù)據(jù)不安全或干擾測試。

解決方案：

通過無痕模式（Incognito Mode）啟動瀏覽器，每個實例都會有獨立的會話。
顯式清除 Cookies，避免會話共享。

代碼示例（Selenium）：

from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.chrome.options import Options
import multiprocessing

def inject_js_selenium(browser_id, url, script):
    print(f"[{browser_id}] 啟動瀏覽器...")

    chrome_options = Options()
    chrome_options.add_argument("--incognito")  # 無痕模式
    chrome_options.add_argument("--disable-cache")
    chrome_options.add_argument("--disable-application-cache")
    
    driver = webdriver.Chrome(service=Service("/path/to/chromedriver"), options=chrome_options)
    driver.get(url)
    
    print(f"[{browser_id}] 注入 JavaScript 代碼...")
    driver.execute_script(script)

    # 清除 Cookie，確保會話隔離
    driver.delete_all_cookies()
    
    print(f"[{browser_id}] 注入完成")
    driver.quit()

if __name__ == "__main__":
    url = "https://www.example.com"
    script = "document.body.style.backgroundColor = 'blue';"
    
    processes = []
    for i in range(5):  # 5個瀏覽器實例
        p = multiprocessing.Process(target=inject_js_selenium, args=(i, url, script))
        p.start()
        processes.append(p)
    
    for p in processes:
        p.join()

關(guān)鍵點：

使用 --incognito 選項：避免不同實例共享 SessionStorage。
清除 Cookie：driver.delete_all_cookies() 確保不會話復(fù)用。

3. 存儲隔離

問題：

數(shù)據(jù)存儲可能會發(fā)生共享（如 IndexedDB、LocalStorage）。
需要在每個實例執(zhí)行前，清除本地存儲和數(shù)據(jù)庫。

解決方案：

通過 JavaScript 代碼清除 IndexedDB、LocalStorage 和 SessionStorage。

代碼示例（Pyppeteer）：

async def clear_storage(page):
    await page.evaluate("""
        indexedDB.databases().then(dbs => {
            for (let db of dbs) {
                indexedDB.deleteDatabase(db.name);
            }
        });
        localStorage.clear();
        sessionStorage.clear();
    """)
    print("存儲清理完畢")

async def inject_js(browser_id, url, script):
    browser = await launch(headless=False, args=['--no-sandbox'])
    page = await browser.newPage()
    await page.goto(url)
    
    # 執(zhí)行 JavaScript 代碼并清理存儲
    await clear_storage(page)
    await page.evaluate(script)

    print(f"[{browser_id}] 執(zhí)行完畢")
    await browser.close()

async def main():
    url = "https://www.example.com"
    script = "document.body.style.backgroundColor = 'yellow';"
    
    tasks = [inject_js(i, url, script) for i in range(5)]
    await asyncio.gather(*tasks)

asyncio.run(main())

關(guān)鍵點：

indexedDB.deleteDatabase()：刪除 IndexedDB 以防止數(shù)據(jù)泄漏。
清理 localStorage 和 sessionStorage，確保不會話存儲干擾。

4. 進程隔離

問題：

共享瀏覽器進程可能導(dǎo)致線程安全問題或數(shù)據(jù)共享。
需要確保每個實例運行在獨立進程中。

解決方案：

通過多進程 multiprocessing 或 asyncio 任務(wù)并發(fā)，確保每個瀏覽器實例是獨立的進程。

代碼示例（多進程 + Selenium）：

import multiprocessing
from selenium import webdriver
from selenium.webdriver.chrome.options import Options

def start_browser(instance_id):
    options = Options()
    options.add_argument("--headless")  
    options.add_argument("--incognito")  # 開啟無痕模式
    options.add_argument(f"--user-data-dir=./chrome_profile_{instance_id}")  # 獨立數(shù)據(jù)目錄

    driver = webdriver.Chrome(options=options)
    driver.get("https://www.example.com")
    
    # 注入 JavaScript 代碼
    driver.execute_script("document.body.style.backgroundColor = 'pink';")
    
    driver.quit()

if __name__ == "__main__":
    num_instances = 5  # 啟動5個實例
    processes = []

    for i in range(num_instances):
        p = multiprocessing.Process(target=start_browser, args=(i,))
        p.start()
        processes.append(p)

    for p in processes:
        p.join()

關(guān)鍵點：

獨立 user-data-dir 目錄，避免多個 Selenium 實例共享數(shù)據(jù)。
multiprocessing 多進程，確保瀏覽器實例的完全隔離。

總結(jié)

隔離類型	方案	實現(xiàn)方式
數(shù)據(jù)隔離	獨立 user-data-dir、清除 LocalStorage	--user-data-dir、localStorage.clear()
會話隔離	開啟 --incognito 無痕模式	delete_all_cookies()
存儲隔離	清除 IndexedDB、LocalStorage、SessionStorage	indexedDB.deleteDatabase()
進程隔離	multiprocessing 啟動多個瀏覽器實例	multiprocessing.Process()