Update app.py

app.py

@@ -4,10 +4,30 @@ from fastapi import FastAPI, Request
 import uvicorn
 import time
 import json
-from urllib.parse import unquote
+from urllib.parse import unquote, urlparse
 
 app = FastAPI()
 
+def convert_cookies_to_dict(cookies):
+    cookies = dict([l.split("=", 1) for l in cookies.split("; ")])
+    return cookies
+
+def get_root_domain(url):
+    # 解析URL
+    parsed_url = urlparse(url)
+    # 获取域名部分
+    domain = parsed_url.netloc
+    
+    # 分割域名部分以获取根域名
+    # 假设根域名是域名的最后两个部分
+    parts = domain.split('.')
+    if len(parts) > 1:
+        # 返回根域名部分
+        return '.'.join(parts[-2:])
+    else:
+        # 如果域名部分少于两个部分，返回整个域名
+        return domain
+        
 @app.get("/")
 def main():
     return {"code": 200,"msg":"Success"}
@@ -40,6 +60,13 @@ def chrome(url:str=None,wait:int=5,header:str=None,cookie:str=None):
     if type(cookie) == str:
         header_array.update({"cookie":unquote(cookie)})
 
+    if 'cookie' in header_array:
+        cookie_array = 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;');
+
     options = Options()
     options.add_argument('--headless')