�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
r�d$ d% � Z? e? � Z@ d& d' � ZA d e< d e= d e> i ZB d( d) � ZC d S)-� N)�_NonClosingTextIOWrapper� text_type�PY2) �byref�POINTER�c_int�c_char�c_char_p�c_void_p� py_object� c_ssize_t�c_ulong�windll� WINFUNCTYPE)� pythonapi)�LPWSTR�LPCWSTR�GetCommandLineW�CommandLineToArgvW�
� � � � 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_buffer�buf�obj�len�itemsize�readonly�ndim�format�shape�strides�
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_SIMPLE�PyObject_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� RawIOBase�isatty)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 � GetLastError�ERROR_OPERATION_ABORTED�time�sleep�OSError�EOF�value)r7 �bZbytes_to_be_read�bufferZcode_units_to_be_readZcode_units_read�rvr, 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_SUCCESS�ERROR_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 �min�MAX_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 �lines�liner, 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-le�strict�line_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 )�zlib�crc32�join�sys�argvr, 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, )�.0�i)� 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_factories�get�filenor r` �msvcrtZsetmode�os�O_BINARY)�frb �errors�funcr, r, r- �_get_windows_console_stream s r� i����i����i�)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
$