File: //usr/lib/python3/dist-packages/click/__pycache__/_winconsole.cpython-38.pyc
U
L�[# � @ s4 d dl Z d dlZd dlZd dlZd dlZd dlZd dlZddlmZm 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)dZ*d Z+d Z,d
Z-d Z.dZ/dZ0dZ1dZ2G dd� dej3�Z4edk�r~dZ5n
d/dd�Z5G dd� de j6�Z7G dd� de7�Z8G dd� de7�Z9G d d � d e:�Z;G d d � d e:�Z<e=� Z>d d!� Z?d"d#� Z@d$d%� ZAd&d'� ZBe
�r d(d)� ZCeC� ZDd*d+� ZEe@eAeBd,�ZFd-d.� ZGdS )0� 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* �3/usr/lib/python3/dist-packages/click/_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` rc r; re r* r* r* r+ rV � s
rV c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �WindowsChunkedWriterz�
Wraps a stream (such as stdout), acting as a transparent proxy for all
attribute access apart from method 'write()' which we wrap to write in
limited chunks due to a Windows limitation on binary console streams.
c C s
|| _ d S r, )� _WindowsChunkedWriter__wrapped)r6 �wrappedr* r* r+ r7 � s z WindowsChunkedWriter.__init__c C s t | j|�S r, )ra rh rb r* r* r+ rc � s z WindowsChunkedWriter.__getattr__c C sH t |�}d}||k rDt|| t�}| j�|||| � � ||7 }q d S )Nr )r rQ rR rh rT )r6 �textZtotal_to_writeZwrittenZto_writer* r* r+ rT � s zWindowsChunkedWriter.writeN)r% r& r' �__doc__r7 rc rT r* r* r* r+ rg � s rg c C sD t r@t�� d d� dkr@| tkr@tt| ttt| ��� t�| � d S )Nr )� r )r �sysZgetwindowsversion�_wrapped_std_streams�setattrrg ra �add)rY r* r* r+ �_wrap_std_stream� s rq 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 �ry c C s$ t t�tt��dddd�}t|| �S rr )r r9 �BufferedWriterrL �
STDOUT_HANDLErV rx r* r* r+ �_get_text_stdout� s �r| c C s$ t t�tt��dddd�}t|| �S rr )r r9 rz rL �
STDERR_HANDLErV rx r* r* r+ �_get_text_stderr� s �r~ c C s t �d�tjdd � ��S )N� r )�zlibZcrc32�joinrm �argvr* r* r* r+ �
_hash_py_argv s r� 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 �hasattrrm r �
startswith)Zargcr� �argr* r� r+ �_get_windows_argv s
r� )r r r c C s| t d k rx|dkrx|dkrxt| d�rx| �� rxt�| �� �}|d k rxts^t| dd �} | d krpd S nt� | �� t
j � || �S d S )N)rs N)rt Nr; rI ) r3 r� r; �_stream_factories�get�filenor ra �msvcrtZsetmode�osZO_BINARY)�frd �errors�funcr* r* r+ �_get_windows_console_stream" s" ���� r� )F)Hr9 r� rm r� rC Zctypesr� Z_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 rw r{ r} r. r- rM rN rB Z STDIN_FILENOZ
STDOUT_FILENOZ
STDERR_FILENOrF rR Z Structurer r3 r: r4 r= rL �objectrV rg �setrn rq ry r| r~ r� Z_initial_argv_hashr� r� r� r* r* r* r+ �<module> s� 4
� ��
$�