Update app.py

app.py

@@ -37,7 +37,7 @@ def main():
     return {"code": 200,"msg":"Success"}
     
 @app.get("/chrome")
-def chrome(url:str=None,wait:int=5,header:str=None,cookie:str=None):
+def chrome(url:str=None,wait:int=5,header:str=None,cookie:str=None,cookie_domain:str=None):
     # 开启捕获HAR数据功能，允许使用 driver.har 进行检索
     seleniumwire_options = {
         'enable_har': True  
@@ -93,6 +93,8 @@ def chrome(url:str=None,wait:int=5,header:str=None,cookie:str=None):
         del header_array['cookie']
         for key, value in cookie_array.items():
             try:
+                if cookie_domain :
+                    target_domain = cookie_domain
                 driver.add_cookie({"name": key, "value": quote(value), "domain": f'.{target_domain}', "path": "/"})
             except Exception as e:
                 print("Error Cookie:")