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 e�Z ej Z e j Z e j Z e j!Z!e j"Z"ee�dej f�Z#ee e�e e e
��dej$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/dZ0dZ1dZ2G dd� dej3�Z4edk�r~dZ5n
d+dd�Z5G dd� de j6�Z7G dd� de7�Z8G dd� de7�Z9G d d � d e:�Z;d d � Z<d d!� Z=d"d#� Z>e
�r�d$d%� Z?e?� Z@d&d'� ZAe<e=e>d(�ZBd)d*� ZCdS ),� 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�CommandLineToArgvWi����i����i�� � i� � � i� c @ sj e Zd Zdefdefdefdefdefdefdefdefd efd
efd efg Z e
rfe � d d
ed f� dS )� Py_buffer�buf�obj�len�itemsize�readonly�ndim�format�shape�strides�
suboffsetsZinternal���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 sT t � }|rtnt}tt| �t|�|� zt|j }|� |j
�W �S tt|�� X d S �N) r �PyBUF_WRITABLE� PyBUF_SIMPLE�PyObject_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� RawIOBase�isatty�r6 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 td��t|dd�}|d }t� }t| j||t|�d �}t� tkrdt �
d� |svt dt� ��|d t kr�dS 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 � GetLastError�ERROR_OPERATION_ABORTED�time�sleep�OSError�EOF�value)r6 �bZbytes_to_be_read�bufferZcode_units_to_be_readZcode_units_read�rvr* 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_SUCCESS�ERROR_NOT_ENOUGH_MEMORYzWindows error %s)rM rN )�errnor* r* r+ �_get_error_message� s
z(_WindowsConsoleWriter._get_error_messagec C sh t |�}t|�}t|t�d }t� }t| j||t|�d � d|j }|dkrd|dkrdt | �
t � ���|S )Nr r ) r r3 �min�MAX_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 |t�r| 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 �lines�liner* 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 t�tt��dddd�}t|| �S �N� utf-16-le�strictT)�line_buffering)r r9 �BufferedReaderr= � STDIN_HANDLErV �Z
buffer_streamrX r* r* r+ �_get_text_stdin� s �rm c C s t tt�dddd�}t|| �S rf )r rL �
STDOUT_HANDLErV rl r* r* r+ �_get_text_stdout� s �ro c C s t tt�dddd�}t|| �S rf )r rL �
STDERR_HANDLErV rl r* r* r+ �_get_text_stderr� s �rq c C s t �d�tjdd � ��S )N� r )�zlibZcrc32�join�sys�argvr* r* r* r+ �
_hash_py_argv� s rw c s� t d�} tt� t| ��� � fdd�td| j�D �}ttd�s�|dd � }t|�dkr�|d }|� d�r�|dkrpq�|dd � }|� d�rHq�qH|dd � S )Nr c s g | ] }� | �qS r* r* )�.0�i�Z 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| d�rv| �� 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_factories�get�filenor ra �msvcrtZsetmode�osZO_BINARY)�frc �errors�funcr* 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
� ��
$�