Update app.py

app.py

@@ -38,6 +38,7 @@ def chrome(url:str=None,wait:int=5,header:str=None,cookie:str=None):
     # 必须有目标url
     if type(url) == str:
         target_url = unquote(url)
+        target_domain = get_root_domain(target_url)
     else:
         return {"code": 500,"msg":"No target URL"}
 
@@ -60,19 +61,21 @@ def chrome(url:str=None,wait:int=5,header:str=None,cookie:str=None):
     if type(cookie) == str:
         header_array.update({"cookie":unquote(cookie)})
 
+    
+
     options = Options()
     options.add_argument('--headless')
     
     driver = webdriver.Chrome(options=options)
 
+    driver.get(target_domain)
+
     if 'cookie' in header_array:
         cookie_array = convert_cookies_to_dict(header_array['cookie'])
         del header_array['cookie']
-        cookie_domain = f'.{get_root_domain(target_url)}'
         for key, value in cookie_array.items():
-            driver.execute_script(f'document.cookie = "{key}={value}; path=/; domain={cookie_domain}; expires=Thu, 01 Jan 2050 00:00:00 GMT;"');
-        print(driver.execute_script(f'return document.cookie'))
-        
+            driver.add_cookie({"name": key, "value": value, "domain": f'.{target_domain}', "path": "/"})
+            
     driver.header_overrides = header_array
     
     driver.get(target_url)