gddlmZddlZddlZddlZddlZddlZ ddlZddl m Z m Z ddl m Z ejdk(rd dZddZdd Zdd Zd d Zed k(reyy#e$rYFwxYw)) annotationsN)ClientConnectionconnect)versionwin32c<ddl}|jd}|jd}d}|jjj |}||k(r t d|j}|jjj ||j|dk(r t dtj|tj}|j||z}|jjj||dk(r t dy) z Enable VT-100 for console output on Windows. See also https://github.com/python/cpython/issues/73245. rNizunable to obtain stdout handlez$unable to query current console modezunable to set console mode) ctypesc_uintwindllkernel32 GetStdHandle RuntimeErrorGetConsoleModebyrefint from_bytessys byteorderSetConsoleMode)r STD_OUTPUT_HANDLEINVALID_HANDLE_VALUE"ENABLE_VIRTUAL_TERMINAL_PROCESSINGhandlecur_mode py_int_modenew_modes H/var/www/openai/venv/lib/python3.12/site-packages/websockets/__main__.pywin_enable_vt100r!s "MM#.%}}R0-2*''445FG ) )?@ @==? == ! ! 0 0h9O PTU UEF FnnXs}}= ==/Q!QR == ! ! 0 0 Ba G;< < Hctjjd|dtjjy)Nz 7 z 8rstdoutwriteflushstrings r print_during_inputr*3s<JJ  (  JJr"ctjjd|dtjjy)Nz  r$r(s r print_over_inputr-Gs9JJ  (-  !JJr"cb|D]=}t|trtd|z"td|jz?|j s]t j dk(rtj}ntj}tjtj|yy)Nz< z < (binary) r) isinstancestrr*hexis_setrplatformsignal CTRL_C_EVENTSIGINToskillgetpid) websocketstopmessagectrl_cs r print_incoming_messagesr>Ss{ gs # tg~ . }w{{}< =  ;;= <<7 "((F]]F  V$ r"cptjddd}|j}|jdd|jdd d |j }|j rt d ty|j|jd tjdk(r t t#|j}t d|jdt)j*}t)j,t.|f}|j1 t3d}|j5|#t$rK}tjjd|dtjj!Yd}~d}~wwxYw#t$$r;}t d|jd|dtj&dYd}~d}~wwxYw#t6t8f$r.|j;|j=t?dYnwxYw|jAy)Nzpython -m websocketszInteractive WebSocket client.F)prog descriptionadd_helpz --version store_true)actionuriz?)metavarnargsz websockets z+the following arguments are required: rzUnable to set terminal to VT100 mode. This is only supported since Win10 anniversary update. Expect weird symbols on the terminal. Error: r,z Connected to .zFailed to connect to z: r)targetargsz> zConnection closed.)!argparseArgumentParseradd_mutually_exclusive_group add_argument parse_argsrprintwebsockets_versionrEerrorrr3r!rstderrr&r'r Exceptionexit threadingEventThreadr>startinputsendKeyboardInterruptEOFErrorsetcloser-join)parsergrouprKexcr:r;threadr<s r mainrfcs  $ $ #3F  / / 1E {<8 ugS9    D || ./01 xx BC ||w   +DHH%  dhhZq)* ?? D  %rf__name__r"r rpsx"  32<<7=<(  % 8v zFe  sAA'&A'