Update app.py

app.py

@@ -1,4 +1,3 @@
-# from selenium import webdriver
 from seleniumwire import webdriver
 from selenium.webdriver.chrome.options import Options
 from fastapi import FastAPI, Request
@@ -8,27 +7,37 @@ import json
 
 app = FastAPI()
 
-def convert_cookies_to_dict(cookies):
-    cookies = dict([l.split("=", 1) for l in cookies.split("; ")])
-    return cookies
-    
 @app.get("/")
 def main():
-    target_url = 'https://test5.container-z.art/test26'
-    wait_time = 10
-    header_list = '{"user-agent": "New York","cookie": "_zap=386b2646-cd4f-48b0-831c-0f4e0fcb84d; d_c0=ATBVpddW2hePTs-uWJUqnuikOgCLTtzDaQ=|1702569488; _xsrf=9Exea3EejXTsAZewjRCns4TPklsjdCpV; __zse_ck=001_xCDDL0hUsjpAaMgmQE7pCfeU+zXi0/T+YTGbyY=OkkWy54lHQjmO1If3BOBMUzDjeY=d26WyAD5j0BG=m7jR7RrpRqv2ImjXDfJPzUEzARb4/TImweWRybZ5aGbVke4B; BEC=9f6899d2c0337126a73fd6e01ed7c740; gdxidpyhxdE=tfBS35OO12TxVYkb6i%5Ce7mQekmAMzv2e1Ah%5CGS4h%5CwU7xS47p9%2Fd4T%5CNEmx7s%2Bu%2Fr8mkLktUMVIPQ404Z1OxrejYyRQ3aSgLL7CBsT2nRhTXpzWaxqr7KyuaHSCMZjsK6OBaCM5GsMTSor4baQvmzWfo5SZqccCgWZTrn4Ii2tgDNH0w%3A1718799277544"}'
-
-    header_array = json.loads(header_list)
-    # cookies = convert_cookies_to_dict(cookies_string)
+    return {"code": 200,"msg":"Success"}
+    
+@app.get("/chrome")
+def chrome(url:str=None,wait:int=5,header:str=None):
     
+    target_url = url
+    wait_time = wait
+    header_list = header
+
+    if target_url == None:
+        return {"code": 500,"msg":"No target URL"}
+
     options = Options()
     options.add_argument('--headless')
-    # for key, value in header_array.items():
-    #     options.add_argument(f'{key}="{value}"')
     
     driver = webdriver.Chrome(options=options)
 
-    driver.header_overrides = header_array
+    try:
+        if header_list == None:
+            pass
+        else:
+            header_array = json.loads(header_list)
+            driver.header_overrides = header_array
+    except ValueError:
+        driver.quit()
+        return {"code": 500,"msg":"The header field is not JSON"}
+    except Exception as e:
+        driver.quit()
+        return {"code": 500,"msg":"Unable to overwrite header"}
     
     driver.get(target_url)