Update app.py

app.py

@@ -14,34 +14,32 @@ def main():
     
 @app.get("/chrome")
 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:
+    # 必须有目标url
+    if type(url) == str:
+        target_url = unquote(url)
+    else:
         return {"code": 500,"msg":"No target URL"}
 
-    if wait_time not in range(0, 31):
+    # 等待时间必须在 0 到 30 之间
+    if wait_time in range(0, 31):
+        wait_time = wait
+    else:
         return {"code": 500,"msg":"The waiting time must be between 0 and 30"}
 
     header_array = {}
-    
+
+    # header可以覆写，但必须传入json
     try:
-        if header_list == None:
-            pass
-        else:
-            # print(unquote(header_list))
-            header_array.update(json.loads(unquote(header_list)))
+        if type(header) == str:
+            header_array.update(json.loads(unquote(header)))
     except Exception as e:
         return {"code": 500,"msg":"The header field is not JSON"}
+        
+    # 如果输入了cookie
+    if type(cookie) == str:
+        header_array.update({"cookie":unquote(cookie)})
 
-    if cookie_string == None:
-        pass
-    else:
-        header_array.update({"cookie":cookie_string})
-            
     options = Options()
     options.add_argument('--headless')