File: //opt/alt/python37/lib/python3.7/site-packages/click/__pycache__/_winconsole.cpython-37.pyc
B
��f� � @ s� U d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlmZ d dlm Z d dlm
Z
d dlm Z d dlm Z d d lm
Z
d d
lmZ d d lmZ d d lmZ d d
lmZ d dlmZ d dlmZ ddlmZ ejdks�t�d dlZd dlmZ d dlmZ e
e
�Z ej Z e j Z e j Z e j Z e j!Z!e j"Z"ee�dej f�Z#ee
e�ee
e ��dej$f�Z%ee e �dej f�Z&e d�Z'e d�Z(e d�Z)d Z*dZ+d Z,dZ-d Z.d Z/dZ0d Z1d Z2d Z3yd d lm4Z4 W n e5k
�r� dZ6Y n(X G d!d"� d"e�Z7e4j8Z8e4j9Z9d=d$d%�Z6G d&d'� d'e j:�Z;G d(d)� d)e;�Z<G d*d+� d+e;�Z=G d,d-� d-�Z>ej?ej@d.�d/d0�ZAej?ej@d.�d1d2�ZBej?ej@d.�d3d4�ZCeAeBeCd5�ZDejEeFejGej?gej@f f eHd6< ej@eId7�d8d9�ZJej@ejKeL ejKeL ejKej@ d:�d;d<�ZMdS )>� N)�byref)�c_char)�c_char_p)�c_int)� c_ssize_t)�c_ulong)�c_void_p)�POINTER)� py_object)� Structure)�DWORD)�HANDLE)�LPCWSTR)�LPWSTR� )�_NonClosingTextIOWrapper�win32)�windll)� WINFUNCTYPE�GetCommandLineW�CommandLineToArgvW� LocalFreei����i����i�� i� � � i� )� pythonapic @ sR e Zd Zdefdefdefdefdefdefdefdefd efd
efd efg Z d S )
� Py_buffer�buf�obj�len�itemsize�readonly�ndim�format�shape�strides�
suboffsetsZinternalN)
�__name__�
__module__� __qualname__r r
r r r � c_ssize_p�_fields_� r, r, �B/opt/alt/python37/lib/python3.7/site-packages/click/_winconsole.pyr K s r Fc C sP t � }|rtnt}tt| �t|�|� zt|j }|�|j �S t
t|�� X d S )N) r �PyBUF_WRITABLE� PyBUF_SIMPLE�PyObject_GetBufferr
r r r � from_addressr �PyBuffer_Release)r �writabler �flagsZ buffer_typer, r, r- �
get_buffer] s
r5 c s$ e Zd Zdd� Z� fdd�Z� ZS )�_WindowsConsoleRawIOBasec C s
|| _ d S )N)�handle)�selfr7 r, r, r- �__init__j s z!_WindowsConsoleRawIOBase.__init__c s t � �� dS )NT)�super�isatty)r8 )� __class__r, r- r; m s
z _WindowsConsoleRawIOBase.isatty)r' r( r) r9 r; �
__classcell__r, r, )r< r- r6 i s r6 c @ s e Zd Zdd� Zdd� ZdS )�_WindowsConsoleReaderc C s dS )NTr, )r8 r, r, r- �readables s z _WindowsConsoleReader.readablec C s� t |�}|sdS |d r td��t|dd�}|d }t� }tt| j�||t|�d �}t� t krht
� d� |s|t dt� � ���|d t
kr�dS d|j S )Nr r z>cannot read odd number of bytes from UTF-16-LE encoded consoleT)r3 g�������?zWindows error: )r �
ValueErrorr5 r � ReadConsoleWr
r7 r � GetLastError�ERROR_OPERATION_ABORTED�time�sleep�OSError�EOF�value)r8 �bZbytes_to_be_read�bufferZcode_units_to_be_readZcode_units_read�rvr, r, r- �readintov s,
z _WindowsConsoleReader.readintoN)r' r( r) r? rL r, r, r, r- r> r s r> c @ s( e Zd Zdd� Zedd� �Zdd� ZdS )�_WindowsConsoleWriterc C s dS )NTr, )r8 r, r, r- r3 � s z _WindowsConsoleWriter.writablec C s" | t kr dS | tkrdS d| � �S )N�
ERROR_SUCCESS�ERROR_NOT_ENOUGH_MEMORYzWindows error )rN rO )�errnor, r, r- �_get_error_message� s
z(_WindowsConsoleWriter._get_error_messagec C sl t |�}t|�}t|t�d }t� }tt| j�||t|�d � d|j }|dkrh|dkrht
| � t � ���|S )Nr r )
r r5 �min�MAX_BYTES_WRITTENr �
WriteConsoleWr
r7 r rH rF rQ rB )r8 rI Zbytes_to_be_writtenr Zcode_units_to_be_writtenZcode_units_writtenZ
bytes_writtenr, r, r- �write� s
z_WindowsConsoleWriter.writeN)r' r( r) r3 � staticmethodrQ rU r, r, r, r- rM � s rM c @ s� e Zd Zejejdd�dd�Zeed�dd��Z ej
e d�d d
�Z ej
ej
dd �d d
�Zeejd�dd�Zed�dd�Zdd� ZdS )�
ConsoleStreamN)� text_stream� byte_stream�returnc C s || _ || _d S )N)� _text_streamrJ )r8 rX rY r, r, r- r9 � s zConsoleStream.__init__)rZ c C s | j jS )N)rJ �name)r8 r, r, r- r\ � s zConsoleStream.name)�xrZ c C sD t |t�r| j�|�S y | �� W n tk
r6 Y nX | j�|�S )N)�
isinstance�strr[ rU �flush� ExceptionrJ )r8 r] r, r, r- rU � s
zConsoleStream.write)�linesrZ c C s x|D ]}| � |� qW d S )N)rU )r8 rb �liner, r, r- �
writelines� s
zConsoleStream.writelines)r\ rZ c C s t | j|�S )N)�getattrr[ )r8 r\ r, r, r- � __getattr__� s zConsoleStream.__getattr__c C s
| j �� S )N)rJ r; )r8 r, r, r- r; � s zConsoleStream.isattyc C s d| j �d| j�d�S )Nz<ConsoleStream name=z
encoding=�>)r\ �encoding)r8 r, r, r- �__repr__� s zConsoleStream.__repr__)r' r( r) �t�TextIO�BinaryIOr9 �propertyr_ r\ �AnyStr�intrU �Iterablerd �Anyrf �boolr; ri r, r, r, r- rW � s rW )�
buffer_streamrZ c C s. t t�tt��dddd�}t�tjt|| ��S )Nz utf-16-le�strictT)�line_buffering) r �io�BufferedReaderr> � STDIN_HANDLErj �castrk rW )rs rX r, r, r- �_get_text_stdin� s rz c C s. t t�tt��dddd�}t�tjt|| ��S )Nz utf-16-lert T)ru ) r rv �BufferedWriterrM �
STDOUT_HANDLErj ry rk rW )rs rX r, r, r- �_get_text_stdout� s r} c C s. t t�tt��dddd�}t�tjt|| ��S )Nz utf-16-lert T)ru ) r rv r{ rM �
STDERR_HANDLErj ry rk rW )rs rX r, r, r- �_get_text_stderr� s r )r r r �_stream_factories)�frZ c C sT t | d�sdS y | �� }W n ttjfk
r4 dS X t�|�}tt|t t
� ���S )N�filenoF) �hasattrr� rF rv �UnsupportedOperation�msvcrtZ
get_osfhandlerr �GetConsoleModer r )r� r� r7 r, r, r- � _is_console� s
r� )r� rh �errorsrZ c C sZ t d k rV|dkrV|dkrVt| �rVt�| �� �}|d k rVt| dd �}|d krNd S ||�S d S )N> � utf-16-leN> Nrt rJ )r5 r� r� �getr� re )r� rh r� �funcrI r, r, r- �_get_windows_console_stream s r� )F)Nrv �sysrD �typingrj �ctypesr r r r r r r r r
r Zctypes.wintypesr r
r r �_compatr �platform�AssertionErrorr� r r r* �kernel32Z GetStdHandlerA rT r� rB r �shell32r r rx r| r~ r/ r. rN rO rC Z STDIN_FILENOZ
STDOUT_FILENOZ
STDERR_FILENOrG rS r � ImportErrorr5 r r0 r2 � RawIOBaser6 r> rM rW rl rk rz r} r r� �Mappingro �Callable�__annotations__rr r� �Optionalr_ r� r, r, r, r- �<module> s�
#
*