Container commited on
Commit
1bf1c75
·
verified ·
1 Parent(s): c05e38a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -8
app.py CHANGED
@@ -60,20 +60,19 @@ def chrome(url:str=None,wait:int=5,header:str=None,cookie:str=None):
60
  if type(cookie) == str:
61
  header_array.update({"cookie":unquote(cookie)})
62
 
63
-
 
 
 
 
 
 
64
 
65
  options = Options()
66
  options.add_argument('--headless')
67
 
68
  driver = webdriver.Chrome(options=options)
69
 
70
- if 'cookie' in header_array:
71
- cookie_array = convert_cookies_to_dict(header_array['cookie'])
72
- del header_array['cookie']
73
- cookie_domain = f'.{get_root_domain(target_url)}'
74
- for key, value in cookie_array.items():
75
- driver.add_cookie({"name": key, "value": value, "domain": cookie_domain, "path": "/"})
76
-
77
  driver.header_overrides = header_array
78
 
79
  driver.get(target_url)
 
60
  if type(cookie) == str:
61
  header_array.update({"cookie":unquote(cookie)})
62
 
63
+ if 'cookie' in header_array:
64
+ cookie_array = convert_cookies_to_dict(header_array['cookie'])
65
+ del header_array['cookie']
66
+ cookie_domain = f'.{get_root_domain(target_url)}'
67
+ for key, value in cookie_array.items():
68
+ driver.execute_script(f'document.cookie = "{key}={value}; path=/; domain={cookie_domain}; expires=Thu, 01 Jan 2050 00:00:00 GMT;');
69
+ print(driver.execute_script(f'return document.cookie'))
70
 
71
  options = Options()
72
  options.add_argument('--headless')
73
 
74
  driver = webdriver.Chrome(options=options)
75
 
 
 
 
 
 
 
 
76
  driver.header_overrides = header_array
77
 
78
  driver.get(target_url)