| 66 | |
| 67 | |
| 68 | class WebSocketConnection(ws_adaptor.WebSocketConnection): |
| 69 | |
| 70 | def __init__(self, context: tornado.websocket.WebSocketHandler): |
| 71 | self.context = context |
| 72 | |
| 73 | def get_query_argument(self, name) -> typing.Optional[str]: |
| 74 | return self.context.get_query_argument(name, None) |
| 75 | |
| 76 | def make_session_info(self) -> dict: |
| 77 | session_info = get_session_info_from_headers(self.context.request.headers) |
| 78 | session_info['user_ip'] = self.context.request.remote_ip |
| 79 | session_info['request'] = self.context.request |
| 80 | session_info['backend'] = 'tornado' |
| 81 | session_info['protocol'] = 'websocket' |
| 82 | return session_info |
| 83 | |
| 84 | def write_message(self, message: dict): |
| 85 | self.context.write_message(json.dumps(message)) |
| 86 | |
| 87 | def closed(self) -> bool: |
| 88 | return not bool(self.context.ws_connection) |
| 89 | |
| 90 | def close(self): |
| 91 | self.context.close() |
| 92 | |
| 93 | |
| 94 | def _webio_handler(applications=None, cdn=True, reconnect_timeout=0, check_origin_func=_is_same_site) \ |
no outgoing calls
no test coverage detected
searching dependent graphs…