From f5b3445968f2be3f13c7f92fe9f969f1a242a2b7 Mon Sep 17 00:00:00 2001 From: hi-T0day Date: Mon, 18 Mar 2019 20:26:39 +0800 Subject: [PATCH] fixing bugs MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 对于自定义的头，代码存在bug，订正 --- aiowebsocket/handshakes.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/aiowebsocket/handshakes.py b/aiowebsocket/handshakes.py index 0b92136..2766bdb 100644 --- a/aiowebsocket/handshakes.py +++ b/aiowebsocket/handshakes.py @@ -36,10 +36,14 @@ def shake_headers(self, host: str, port: int, resource: str = '/', if self.headers: # Allow the use of custom header if isinstance(self.headers, list): - return '\r\n'.join(self.headers) + '\r\n' + self.headers.insert(0, 'GET {} HTTP/1.1'.format(resource)) + self.headers.append('\r\n') + return '\r\n'.join(self.headers) if isinstance(self.headers, dict): head = ['{}:{}'.format(k, item) for k, item in self.headers.items()] - return '\r\n'.join(head) + '\r\n' + self.headers.insert(0, 'GET {} HTTP/1.1'.format(resource)) + self.headers.append('\r\n') + return '\r\n'.join(self.headers) bytes_key = bytes(random.getrandbits(8) for _ in range(16)) key = base64.b64encode(bytes_key).decode()