Update app.py

app.py

@@ -13,11 +13,12 @@ def main():
     return {"code": 200,"msg":"Success"}
     
 @app.get("/chrome")
-def chrome(url:str=None,wait:int=5,header:str=None):
+def chrome(url:str=None,wait:int=5,header:str=None,cookie:str=None):
     
     target_url = url
     wait_time = wait
     header_list = header
+    cookie_string = cookie
 
     if target_url == None:
         return {"code": 500,"msg":"No target URL"}
@@ -25,24 +26,28 @@ def chrome(url:str=None,wait:int=5,header:str=None):
     if wait_time not in range(0, 31):
         return {"code": 500,"msg":"The waiting time must be between 0 and 30"}
 
-    options = Options()
-    options.add_argument('--headless')
+    header_array = {}
     
-    driver = webdriver.Chrome(options=options)
-
     try:
         if header_list == None:
             pass
         else:
             # print(unquote(header_list))
-            header_array = json.loads(unquote(header_list))
-            driver.header_overrides = header_array
-    except ValueError:
-        driver.quit()
-        return {"code": 500,"msg":"The header field is not JSON"}
+            header_array.update(json.loads(unquote(header_list)))
     except Exception as e:
-        driver.quit()
-        return {"code": 500,"msg":"Unable to overwrite header"}
+        return {"code": 500,"msg":"The header field is not JSON"}
+
+    if cookie_string == None:
+        pass
+    else:
+        header_array.update({"cookie":cookie_string})
+            
+    options = Options()
+    options.add_argument('--headless')
+    
+    driver = webdriver.Chrome(options=options)
+
+    driver.header_overrides = header_array
     
     driver.get(target_url)