跳到主要内容

WebSocketRoute

每当通过 page.route_web_socket() 或 browser_context.route_web_socket() 设置 WebSocket 路由时，WebSocketRoute 对象允许像实际服务器那样处理 WebSocket 连接。

模拟

默认情况下，路由的 WebSocket 不会连接到服务器。通过这种方式，你可以模拟整个 WebSocket 通信过程。下面是一个响应 "request" 并发送 "response" 的示例。

def message_handler(ws: WebSocketRoute, message: Union[str, bytes]):
  if message == "request":
    ws.send("response")

page.route_web_socket("wss://example.com/ws", lambda ws: ws.on_message(
    lambda message: message_handler(ws, message)
))

由于我们没有在 WebSocket 路由处理程序中调用 web_socket_route.connect_to_server，Playwright 会假定 WebSocket 将被模拟，并自动在页面内部建立 WebSocket 连接。

这是另一个处理 JSON 消息的示例

def message_handler(ws: WebSocketRoute, message: Union[str, bytes]):
  json_message = json.loads(message)
  if json_message["request"] == "question":
    ws.send(json.dumps({ "response": "answer" }))

page.route_web_socket("wss://example.com/ws", lambda ws: ws.on_message(
    lambda message: message_handler(ws, message)
))

拦截

或者，你可能希望连接到实际服务器，但在中间拦截消息，并修改或阻止它们。调用 web_socket_route.connect_to_server 会返回一个服务器端的 WebSocketRoute 实例，你可以用它向服务器发送消息或处理来自服务器的消息。

下面是一个修改页面发送到服务器的某些消息的示例。来自服务器发送到页面的消息保持不变，这依赖于默认转发。

def message_handler(server: WebSocketRoute, message: Union[str, bytes]):
  if message == "request":
    server.send("request2")
  else:
    server.send(message)

def handler(ws: WebSocketRoute):
  server = ws.connect_to_server()
  ws.on_message(lambda message: message_handler(server, message))

page.route_web_socket("/ws", handler)

连接到服务器后，页面和服务器之间的所有**消息默认会自动转发**。

然而，如果你在原始路由上调用 web_socket_route.on_message()，从页面发送到服务器的消息将不再**自动转发**，而是应该由 handler 处理。

类似地，在服务器端 WebSocket 上调用 web_socket_route.on_message() 将**停止自动转发**从服务器到页面的消息，并且 handler 应该负责处理它们。

以下示例阻止了双向的某些消息。由于它在双向都调用了 web_socket_route.on_message()，所以完全没有自动转发。

def ws_message_handler(server: WebSocketRoute, message: Union[str, bytes]):
  if message != "blocked-from-the-page":
    server.send(message)

def server_message_handler(ws: WebSocketRoute, message: Union[str, bytes]):
  if message != "blocked-from-the-server":
    ws.send(message)

def handler(ws: WebSocketRoute):
  server = ws.connect_to_server()
  ws.on_message(lambda message: ws_message_handler(server, message))
  server.on_message(lambda message: server_message_handler(ws, message))

page.route_web_socket("/ws", handler)

方法​

close​

添加于: v1.48 webSocketRoute.close

关闭 WebSocket 连接的其中一端。

用法

web_socket_route.close()
web_socket_route.close(**kwargs)

参数

返回值

on_close​

添加于: v1.48 webSocketRoute.on_close

允许处理 WebSocket 关闭事件 (WebSocket.close)。

默认情况下，关闭连接的一侧（无论是在页面中还是在服务器上）都会关闭另一侧。然而，当设置了 web_socket_route.on_close() 处理程序时，默认的关闭转发功能被禁用，处理程序应该负责处理。

用法

web_socket_route.on_close(handler)

参数

on_message​

添加于: v1.48 webSocketRoute.on_message

此方法允许处理由 WebSocket 发送的消息，无论消息来自页面还是服务器。

当在原始 WebSocket 路由上调用时，此方法处理从页面发送的消息。你可以通过使用 web_socket_route.send() 响应这些消息，或将它们转发到 web_socket_route.connect_to_server 返回的服务器端连接，或执行其他操作来处理这些消息。

一旦调用此方法，消息不会再自动转发到服务器或页面 - 你应该通过调用 web_socket_route.send() 手动进行转发。详情请参阅顶部的示例。

再次调用此方法将用新的处理程序覆盖现有的。

用法

web_socket_route.on_message(handler)

参数

send​

添加于: v1.48 webSocketRoute.send

向 WebSocket 发送消息。当在原始 WebSocket 上调用时，将消息发送到页面。当在 web_socket_route.connect_to_server 的结果上调用时，将消息发送到服务器。详情请参阅顶部的示例。

用法

web_socket_route.send(message)

参数

属性​

connect_to_server​

添加于: v1.48 webSocketRoute.connect_to_server

默认情况下，路由的 WebSocket 不连接到服务器，因此你可以模拟整个 WebSocket 通信。此方法连接到实际的 WebSocket 服务器，并返回服务器端的 WebSocketRoute 实例，从而能够发送和接收来自服务器的消息。

连接到服务器后

详情请参阅顶部的示例。

用法

web_socket_route.connect_to_server

返回值

url​

添加于: v1.48 webSocketRoute.url

在页面中创建的 WebSocket 的 URL。

用法

web_socket_route.url

返回值