U Mfg@sddlmZddlZddlZddlmZddlmZmZddl m Z m Z m Z ddl mZmZddlmZmZdd lmZmZd d d d gZddddd Zddddd Zdddddd Zdddddd ZdS)) annotationsN)List)HeadersMultipleValuesError) InvalidHeaderInvalidHeaderValueInvalidUpgrade)parse_connection parse_upgrade)ConnectionOptionUpgradeProtocol) accept_key generate_key build_request check_requestbuild_responsecheck_responserstr)headersreturncCs*t}d|d<d|d<||d<d|d<|S)z Build a handshake request to send to the server. Update request headers passed in argument. Args: headers: Handshake request headers. Returns: str: ``key`` that must be passed to :func:`check_response`. websocketUpgrade ConnectionSec-WebSocket-Key13Sec-WebSocket-Version)rrkeyr?/tmp/pip-unpacked-wheel-yzabpfcc/websockets/legacy/handshake.pyrs c Cstdd|dDg}tdd|Ds;sz!check_request..rcss|]}|dkVqdSupgradeNlowerr"rrr >sz check_request.., cSsg|] }t|qSrr r"rrr r%BsrrrrNz,more than one Sec-WebSocket-Key header foundT)validaterz0more than one Sec-WebSocket-Version header foundr)sumget_allanyr joinlenr)KeyErrorrrbase64 b64decodeencodebinasciiErrorr)r connectionr'Zs_w_keyexcZraw_keyZ s_w_versionrrr r&sV     None)rrrcCs d|d<d|d<t||d<dS)z Build a handshake response to send to the client. Update response headers passed in argument. Args: headers: Handshake response headers. key: Returned by :func:`check_request`. rrrSec-WebSocket-AcceptN)acceptrrrr rjs c Cstdd|dDg}tdd|Ds.rcss|]}|dkVqdSr&r(r"rrr r*sz!check_response.. cSsg|] }t|qSrr,r"rrr r%srr-rrr+r>Nz/more than one Sec-WebSocket-Accept header found) r0r1r2r r3r4r)r5rrr?r)rrr;r'Z s_w_acceptr<rrr rzs2  ) __future__rr6r9typingrZdatastructuresrr exceptionsrrr rr r r r utilsrr?r__all__rrrrrrrr s   D