playwright保持网站的登录状态的几种方法

目录

• 方法 1：保存和加载浏览器上下文的存储状态
• 实现步骤：
• 方法 2：使用持久化上下文
• 实现步骤：
• 方法 3：使用 Cookies 手动保存和加载会话
• 实现步骤：
• 方法选择

在使用 Playwright 自动化测试时，为了保持网站的登录状态，可以通过以下方法实现：

Playwright 提供了 storageState 功能，可以保存当前的浏览器存储状态（包括登录的会话信息），并在后续会话中加载。

1. 登录并保存状态
   在第一次运行时，执行登录操作并将状态保存到文件。

   from playwright.sync_api import sync_playwright
   
   def save_login_state():
       with sync_playwright() as p:
           browser = p.chromium.launch(headless=False)
           context = browser.new_context()
           page = context.new_page()
   
           # 打开登录页面并完成登录
           page.goto("https://example.com/login")
           page.fill("input[name='username']", "your_username")
           page.fill("input[name='password']", "your_password")
           page.click("button[type='submit']")
   
           # 等待登录完成
           page.wait_for_url("https://example.com/dashboard")
   
           # 保存登录状态到文件
           context.storage_state(path="state.json")
           browser.close()
   
   save_login_state()
   

2. 加载保存的状态
   在后续运行中，加载保存的状态文件，避免重复登录。

   from playwright.sync_api import sync_playwright
   
   def use_saved_state():
       with sync_playwright() as p:
           browser = p.chromium.launch(headless=False)
           # 使用之前保存的存储状态
           context = browser.new_context(storage_state="state.json")
           page = context.new_page()
   
           # 直接访问需要登录的页面
           page.goto("https://example.com/dashboard")
           print(page.title())
           browser.close()
   
   use_saved_state()
   

Playwright 支持使用带有持久化存储的浏览器上下文，自动保留登录状态。

1. 创建持久化上下文
   指定一个文件夹来保存浏览器会话数据：

   from playwright.sync_api import sync_playwright
   
   def use_persistent_context():
       with sync_playwright() as p:
           # 创建持久化上下文
           user_data_dir = "user_data"
           browser = p.chromium.launch_persistent_context(
               user_data_dir, headless=False
           )
           page = browser.new_page()
   
           # 打开需要登录的页面
           page.goto("https://example.com/login")
   
           # 如果首次访问，登录后数据会自动保存到 user_data_dir
           page.fill("input[name='username']", "your_username")
           page.fill("input[name='password']", "your_password")
           page.click("button[type='submit']")
           page.wait_for_url("https://example.com/dashboard")
   
           browser.close()
   
   use_persistent_context()
   

2. 后续运行直接加载会话数据
   在后续运行中，直接加载同一 user_data_dir，会话数据会自动保留：

   from playwright.sync_api import sync_playwright
   
   def use_existing_persistent_context():
       with sync_playwright() as p:
           # 使用之前的持久化上下文
           browser = p.chromium.launch_persistent_context(
               "user_data", headless=False
           )
           page = browser.new_page()
   
           # 访问需要登录的页面
           page.goto("https://example.com/dashboard")
           print(page.title())
           browser.close()
   
   use_existing_persistent_context()
   

通过手动保存和加载 Cookies 来保持登录状态。

1. 保存 Cookies

   from playwright.sync_api import sync_playwright
   import json
   
   def save_cookies():
       with sync_playwright() as p:
           browser = p.chromium.launch(headless=False)
           context = browser.new_context()
           page = context.new_page()
   
           page.goto("https://example.com/login")
           page.fill("input[name='username']", "your_username")
           page.fill("input[name='password']", "your_password")
           page.click("button[type='submit']")
           page.wait_for_url("https://example.com/dashboard")
   
           # 保存 Cookies 到文件
           cookies = context.cookies()
           with open("cookies.json", "w") as f:
               json.dump(cookies, f)
           browser.close()
   
   save_cookies()
   

2. 加载 Cookies

   from playwright.sync_api import sync_playwright
   import json
   
   def load_cookies():
       with sync_playwright() as p:
           browser = p.chromium.launch(headless=False)
           context = browser.new_context()
   
           # 加载 Cookies
           with open("cookies.json", "r") as f:
               cookies = json.load(f)
               context.add_cookies(cookies)
   
           page = context.new_page()
           page.goto("https://example.com/dashboard")
           print(page.title())
           browser.close()
   
   load_cookies()
   

• 方法 1（存储状态）：适合短期保留登录状态，并易于分享状态文件。
• 方法 2（持久化上下文）：适合长期保留会话（类似普通浏览器），无需额外管理文件。
• 方法 3（Cookies 管理）：适合需要灵活操作 Cookies 的场景。

到此这篇关于playwright保持网站的登录状态的几种方法的文章就介绍到这了,更多相关playwright 登录状态内容请搜索风君子博客以前的文章或继续浏览下面的相关文章希望大家以后多多支持风君子博客！

您可能感兴趣的文章:

• Playwright中如何保持登录状态