U g@sddlmZddlZddlZddlZddlZddlZz ddlZWnek rTYnXddl m Z m Z ddl m Z ejdkrddd d Zd dd d dZd dd ddZddddddZddddZedkredS)) annotationsN)ClientConnectionconnect)versionwin32None)returncCsddl}|d}|d}d}|jj|}||kr>td|}|jj|||dkrhtdt |t j }|||B}|jj ||dkrtddS) 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_uintwindllkernel32Z GetStdHandle RuntimeErrorZGetConsoleModebyrefint from_bytessys byteorderZSetConsoleMode)r ZSTD_OUTPUT_HANDLEZINVALID_HANDLE_VALUEZ"ENABLE_VIRTUAL_TERMINAL_PROCESSINGhandleZcur_modeZ py_int_modeZnew_moder7/tmp/pip-unpacked-wheel-dx_q7dq3/websockets/__main__.pywin_enable_vt100s  rstr)stringr cCs"tjd|dtjdS)Nz 7 z 8rstdoutwriteflushrrrrprint_during_input3s r!cCs"tjd|dtjdS)Nz  rr rrrprint_over_inputGs r#rzthreading.Event) websocketstopr cCsf|D],}t|tr td|qtd|q|sbtjdkrLtj}ntj }t t |dS)Nz< z < (binary) r) isinstancerr!hexis_setrplatformsignalZ CTRL_C_EVENTSIGINToskillgetpid)r$r%messageZctrl_crrrprint_incoming_messagesSs  r0c Cstjdddd}|}|jddd|jdd d d |}|jrVtd tdS|jdkrj| d t j dkrz t Wn@t k r}z"t jd|dt jW5d}~XYnXzt|j}WnFtk r}z&td|jd|dt dW5d}~XYnXtd|jdt}tjt||fd}|ztd}||qLWn2ttfk r||tdYnX| dS)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"zFailed to connect to z: .rz Connected to )targetargsz> zConnection closed.)!argparseArgumentParseradd_mutually_exclusive_group add_argument parse_argsrprintwebsockets_versionr6errorrr)rrstderrrrr Exceptionexit threadingEventThreadr0startinputsendKeyboardInterruptEOFErrorsetcloser#join)parsergroupr<excr$r%threadr/rrrmaincsN     rW__main__) __future__rr=r,r*rrHreadline ImportErrorZ sync.clientrrrrCr)rr!r#r0rW__name__rrrrs&     ;