U
Wn @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZm Z m
Z
d dlm Z m Z m
Z
mZmZmZmZmZmZmZmZ z d dlmZ ejZejZW n ek
r dZY nX d dlmZm Z e eZ ej Z e j Z e j Z e j!Z!e j"Z"eedej fZ#ee ee e e
dej$fZ%e dZ&e d Z'e d
Z(d Z)d Z*d Z+d Z,d
Z-d Z.d Z/dZ0dZ1dZ2G dd dej3Z4edkr~dZ5n
d+ddZ5G dd de j6Z7G dd de7Z8G dd de7Z9G d d d e:Z;d d Z<d d! Z=d"d# Z>e
rd$d% Z?e? Z@d&d' ZAe<e=e>d(ZBd)d* ZCdS ), N)_NonClosingTextIOWrapper text_typePY2) byrefPOINTERc_intc_charc_char_pc_void_p py_object c_ssize_tc_ulongwindll WINFUNCTYPE) pythonapi)LPWSTRLPCWSTRGetCommandLineWCommandLineToArgvWiii i i c @ sj e Zd Zdefdefdefdefdefdefdefdefd efd
efd efg Z e
rfe d d
ed f dS ) Py_bufferbufobjlenitemsizereadonlyndimformatshapestrides
suboffsetsZinternalZ
smalltabler N) __name__
__module__ __qualname__r
r r r r c_ssize_pZ_fields_r insert r* r* /_winconsole.pyr @ s r Fc C sT t }|rtnt}tt| t|| zt|j }| |j
W S tt| X d S N) r PyBUF_WRITABLE PyBUF_SIMPLEPyObject_GetBufferr r PyBuffer_Releaser r Z from_addressr )r writabler flagsZ buffer_typer* r* r+
get_bufferX s
r3 c @ s e Zd Zdd Zdd ZdS )_WindowsConsoleRawIOBasec C s
|| _ d S r, )handle)selfr5 r* r* r+ __init__e s z!_WindowsConsoleRawIOBase.__init__c C s t j| dS NT)io RawIOBaseisattyr6 r* r* r+ r; h s z _WindowsConsoleRawIOBase.isattyN)r% r& r' r7 r; r* r* r* r+ r4 c s r4 c @ s e Zd Zdd Zdd ZdS )_WindowsConsoleReaderc C s dS r8 r* r< r* r* r+ readableo s z _WindowsConsoleReader.readablec C s t |}|sdS |d r tdt|dd}|d }t }t| j||t|d }t tkrdt
d |svt dt |d t krdS d|j
S )Nr r z>cannot read odd number of bytes from UTF-16-LE encoded consoleT)r1 g?zWindows error: %s)r
ValueErrorr3 r
ReadConsoleWr5 r GetLastErrorERROR_OPERATION_ABORTEDtimesleepOSErrorEOFvalue)r6 bZbytes_to_be_readbufferZcode_units_to_be_readZcode_units_readrvr* r* r+ readintor s&
z _WindowsConsoleReader.readintoN)r% r& r' r> rK r* r* r* r+ r= m s r= c @ s( e Zd Zdd Zedd Zdd ZdS )_WindowsConsoleWriterc C s dS r8 r* r< r* r* r+ r1 s z _WindowsConsoleWriter.writablec C s | t kr dS | tkrdS d| S )N
ERROR_SUCCESSERROR_NOT_ENOUGH_MEMORYzWindows error %s)rM rN )errnor* r* r+ _get_error_message s
z(_WindowsConsoleWriter._get_error_messagec C sh t |}t|}t|td }t }t| j||t|d d|j }|dkrd|dkrdt |
t |S )Nr r ) r r3 minMAX_BYTES_WRITTENr
WriteConsoleWr5 r rG rE rP rA )r6 rH Zbytes_to_be_writtenr Zcode_units_to_be_writtenZcode_units_writtenZ
bytes_writtenr* r* r+ write s
z_WindowsConsoleWriter.writeN)r% r& r' r1 staticmethodrP rT r* r* r* r+ rL s
rL c @ sH e Zd Zdd Zedd Zdd Zdd Zd d
Zd d Z d
d Z
dS )
ConsoleStreamc C s || _ || _d S r, ) _text_streamrI )r6 text_streamZ byte_streamr* r* r+ r7 s zConsoleStream.__init__c C s | j jS r, )rI namer< r* r* r+ rY s zConsoleStream.namec C sD t |tr| j|S z | W n tk
r6 Y nX | j|S r, )
isinstancer rW rT flush ExceptionrI )r6 xr* r* r+ rT s
zConsoleStream.writec C s |D ]}| | qd S r, )rT )r6 linesliner* r* r+
writelines s zConsoleStream.writelinesc C s t | j|S r, )getattrrW )r6 rY r* r* r+ __getattr__ s zConsoleStream.__getattr__c C s
| j S r, )rI r; r< r* r* r+ r; s zConsoleStream.isattyc C s d| j | jf S )Nz#<ConsoleStream name=%r encoding=%r>)rY encodingr< r* r* r+ __repr__ s zConsoleStream.__repr__N) r% r& r' r7 propertyrY rT r` rb r; rd r* r* r* r+ rV s
rV c C s$ t tttdddd}t|| S N utf-16-lestrictT)line_buffering)r r9 BufferedReaderr= STDIN_HANDLErV Z
buffer_streamrX r* r* r+ _get_text_stdin s rm c C s t ttdddd}t|| S rf )r rL
STDOUT_HANDLErV rl r* r* r+ _get_text_stdout s ro c C s t ttdddd}t|| S rf )r rL
STDERR_HANDLErV rl r* r* r+ _get_text_stderr s rq c C s t dtjdd S )N r )zlibZcrc32joinsysargvr* r* r* r+
_hash_py_argv s rw c s t d} tt t| fddtd| jD }ttds|dd }t|dkr|d }| dr|dkrpq|dd }| drHqqH|dd S )Nr c s g | ] } | qS r* r* ).0iZ argv_unicoder* r+
<listcomp> s z%_get_windows_argv.<locals>.<listcomp>frozenr -)z-cz-m)
r r r r rangerG hasattrru r
startswith)Zargcrv argr* rz r+ _get_windows_argv s
r )r r r c C sz t d k rv|dkrv|dkrvt| drv| rvt| }|d k rvts\t| d} | d krnd S nt | t
j || S d S )N)rg N)rh Nr; rI ) r3 r r; _stream_factoriesgetfilenor ra msvcrtZsetmodeosZO_BINARY)frc errorsfuncr* r* r+ _get_windows_console_stream s"
r )F)Dr9 r ru rs rC Zctypesr Z
click._compatr r r r r r r r r
r r r
r r r r/ r0 ImportErrorZctypes.wintypesr r r( Zkernel32Z GetStdHandler@ rS rA r Zshell32r rk rn rp r. r- rM rN rB Z STDIN_FILENOZ
STDOUT_FILENOZ
STDERR_FILENOrF rR Z Structurer r3 r: r4 r= rL objectrV rm ro rq rw Z_initial_argv_hashr r r r* r* r* r+ <module> s 4
$