Update app.py

app.py

@@ -26,7 +26,6 @@ def get_root_domain(url):
     else:
         return domain
 
-# 网络抓包内容过滤
 def filter_type(_type: str):
     types = [
         'application/javascript', 'application/x-javascript', 'text/css', 'webp', 'image/png', 'image/gif',
@@ -42,10 +41,10 @@ def main():
     
 @app.get("/chrome")
 def chrome(url:str=None,wait:int=5,header:str=None,cookie:str=None):
-    # 设置日志性能监听参数
+    
     caps = {
         "browserName": "chrome",
-        'goog:loggingPrefs': {'performance': 'ALL'}  
+        'goog:loggingPrefs': {'performance': 'ALL'}  # 开启日志性能监听
     }
     
     # 必须有目标url
@@ -79,8 +78,6 @@ def chrome(url:str=None,wait:int=5,header:str=None,cookie:str=None):
 
     # 设置为无头模式
     options.add_argument('--headless')
-
-    # 开启日志性能监听
     for key, value in caps.items():
         options.set_capability(key, value)
 
@@ -126,9 +123,7 @@ def chrome(url:str=None,wait:int=5,header:str=None,cookie:str=None):
 
     # 完全加载完成时，页面是否有发生过 301 302 跳转过
     is_jump = (target_url != current_url)
-
-    network = []
-    # 从日志性能监听中抓取网络包（Network），并进行过滤
+    
     performance_log = driver.get_log('performance')  # 获取名称为 performance 的日志
     for packet in performance_log:
         message = json.loads(packet.get('message')).get('message')  # 获取message的数据
@@ -141,10 +136,9 @@ def chrome(url:str=None,wait:int=5,header:str=None,cookie:str=None):
         url = message.get('params').get('response').get('url')  # 获取 该请求  url
         try:
             resp = driver.execute_cdp_cmd('Network.getResponseBody', {'requestId': requestId})  # selenium调用 cdp
-            request_headers = driver.execute_cdp_cmd('Network.getRequestHeaders', {'requestId': requestId})
-            network.append({"url":url, "type":packet_type})
+            print(f'type: {packet_type} url: {url}')
             print(f'response: {resp}')
-            print(f'request_headers: {request_headers}')
+            print()
         except WebDriverException:  # 忽略异常
             pass
 
@@ -152,14 +146,11 @@ def chrome(url:str=None,wait:int=5,header:str=None,cookie:str=None):
         "url": current_url,
         "page_source": page_source,
         "cookies": cookies,
-        "is_jump": is_jump,
-        "network": network,
+        "is_jump": is_jump
     }
     
     driver.quit()
     return {"code": 200,"data":data}
 
 if __name__ == '__main__':
-    uvicorn.run(app='app:app', host="0.0.0.0", port=7860)
-    
-
+    uvicorn.run(app='app:app', host="0.0.0.0", port=7860)