B ]bd@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 ConnectionzSec-WebSocket-Key13zSec-WebSocket-Version)r)rkeyr:lib/python3.7/site-packages/websockets/legacy/handshake.pyrs c Cstdd|dDg}tdd|Ds;sz!check_request..rcss|]}|dkVqdS)upgradeN)lower)rrrrr >sz check_request..z, cSsg|] }t|qSr)r )rrrrrr BsrrrzSec-WebSocket-KeyNz,more than one Sec-WebSocket-Key header foundT)ZvalidatezSec-WebSocket-Versionz0more than one Sec-WebSocket-Version header foundr)sumget_allanyr joinlenr"KeyErrorrrbase64Z b64decodeencodebinasciiErrorr)r connectionr!Zs_w_keyexcZraw_keyZ s_w_versionrrrr&sB     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`. rrrzSec-WebSocket-AcceptN)accept)rrrrrrjs c Cstdd|dDg}tdd|Ds.rcss|]}|dkVqdS)r!N)r")rrrrrr#sz!check_response.. cSsg|] }t|qSr)r )rrrrrr srr$rrz, zSec-WebSocket-AcceptNz/more than one Sec-WebSocket-Accept header found) r&r'r(r r)r*r"r+rrr3r)rrr0r!Z s_w_acceptr1rrrrzs$  )Z __future__rr,r.typingrZdatastructuresrr exceptionsrrr rr r r r Zutilsrr3r__all__rrrrrrrrs   D