��W�P � @ s� 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 e j d d k Z e j j
d � Z d Z e j
d � Z d d � Z d d
� Z d d � Z d
d � Z Gd d � d e j � Z Gd d � d e � Z e r�e Z e Z e Z e e f Z d d � Z e Z d d � Z e j
d � Z! e rWd d l" Z" d d � Z# n d d � Z# d d � Z$ d d � Z% d d � Z& d! d" � Z' d d d# d$ � Z( d d d% d& � Z) d d d' d( � Z* d) d* � Z+ nd d l Z e Z e, Z e f Z e- Z d+ d � Z$ d, d � Z d- d � Z d. d/ d0 � Z. d. d1 d2 � Z/ d3 d4 � Z0 d5 d6 � Z1 d7 d8 � Z2 d9 d: � Z3 d; d< � Z4 d= d> � Z5 d? d � Z% d@ d � Z& dA d" � Z' d d dB d$ � Z( d d dC d& � Z) d d dD d( � Z* dE d* � Z+ d dF dG � Z6 dH d dI d. dJ dK � Z7 e8 e dL � r7e j9 Z: dM Z; n e j< Z: e Z; GdN dO � dO e � Z= d Z> d Z? d Z@ dP dQ � ZA d d dR dS � ZB e rdT Z dU dV lC mD ZD dW dX � ZE e r�dY dZ d[ � Z y d d l? Z? Wn eF k
r�Yq.Xe � ZG d d\ d] � Z> d^ d_ � Z@ n d` dX � ZE da d � ZD db dc � ZH dd de � ZI df dg � ZJ eJ dh d � e( � ZK eJ di d � e) � ZL eJ dj d � e* � ZM dk e% dl e& dm e' i ZN dk e( dl e) dm e* i ZO d S)n� N)�WeakKeyDictionary� �win�P z\[((?:\d|;)*)([a-zA-Z])c C s t j � p t j � S)N)�sys�getfilesystemencoding�getdefaultencoding� r r � /_compat.py�get_filesystem_encoding s r c C s@ | d k r t | � } | d k r* d } t | | | d d �S)N�replace�line_bufferingT)�get_best_encoding�_NonClosingTextIOWrapper)�stream�encoding�errorsr r r
�_make_text_stream s r c C s7 y t j | � j d k SWn t k
r2 d SYn Xd S)z$Checks if a given encoding is ascii.�asciiFN)�codecs�lookup�name� LookupError)r r r r
�is_ascii_encoding s
r c C s2 t | d d � p t j � } t | � r. d S| S)z1Returns the default stream encoding if not found.r Nzutf-8)�getattrr r r )r �rvr r r
r & s r c @ sR e Z d Z d d � Z e r6 d d � Z d d � Z d d � Z d d
� Z d S) r c K s3 t | � | _ } t j j | | | | | � d S)N)� _FixupStream�_stream�io�
TextIOWrapper�__init__)�selfr r r �extrar r r
r 0 s z!_NonClosingTextIOWrapper.__init__c C sg t | t � s t | � rT y | j � Wn t k
r= Yn X| j j t | � � St j j | | � S)N) �
isinstance�str�is_bytes�flush� Exception�buffer�writer r )r! �xr r r
r) 8 s
z _NonClosingTextIOWrapper.writec C s" x | D] } | j | � q Wd S)N)r) )r! �lines�liner r r
�
writelinesA s
z#_NonClosingTextIOWrapper.writelinesc C s' y | j � Wn t k
r" Yn Xd S)N)�detachr' )r! r r r
�__del__E s
z _NonClosingTextIOWrapper.__del__c C s
| j j � S)N)r �isatty)r! r r r
r0 K s z _NonClosingTextIOWrapper.isattyN) �__name__�
__module__� __qualname__r �PY2r) r- r/ r0 r r r r
r . s r c @ s^ e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d S)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)N)r )r! r r r r
r V s z_FixupStream.__init__c C s t | j | � S)N)r r )r! r r r r
� __getattr__Y s z_FixupStream.__getattr__c C sQ t | j d d � } | d k r+ | | � St rA | j j | � S| j j | � S)N�read1)r r r4 �readline�read)r! �size�fr r r
r6 \ s
z_FixupStream.read1c C sY t | j d d � } | d k r( | � Sy | j j d � Wn t k
rT d SYn Xd S)N�readabler FT)r r r8 r' )r! r* r r r
r; g s
z_FixupStream.readablec C s� t | j d d � } | d k r( | � Sy | j j d � Wn? t k
r} y | j j d � Wn t k
rx d SYn XYn Xd S)N�writable� � FT)r r r) r' )r! r* r r r
r< q s
z_FixupStream.writablec C sb t | j d d � } | d k r( | � Sy | j j | j j � � Wn t k
r] d SYn Xd S)N�seekableFT)r r �seek�tellr' )r! r* r r r
r? ~ s
z_FixupStream.seekableN)
r1 r2 r3 �__doc__r r5 r6 r; r< r? r r r r
r P s
r c C s
| j � S)N)� iteritems)r* r r r
�<lambda>� s rD c C s t | t t f � S)N)r# r( � bytearray)r* r r r
r% � s r% z^[a-zA-Z_][a-zA-Z0-9_]*$c C s< y | j � } Wn t k
r$ Yn Xt j | t j � | S)N)�filenor' �msvcrtZsetmode�os�O_BINARY)r: rF r r r
�set_binary_mode� s
rJ c C s | S)Nr )r* r r r
rD � s c C s t j | � d k S)N)�_identifier_re�search)r* r r r
� isidentifier� s rM c C s
t t j � S)N)rJ r �stdinr r r r
�get_binary_stdin� s rO c C s
t t j � S)N)rJ r �stdoutr r r r
�get_binary_stdout� s rQ c C s
t t j � S)N)rJ r �stderrr r r r
�get_binary_stderr� s rS c C s8 t t j | | � } | d k r% | St t j | | � S)N)�_get_windows_console_streamr rN r )r r r r r r
�get_text_stdin� s rU c C s8 t t j | | � } | d k r% | St t j | | � S)N)rT r rP r )r r r r r r
�get_text_stdout� s rV c C s8 t t j | | � } | d k r% | St t j | | � S)N)rT r rR r )r r r r r r
�get_text_stderr� s rW c C s( t | t � r$ | j t � d � } | S)Nr )r# �bytes�decoder )�valuer r r
�filename_to_ui� s r[ c C s
| j � S)N)rM )r* r r r
rD � s c C s t | j � � S)N)�iter�items)r* r r r
rD � s c C s t | t t t f � S)N)r# rX �
memoryviewrE )r* r r r
r% � s Fc C s7 y t | j d � t � SWn t k
r2 | SYn Xd S)Nr )r# r8 rX r' )r �defaultr r r
�_is_binary_reader� s
r` c C sX y | j d � Wn@ t k
rS y | j d � d SWn t k
rJ Yn X| SYn Xd S)Nr> r= FT)r) r' )r r_ r r r
�_is_binary_writer� s
ra c C sH t | d � r | St | d d � } | d k rD t | d � rD | Sd S)NFr( T)r` r )r �bufr r r
�_find_binary_reader� s
rc c C sH t | d � r | St | d d � } | d k rD t | d � rD | Sd S)NFr( T)ra r )r rb r r r
�_find_binary_writer s
rd c C s t t | d d � p d � S)z3A stream is misconfigured if its encoding is ASCII.r Nr )r r )r r r r
�_stream_is_misconfigured s re c C sZ t | d d � } t | d d � } | | k r@ | | k r@ d S| d k rV | d k Sd S)Nr r TF)r )r r r Zstream_encodingZ
stream_errorsr r r
�_is_compatible_text_stream s
rf c C s� t | d � r | } nO | d k r5 t | � r5 | St | | | � rK | St | � } | d k rg | S| d k ry d } t | | | � S)NFr )r` re rf rc r )Z text_readerr r Z
binary_readerr r r
�_force_correct_text_reader' s rg c C s� t | d � r | } nO | d k r5 t | � r5 | St | | | � rK | St | � } | d k rg | S| d k ry d } t | | | � S)NFr )ra re rf rd r )Z text_writerr r Z
binary_writerr r r
�_force_correct_text_writerC s rh c C s+ t t j � } | d k r' t d � � | S)Nz6Was not able to determine binary stream for sys.stdin.)rc r rN � RuntimeError)�readerr r r
rO _ s c C s+ t t j � } | d k r' t d � � | S)Nz7Was not able to determine binary stream for sys.stdout.)rd r rP ri )�writerr r r
rQ f s c C s+ t t j � } | d k r' t d � � | S)Nz7Was not able to determine binary stream for sys.stderr.)rd r rR ri )rk r r r
rS m s c C s8 t t j | | � } | d k r% | St t j | | � S)N)rT r rN rg )r r r r r r
rU t s c C s8 t t j | | � } | d k r% | St t j | | � S)N)rT r rP rh )r r r r r r
rV z s c C s8 t t j | | � } | d k r% | St t j | | � S)N)rT r rR rh )r r r r r r
rW � s c C sI t | t � r' | j t � d � } n | j d d � j d d � } | S)Nr zutf-8�surrogateescape)r# rX rY r �encode)rZ r r r
r[ � s
c C sa t | d � r | j } n! | d k r0 | } n t | � } t | t � r] | j d d � } | S)N�strerrorzutf-8r )�hasattrrn r$ r# rX rY )�er_ �msgr r r
�
get_streerror� s rr �r�strictc C s� | d k r| d | k rJ d | k r1 t � d f St d | d | � d f Sd | k rc t � d f St d | d | � d f S| s� | d k r� t | | � d f St j | | d | d | �d f Sd | k r� t d � � d
| k r� t d � � d | k r t d � � d
d l } | j d t j
j | � d d � \ } } | d k rnt j | | d | d | �} n t j | | � } t
| | | � d f S)N�-�w�bFr 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 �dir�prefixz.__atomic-write)rQ rV rO rU �openr �
ValueError�tempfile�mkstemprH �path�dirname�fdopen� _AtomicFile) �filename�moder r Zatomicr} �fd� tmp_filenamer: r r r
� open_stream� s4
" !r� r Tc @ sm e Z d Z d d � Z e d d � � Z d d d � Z d d � Z d
d � Z d d
� Z d d � Z
d S)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)N)r� )r! r r r
r � s z_AtomicFile.nameFc C si | j r
d S| j j � t sI y t j | j � Wn t k
rH Yn Xt | j | j � d | _ d S)NT)
r� r� �close� _can_replacerH �remover� �OSError�_replacer� )r! �deleter r r
r� � s
z_AtomicFile.closec C s t | j | � S)N)r r� )r! r r r r
r5 � s z_AtomicFile.__getattr__c C s | S)Nr )r! r r r
� __enter__� s z_AtomicFile.__enter__c C s | j d | d k � d S)Nr� )r� )r! �exc_type� exc_value�tbr r r
�__exit__� s z_AtomicFile.__exit__c C s
t | j � S)N)�reprr� )r! r r r
�__repr__� s z_AtomicFile.__repr__N) r1 r2 r3 r �propertyr r� r5 r� r� r� r r r r
r� � s r� c C s t j d | � S)Nr= )�_ansi_re�sub)rZ r r r
�
strip_ansi� s r� c C s1 | d k r, | d k r! t j } t | � S| S)N)r rN r0 )r �colorr r r
�should_strip_ansi s
r� �O � )rT c C s d d l } | j � S)Nr )�locale�getpreferredencoding)r� r r r
�_get_argv_encoding s r� r= c C sE t j j � | r) t � } | j | � t � } | j � j d � S)Nz
)r rR r& �_default_text_stdoutr) �_default_text_stdinr7 �rstrip)�promptrP rN r r r
� raw_input s
r� c s� y t j | � } Wn t k
r- d } Yn X| d k r> | St | | � } t j | d | �� � j } | j � � � f d d � } | | _ y | t | <Wn t k
r� Yn X| 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* y � | � SWn � j � � Yn Xd S)N)Z reset_all)�s)�_write� 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� r
�auto_wrap_for_ansi( s"
r� c C s9 t j j t j j � j } | j | j | j | j f S)N) r� �win32ZGetConsoleScreenBufferInfo�STDOUTZsrWindowZRightZLeftZBottomZTop)r r r r
�get_winterm_sizeG s r� c C s t t j d d � p t � S)Nr )r r rN r r r r r
r� L s c G s d S)Nr )r* r r r
rD O s c C s t t | � � S)N)�lenr� )r* r r r
�term_lenR s r� c C s+ y | j � SWn t k
r& d SYn Xd S)NF)r0 r' )r r r r
r0 V s
r0 c s% t � � � � � f d d � } | S)Nc sw � � } y � j | � } Wn t k
r6 d } Yn X| d k rG | S� � } y | � | <Wn t k
rr Yn X| S)N)r� r' )r r )�cache�src_func� wrapper_funcr r
�func_ s
z&_make_cached_stream_func.<locals>.func)r )r� r� r� r )r� r� r� r
�_make_cached_stream_func] s r� c C s t j S)N)r rN r r r r
rD q s c C s t j S)N)r rP r r r r
rD s s c C s t j S)N)r rR r r r r
rD u s rN rP rR )P�rer rH r r �weakrefr � version_infor4 �platform�
startswithZWINZDEFAULT_COLUMNS�compiler� r r r r r r �objectr �unicodeZ text_typer$ rX r� Z string_typesrC �xrangeZ
range_typer% rK rG rJ rM rO rQ rS rU rV rW r[ �input�ranger` ra rc rd re rf rg rh rr r� ro r r� r� �renamer� r� r� r� r� r� Z _winconsolerT r� � ImportErrorr� r� r0 r� r� r� Z_default_text_stderrZbinary_streamsZ text_streamsr r r r
�<module> s� "9
0 %