Wn @ s d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z m Z m
Z
d d l m Z m Z m
Z
m Z m Z m Z m Z m Z m Z m Z m Z y& d d l m Z e j Z e j Z Wn e k
r d Z Yn Xd d l m Z m Z e e Z e j Z e j Z e j Z e j! Z! e j" Z" e e d e j f Z# e e e e e e
d e j$ f Z% e d* Z& e d+ Z' e d, Z( d Z) d Z* d Z+ d Z, d
Z- d Z. d Z/ d Z0 d Z1 d Z2 Gd d d e j3 Z4 e d k r"d Z5 n d d d Z5 Gd d d e j6 Z7 Gd d d e7 Z8 Gd d d e7 Z9 Gd d d e: Z; d d Z< d d! Z= d" d# Z> e
rd$ d% Z? e? Z@ d& d' ZA d e< d e= d e> i ZB d( d) ZC d S)- N)_NonClosingTextIOWrapper text_typePY2) byrefPOINTERc_intc_charc_char_pc_void_p py_object c_ssize_tc_ulongwindll WINFUNCTYPE) pythonapi)LPWSTRLPCWSTRGetCommandLineWCommandLineToArgvW
i s i c @ s e Z d Z d e f d e f d e f d e f d e f d e f d e f d e f d e f d
e f d e f g Z e
r e j d d
e d f d S) Py_bufferbufobjlenitemsizereadonlyndimformatshapestrides
suboffsetsZinternalr Z
smalltabler N) __name__
__module__ __qualname__r
r r r r c_ssize_pZ_fields_r insert r, r, /_winconsole.pyr @ s r Fc C sp t } | r t n t } t t | t | | z! t | j } | j | j SWd t
t | Xd S)N) r PyBUF_WRITABLE PyBUF_SIMPLEPyObject_GetBufferr r r r Z from_addressr PyBuffer_Release)r writabler flagsZ buffer_typer, r, r-
get_bufferX s
r4 c @ s( e Z d Z d d Z d d Z d S)_WindowsConsoleRawIOBasec C s
| | _ d S)N)handle)selfr6 r, r, r- __init__e s z!_WindowsConsoleRawIOBase.__init__c C s t j j | d S)NT)io RawIOBaseisatty)r7 r, r, r- r; h s z _WindowsConsoleRawIOBase.isattyN)r' r( r) r8 r; r, r, r, r- r5 c s r5 c @ s( e Z d Z d d Z d d Z d S)_WindowsConsoleReaderc C s d S)NTr, )r7 r, r, r- readableo s z _WindowsConsoleReader.readablec C s t | } | s d S| d r, t d t | d d } | d } t } t | j | | t | d } t t k r t j
d | s t d t | d t k r d Sd | j
S)Nr r z>cannot read odd number of bytes from UTF-16-LE encoded consoler2 Tg?zWindows error: %s)r
ValueErrorr4 r
ReadConsoleWr6 r GetLastErrorERROR_OPERATION_ABORTEDtimesleepOSErrorEOFvalue)r7 bZbytes_to_be_readbufferZcode_units_to_be_readZcode_units_readrvr, r, r- readintor s"
z _WindowsConsoleReader.readintoN)r' r( r) r= rJ r, r, r, r- r< m s r< c @ s: e Z d Z d d Z e d d Z d d Z d S)_WindowsConsoleWriterc C s d S)NTr, )r7 r, r, r- r2 s z _WindowsConsoleWriter.writablec C s( | t k r d S| t k r d Sd | S)N
ERROR_SUCCESSERROR_NOT_ENOUGH_MEMORYzWindows error %s)rL rM )errnor, r, r- _get_error_message s
z(_WindowsConsoleWriter._get_error_messagec C s t | } t | } t | t d } t } t | j | | t | d d | j } | d k r | d k r t | j
t | S)Nr r ) r r4 minMAX_BYTES_WRITTENr
WriteConsoleWr6 r rF rD rO r@ )r7 rG Zbytes_to_be_writtenr Zcode_units_to_be_writtenZcode_units_writtenZ
bytes_writtenr, r, r- write s
z_WindowsConsoleWriter.writeN)r' r( r) r2 staticmethodrO rS r, r, r, r- rK s rK c @ sj e Z d Z d d Z e d d Z d d Z d d Z d d
Z d d Z d
d Z
d S)
ConsoleStreamc C s | | _ | | _ d S)N) _text_streamrH )r7 text_streamZ byte_streamr, r, r- r8 s zConsoleStream.__init__c C s
| j j S)N)rH name)r7 r, r, r- rX s zConsoleStream.namec C sR t | t r | j j | Sy | j Wn t k
rA Yn X| j j | S)N)
isinstancer rV rS flush ExceptionrH )r7 xr, r, r- rS s
zConsoleStream.writec C s" x | D] } | j | q Wd S)N)rS )r7 linesliner, r, r-
writelines s
zConsoleStream.writelinesc C s t | j | S)N)getattrrV )r7 rX r, r, r- __getattr__ s zConsoleStream.__getattr__c C s
| j j S)N)rH r; )r7 r, r, r- r; s zConsoleStream.isattyc C s d | j | j f S)Nz#<ConsoleStream name=%r encoding=%r>)rX encoding)r7 r, r, r- __repr__ s zConsoleStream.__repr__N) r' r( r) r8 propertyrX rS r_ ra r; rc r, r, r, r- rU s rU c C s4 t t j t t d d d d } t | | S)Nz utf-16-lestrictline_bufferingT)r r9 BufferedReaderr< STDIN_HANDLErU )
buffer_streamrW r, r, r- _get_text_stdin s rj c C s+ t t t d d d d } t | | S)Nz utf-16-lere rf T)r rK
STDOUT_HANDLErU )ri rW r, r, r- _get_text_stdout s rl c C s+ t t t d d d d } t | | S)Nz utf-16-lere rf T)r rK
STDERR_HANDLErU )ri rW r, r, r- _get_text_stderr s rn c C s# t j d j t j d d S)N r )zlibcrc32joinsysargvr, r, r, r-
_hash_py_argv s ru c s t d } t t t | f d d t d | j D } t t d s | d d } x] t | d k r | d } | j d s | d k r P| d d } | j d rk Pqk W| d d S)
Nr c s g | ] } | q Sr, r, ).0i) argv_unicoder, r-
<listcomp> s z%_get_windows_argv.<locals>.<listcomp>frozenr --c-m)r| r} )
r r r r rangerF hasattrrs r
startswith)Zargcrt argr, )rx r- _get_windows_argv s %
r c C s t d k r | d k r | d k r t | d r | j r t j | j } | d k r t s t | d } | d k r d Sn t j | j t
j | | Sd S)N utf-16-lere r; rH )r N)zstrictN) r4 r r; _stream_factoriesgetfilenor r` msvcrtZsetmodeosO_BINARY)frb errorsfuncr, r, r- _get_windows_console_stream s r iii)Dr9 r rs rp rB Zctypesr Z
click._compatr r r r r r r r r
r r r
r r r r0 r1 ImportErrorZctypes.wintypesr r r* Zkernel32Z GetStdHandler? rR r@ r Zshell32r rh rk rm r/ r. rL rM rA Z STDIN_FILENOZ
STDOUT_FILENOZ
STDERR_FILENOrE rQ Z Structurer r4 r: r5 r< rK objectrU rj rl rn ru Z_initial_argv_hashr r r r, r, r, r- <module> st L
$