artin/webarena
1
0
Fork 0
mirror of https://github.com/web-arena-x/webarena.git synced 2026-02-06 11:16:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

add options to renew cookie for selected sites

Browse Source
This commit is contained in:
alexisxy 2023-10-20 19:30:24 -04:00
parent 4b86d435b9
commit df87757d47
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
browser_env/auto_login.py
View File

@ -2,6 +2,7 @@
import argparse
import glob
import os
import time
from concurrent.futures import ThreadPoolExecutor
from itertools import combinations
from pathlib import Path
@ -40,10 +41,11 @@ def is_expired(
context_manager = sync_playwright()
playwright = context_manager.__enter__()
browser = playwright.chromium.launch(headless=HEADLESS, slow_mo=SLOW_MO)
browser = playwright.chromium.launch(headless=True, slow_mo=SLOW_MO)
context = browser.new_context(storage_state=storage_state)
page = context.new_page()
page.goto(url)
time.sleep(1)
d_url = page.url
content = page.content()
context_manager.__exit__()
@ -151,4 +153,7 @@ if __name__ == "__main__":
if not args.site_list:
main()
else:
renew_comb(args.site_list, auth_folder=args.auth_folder)
if "all" in args.site_list:
main(auth_folder=args.auth_folder)
else:
renew_comb(args.site_list, auth_folder=args.auth_folder)