Update app.py

app.py

@@ -12,22 +12,37 @@ app = FastAPI()
 
 @app.get("/")
 def main():
+    target_url = 'https://test5.container-z.art/test26'
+    wait_time = 10
+    
     options = Options()
     options.add_argument('--headless')
     driver = webdriver.Chrome(options=options)
-    driver.get('https://test5.container-z.art/test26')
+    driver.get(target_url)
+
+    time.sleep(wait_time)
 
-    # WebDriverWait(driver, 10).until(EC.presence_of_element_located((By.TAG_NAME, "body")))
-    time.sleep(10)
+    # 获取当前URL
+    current_url = driver.current_url
     
+    # 获取页面源代码
+    page_source = driver.page_source
+    
+    # 获取cookie
     cookies = driver.get_cookies()
-    print(cookies)
+
+    # 是否有跳转过
+    is_jump = (target_url != current_url)
     
-    current_url = driver.current_url
-    print(current_url)
+    data = {
+        "url": current_url,
+        "page_source": page_source,
+        "cookies": cookies,
+        "is_jump": is_jump
+    }
     
     driver.quit()
-    return {"code": 200,"msg":"Success"}
+    return {"code": 200,"data":data}
 
 if __name__ == '__main__':
     uvicorn.run(app='app:app', host="0.0.0.0", port=7860)