U
��W�P � @ s0 d dl Z d dlZd dlZd dlZd dlZd dlmZ ejd dkZej �
d�Z dZ e �
d�Zdd� Zd d
� Zd d � Zd
d� ZG dd� dej�ZG dd� de�Ze�r<eZeZeZeefZ dd� Z e Z dd� Z e �
d�Z!e r�d dl"Z"dd� Z#ndd� Z#dd � Z$d d � Z%d d � Z&d!d"� Z'dld#d$�Z(dmd%d&�Z)dnd'd(�Z*d)d*� Z+n�d dlZeZe,ZefZ e-Z d+d� Z$d,d� Z d-d� Z dod/d0�Z.dpd1d2�Z/d3d4� Z0d5d6� Z1d7d8� Z2d9d:� Z3d;d<� Z4d=d>� Z5d?d � Z%d@d � Z&dAd"� Z'dqdBd$�Z(drdCd&�Z)dsdDd(�Z*dEd*� Z+dtdFdG�Z6dudJdK�Z7e8edL��r ej9Z:dMZ;n ej<Z:e Z;G dNdO� dOe�Z=dZ>dZ?dZ@dPdQ� ZAdvdRdS�ZBe �r�dTZ dUdVlCmDZD dWdX� ZEe�r�dwdZd[�Zz d dl?Z?W n eFk
�r� Y nX e� ZGdxd\d]�Z>d^d_� Z@nd`dX� ZEdad� ZDdbdc� ZHddde� ZIdfdg� ZJeJdhd� e(�ZKeJdid� e)�ZLeJdjd� e*�ZMe%e&e'dk�ZNe(e)e*dk�ZOdS )y� N��WeakKeyDictionary� �win�P z\[((?:\d|;)*)([a-zA-Z])c C s t �� pt �� S �N)�sys�getfilesystemencoding�getdefaultencoding� r r � /_compat.py�get_filesystem_encoding s r
c C s, |d krt | �}|d kr d}t| ||dd�S )N�replaceT)�line_buffering)�get_best_encoding�_NonClosingTextIOWrapper)�stream�encoding�errorsr r r �_make_text_stream s �r c C s. zt �| �jdkW S tk
r( Y dS X dS )z$Checks if a given encoding is ascii.�asciiFN)�codecs�lookup�name� LookupError)r r r r �is_ascii_encoding s r c C s$ t | dd�pt�� }t|�r dS |S )z1Returns the default stream encoding if not found.r N�utf-8)�getattrr r
r �r �rvr r r r & s r c @ s8 e Zd Zdd� Zer$dd� Zdd� Zdd� Zd d
� Zd S ) r c K s( t |� | _}tjj| |||f|� d S r )� _FixupStream�_stream�io�
TextIOWrapper�__init__)�selfr r r Zextrar r r r$ 0 s z!_NonClosingTextIOWrapper.__init__c C sR t |t�st|�rDz | �� W n tk
r2 Y nX | j�t|��S tj�| |�S r ) �
isinstance�str�is_bytes�flush� Exception�buffer�writer" r# �r% �xr r r r, 8 s z _NonClosingTextIOWrapper.writec C s |D ]}| � |� qd S r )r, )r% �lines�liner r r �
writelinesA s z#_NonClosingTextIOWrapper.writelinesc C s& z | � � W n tk
r Y nX d S r )�detachr* �r% r r r �__del__E s z _NonClosingTextIOWrapper.__del__c C s
| j �� S r )r! �isattyr3 r r r r5 K s z _NonClosingTextIOWrapper.isattyN) �__name__�
__module__� __qualname__r$ �PY2r, r1 r4 r5 r r r r r . s r c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )r z�The new io interface needs more from streams than streams
traditionally implement. As such, this fix-up code is necessary in
some circumstances.
c C s
|| _ d S r )r! )r% r r r r r$ V s z_FixupStream.__init__c C s t | j|�S r )r r! �r% r r r r � __getattr__Y s z_FixupStream.__getattr__c C s: t | jdd �}|d k r ||�S tr.| j�|�S | j�|�S )N�read1)r r! r9 �readline�read)r% �size�fr r r r< \ s z_FixupStream.read1c C sH t | jdd �}|d k r |� S z| j�d� W n tk
rB Y dS X dS )N�readabler FT)r r! r> r* r- r r r rA g s z_FixupStream.readablec C sp t | jdd �}|d k r |� S z| j�d� W n> tk
rj z| j�d� W n tk
rd Y Y dS X Y nX dS )N�writable� � FT)r r! r, r* r- r r r rB q s z_FixupStream.writablec C sN t | jdd �}|d k r |� S z| j�| j�� � W n tk
rH Y dS X dS )N�seekableFT)r r! �seek�tellr* r- r r r rE ~ s z_FixupStream.seekableN)
r6 r7 r8 �__doc__r$ r; r< rA rB rE r r r r r P s
r c C s | � � S r )� iteritems�r. r r r �<lambda>� rD rK c C s t | ttf�S r )r& r+ � bytearrayrJ r r r r( � s r( z^[a-zA-Z_][a-zA-Z0-9_]*$c C s4 z | � � }W n tk
r Y nX t�|tj� | S r )�filenor* �msvcrtZsetmode�osZO_BINARY)r@ rM r r r �set_binary_mode� s rP c C s | S r r rJ r r r rK � rD c C s t �| �d k S r )�_identifier_re�searchrJ r r r � isidentifier� s rS c C s
t tj�S r )rP r �stdinr r r r �get_binary_stdin� s rU c C s
t tj�S r )rP r �stdoutr r r r �get_binary_stdout� s rW c C s
t tj�S r )rP r �stderrr r r r �get_binary_stderr� s rY c C s( t tj| |�}|d k r|S ttj| |�S r )�_get_windows_console_streamr rT r �r r r r r r �get_text_stdin� s r\ c C s( t tj| |�}|d k r|S ttj| |�S r )rZ r rV r r[ r r r �get_text_stdout� s r] c C s( t tj| |�}|d k r|S ttj| |�S r )rZ r rX r r[ r r r �get_text_stderr� s r^ c C s t | t�r| �t� d�} | S )Nr )r& �bytes�decoder
��valuer r r �filename_to_ui� s
rc c C s | � � S r )rS rJ r r r rK � rD c C s t | �� �S r )�iter�itemsrJ r r r rK � rD c C s t | tttf�S r )r& r_ �
memoryviewrL rJ r r r r( � s Fc C s0 zt | �d�t�W S tk
r* | Y S X d S �Nr )r& r> r_ r* �r �defaultr r r �_is_binary_reader� s rj c C sT z| � d� W n@ tk
rN z| � d� W Y dS tk
rD Y nX | Y S X dS )NrD rC FT)r, r* rh r r r �_is_binary_writer� s
rk c C s4 t | d�r| S t| dd �}|d k r0t |d�r0|S d S �NFr+ T)rj r �r Zbufr r r �_find_binary_reader� s
rn c C s4 t | d�r| S t| dd �}|d k r0t |d�r0|S d S rl )rk r rm r r r �_find_binary_writer s
ro c C s t t| dd�pd�S )z3A stream is misconfigured if its encoding is ASCII.r Nr )r r �r r r r �_stream_is_misconfigured s rq c C s@ t | dd �}t | dd �}||kr,||kr,dS |d kr<|d k S dS )Nr r TF)r )r r r Zstream_encodingZ
stream_errorsr r r �_is_compatible_text_stream s rr c C s` t | d�r| }n8|d kr$t| �s$| S t| ||�r4| S t| �}|d krH| S |d krTd}t|||�S �NFr )rj rq rr rn r )Z text_readerr r Z
binary_readerr r r �_force_correct_text_reader' s
rt c C s` t | d�r| }n8|d kr$t| �s$| S t| ||�r4| S t| �}|d krH| S |d krTd}t|||�S rs )rk rq rr ro r )Z text_writerr r Z
binary_writerr r r �_force_correct_text_writerC s
ru c C s t tj�} | d krtd��| S )Nz6Was not able to determine binary stream for sys.stdin.)rn r rT � RuntimeError)�readerr r r rU _ s
c C s t tj�} | d krtd��| S )Nz7Was not able to determine binary stream for sys.stdout.)ro r rV rv ��writerr r r rW f s
c C s t tj�} | d krtd��| S )Nz7Was not able to determine binary stream for sys.stderr.)ro r rX rv rx r r r rY m s
c C s( t tj| |�}|d k r|S ttj| |�S r )rZ r rT rt r[ r r r r\ t s c C s( t tj| |�}|d k r|S ttj| |�S r )rZ r rV ru r[ r r r r] z s c C s( t tj| |�}|d k r|S ttj| |�S r )rZ r rX ru r[ r r r r^ � s c C s2 t | t�r| �t� d�} n| �dd��dd�} | S )Nr r �surrogateescape)r& r_ r` r
�encodera r r r rc � s
�c C sB t | d�r| j}n|d k r |}nt| �}t|t�r>|�dd�}|S )N�strerrorr r )�hasattrr| r' r& r_ r` )�eri �msgr r r �
get_streerror� s
r� �r�strictc C s | dkrTd|kr2d|kr"t � dfS t||d�dfS d|krDt� dfS t||d�dfS |s�|d krnt| |�dfS tj| |||d�dfS d|kr�td��d |kr�td
��d|kr�td ��d d l}|jt j
� | �d
d�\}}|d k r�tj||||d�}n t � ||�}t
||| �dfS )N�-�w�bF)r r T�az�Appending to an existing file is not supported, because that would involve an expensive `copy`-operation to a temporary file. Open the file in normal `w`-mode and copy explicitly if that's what you're after.r. z&Use the `overwrite`-parameter instead.z,Atomic writes only make sense with `w`-mode.r z.__atomic-write)�dir�prefix)rW r] rU r\ �openr" �
ValueError�tempfileZmkstemprO �path�dirname�fdopen� _AtomicFile) �filename�moder r Zatomicr� �fd� tmp_filenamer@ r r r � open_stream� s8
��
r� r Tc @ sJ e Zd Zdd� Zedd� �Zddd�Zdd � Zd
d � Zd d
� Z dd� Z
dS )r� c C s || _ || _|| _d| _d S �NF)�_f�
_tmp_filename�_real_filename�closed)r% r@ r� Z
real_filenamer r r r$ � s z_AtomicFile.__init__c C s | j S r )r� r3 r r r r � s z_AtomicFile.nameFc C sV | j r
d S | j�� ts>zt�| j� W n tk
r< Y nX t| j | j� d| _ d S )NT)
r� r� �close� _can_replacerO �remover� �OSError�_replacer� )r% �deleter r r r� � s
z_AtomicFile.closec C s t | j|�S r )r r� r: r r r r; � s z_AtomicFile.__getattr__c C s | S r r r3 r r r � __enter__� s z_AtomicFile.__enter__c C s | j |d k d� d S )N)r� )r� )r% �exc_type� exc_value�tbr r r �__exit__� s z_AtomicFile.__exit__c C s
t | j�S r )�reprr� r3 r r r �__repr__� s z_AtomicFile.__repr__N)F) r6 r7 r8 r$ �propertyr r� r; r� r� r� r r r r r� � s
r� c C s t �d| �S )NrC )�_ansi_re�subra r r r �
strip_ansi� s r� c C s& |d kr | d krt j} t| � S | S r )r rT r5 )r �colorr r r �should_strip_ansi s
r� �O � )rZ c C s dd l } | �� S rg )�localeZgetpreferredencoding)r� r r r �_get_argv_encoding s r� rC c C s2 t j�� | r t� }|�| � t� }|�� �d�S )Nz
)r rX r) �_default_text_stdoutr, �_default_text_stdinr= �rstrip)�promptrV rT r r r � raw_input s
r� c s� zt �| �}W n tk
r& d}Y nX |dk r4|S t| |�}tj| |d���j}|j� � �fdd�}||_z |t | < W n tk
r� Y nX |S )a This function wraps a stream so that calls through colorama
are issued to the win32 console API to recolor on demand. It
also ensures to reset the colors if a write call is interrupted
to not destroy the console afterwards.
N)�stripc s& z
� | �W S �� � � Y nX d S r )Z reset_all)�s�Z_writeZ ansi_wrapperr r � _safe_write9 s
z'auto_wrap_for_ansi.<locals>._safe_write)�_ansi_stream_wrappers�getr* r� �coloramaZ AnsiToWin32r r, )r r� �cachedr� r r� r r� r �auto_wrap_for_ansi( s"
r� c C s* t j�t jj�j} | j| j | j| j fS r ) r� Zwin32ZGetConsoleScreenBufferInfoZSTDOUTZsrWindowZRightZLeftZBottomZTop)r r r r �get_winterm_sizeG s �r� c C s t tjdd �pt� S )Nr )r r rT r
r r r r r� L s c G s d S r r rJ r r r rK O rD c C s t t| ��S r )�lenr� rJ r r r �term_lenR s r� c C s& z
| � � W S tk
r Y dS X d S r� )r5 r* rp r r r r5 V s
r5 c s t � � � ��fdd�}|S )Nc sf �� } z� � | �}W n tk
r, d }Y nX |d k r:|S �� }z |� | < W n tk
r` Y nX |S r )r� r* r ��cache�src_func� wrapper_funcr r �func_ s
z&_make_cached_stream_func.<locals>.funcr )r� r� r� r r� r �_make_cached_stream_func] s r� c C s t jS r )r rT r r r r rK q rD c C s t jS r )r rV r r r r rK s rD c C s t jS r )r rX r r r r rK u rD )rT rV rX )NN)NN)NN)F)F)NN)NN)NN)N)r� Nr� F)NN)rC )N)P�rer" rO r r �weakrefr � version_infor9 �platform�
startswithZWINZDEFAULT_COLUMNS�compiler� r
r r r r# r �objectr ZunicodeZ text_typer' r_ r� Z string_typesrI ZxrangeZ
range_typer( rQ rN rP rS rU rW rY r\ r] r^ rc �input�rangerj rk rn ro rq rr rt ru r� r� r} r r� r� �renamer� r� r� r� r� r� Z _winconsolerZ r� � ImportErrorr� r� r5 r� r� r� Z_default_text_stderrZbinary_streamsZ text_streamsr r r r �<module> s�
"9
�
1 %
� � ���