Update app.py

app.py

@@ -17,41 +17,44 @@ def main():
     wait_time = 10
     header_list = '{"name": "John", "age": 30, "User-Agent": "New York"}'
 
-    options = Options()
-    options.add_argument('--headless')
-
-    header_array = json.loads(header_list)
-    for key, value in header_array.items():
-        options.add_argument(f'--header="{key}: {value}"')
+    try:
+        options = Options()
+        options.add_argument('--headless')
+        driver = webdriver.Chrome(options=options)
+
+        header_array = json.loads(header_list)
+        for key, value in header_array.items():
+            driver.add_header(str(key), str(value))
+            
+        driver.get(target_url)
+
+        print(driver.page_source)
         
-    driver = webdriver.Chrome(options=options)
-    driver.get(target_url)
-
-    print(driver.page_source)
-    
-    time.sleep(wait_time)
-
-    # 获取当前URL
-    current_url = driver.current_url
-    
-    # 获取页面源代码
-    page_source = driver.page_source
+        time.sleep(wait_time)
     
-    # 获取cookie
-    cookies = driver.get_cookies()
-
-    # 是否有跳转过
-    is_jump = (target_url != current_url)
-    
-    data = {
-        "url": current_url,
-        "page_source": page_source,
-        "cookies": cookies,
-        "is_jump": is_jump
-    }
+        # 获取当前URL
+        current_url = driver.current_url
+        
+        # 获取页面源代码
+        page_source = driver.page_source
+        
+        # 获取cookie
+        cookies = driver.get_cookies()
     
-    driver.quit()
-    return {"code": 200,"data":data}
+        # 是否有跳转过
+        is_jump = (target_url != current_url)
+        
+        data = {
+            "url": current_url,
+            "page_source": page_source,
+            "cookies": cookies,
+            "is_jump": is_jump
+        }
+        
+        driver.quit()
+        return {"code": 200,"data":data}
+    except Exception as e:
+        return {"code": 500,"msg":e}
 
 if __name__ == '__main__':
     uvicorn.run(app='app:app', host="0.0.0.0", port=7860)