o Yii@sddlmZddlZddlZddlZddlmZddlm Z ddl m Z m Z ddl mZddlmZmZgd ZejGd d d ZdddZdddZefdddZdS)) annotationsN)Headers) InvalidProxy)build_authorization_basic build_host) USER_AGENT)DELIMS WebSocketURI) get_proxy parse_proxyProxyc@sPeZdZUdZded<ded<ded<dZded <dZded <edd dZdS)r a Proxy address. Attributes: scheme: ``"socks5h"``, ``"socks5"``, ``"socks4a"``, ``"socks4"``, ``"https"``, or ``"http"``. host: Normalized to lower case. port: Always set even if it's the default. username: Available when the proxy address contains `User Information`_. password: Available when the proxy address contains `User Information`_. .. _User Information: https://datatracker.ietf.org/doc/html/rfc3986#section-3.2.1 strschemehostintportN str | Noneusernamepasswordreturntuple[str, str] | NonecCs(|jdurdS|jdusJ|j|jfS)N)rr)selfrJ/home/kim/smarthome/.venv/lib/python3.10/site-packages/websockets/proxy.py user_info(s  zProxy.user_info)rr) __name__ __module__ __qualname____doc____annotations__rrpropertyrrrrrr s   r proxyrrcCs8tj|}|jdvrt|d|jd|jdurt|d|jdvr)t|d|jdkr3t|d |jdkr=t|d |j}|j}|j pN|jd krMd nd }|j }|j }|durb|durbt|dz| dWn*t y| d}|dur|dusJtjj|td}tjj|td}Ynwt|||||S)z Parse and validate a proxy. Args: proxy: proxy. Returns: Parsed proxy. Raises: InvalidProxy: If ``proxy`` isn't a valid proxy. )Zsocks5hZsocks5Zsocks4aZsocks4httpshttpzscheme z isn't supportedNzhostname isn't provided)/zpath is meaninglessr%zquery is meaninglesszfragment is meaninglessr#iPz"username provided without passwordasciiidna)safe)urllibparseurlparserrhostnamepathqueryfragmentrrrencodeUnicodeEncodeErrordecodequoter r )r"parsedrrrrrrrrr 0s:              r urir rcCstj|jd|jrdStj}|jrgd}ngd}|D] }||}|durB|dkr>|dr>d|dd}|Sq"dS) zU Return the proxy to use for connecting to the given WebSocket URI, if any. :N)wsssocksr#)wsr:r#r$r:zhttp://z socks5h://) r+request proxy_bypassrr getproxiessecureget startswith)r7proxiesschemesrr"rrrr bs   r ws_uriuser_agent_headerbytescCst|j|j|jdd}t}t|j|j|j|d<|dur!||d<|jdur6|jdus-Jt|j|j|d<d|d| S)NT)Zalways_include_portHostz User-AgentzProxy-AuthorizationzCONNECT z HTTP/1.1 ) rrrr@rrrrr2 serialize)r"rErFrheadersrrrprepare_connect_requests rK)r"rrr )r7r rr)r"r rEr rFrrrG) __future__r dataclasses urllib.parser+urllib.requestZdatastructuresr exceptionsrrJrrZhttp11rr7r r __all__ dataclassr r r rKrrrrs      2'