def __init__(self, **kw):
super().__init__(**kw)
self._pending_responses = {}
self._parser = frame_parser(kind=2, pyparser=True)
actor = get_actor()
if actor.is_arbiter():
self.bind_event('connection_lost', self._connection_lost)
def parse_frame(self, websocket, frame):
parser = frame_parser(kind=1)
frame = parser.decode(frame)
wsclient = websocket.cache.wsclient
websocket.connection.reset_mock()
msg = _json.loads(frame.body[1:])[0]
return wsclient.protocol.decode(msg)
def parse_frame(self, websocket, frame):
parser = frame_parser(kind=1)
frame = parser.decode(frame)
wsclient = websocket.cache.wsclient
websocket.connection.reset_mock()
msg = _json.loads(frame.body[1:])[0]
return wsclient.protocol.decode(msg)
def handle_101(response, exc=None):
'''Websocket upgrade as ``on_headers`` event.'''
if not exc and response.status_code == 101:
connection = response.connection
request = response._request
handler = request.websocket_handler
parser = frame_parser(kind=1)
if not handler:
handler = WS()
connection.upgrade(partial(WebSocketClient, response, handler, parser))
response.finished()
response.request_again = connection.current_consumer()
def handle_101(response, exc=None):
'''Websocket upgrade as ``on_headers`` event.'''
if response.status_code == 101:
connection = response.connection
request = response._request
handler = request.websocket_handler
parser = frame_parser(kind=1)
if not handler:
handler = WS()
body = response.recv_body()
connection.upgrade(partial(WebSocketClient, response, handler, parser))
response.finished()
consumer = connection.current_consumer()
consumer.data_received(body)
response.request_again = consumer
def handle_101(response, exc=None):
"""Websocket upgrade as ``on_headers`` event."""
if response.status_code == 101:
connection = response.connection
request = response._request
handler = request.websocket_handler
parser = frame_parser(kind=1)
if not handler:
handler = WS()
body = response.recv_body()
connection.upgrade(partial(WebSocketClient, response, handler, parser))
response.finished()
consumer = connection.current_consumer()
consumer.data_received(body)
response.request_again = consumer
def _factory(self, **params):
params['pyparser'] = self.pyparser
return frame_parser(**params)
def parser(self, pyparser=False, **kw):
return frame_parser(**kw)
def parser(self, pyparser=True, **kw):
return frame_parser(pyparser=True, **kw)
def _factory(self, **params):
params['pyparser'] = self.pyparser
return frame_parser(**params)
def parser(self, kind=0):
return frame_parser(pyparser=True, kind=kind)
def parser(self, kind=0):
return frame_parser(kind=kind)
def parser(self, kind=0):
return frame_parser(pyparser=True, kind=kind)
def parser(self, kind=0):
return frame_parser(kind=kind)
def parser(self, pyparser=True, **kw):
return frame_parser(pyparser=True, **kw)
def parser(self, pyparser=False, **kw):
return frame_parser(**kw)