U
WP @ s0 d dl Z d dlZd dlZd dlZd dlZd dlmZ ejd dkZej
dZ dZ e
dZdd Zd d
Zd d Zd
d ZG dd dejZG dd deZer<eZeZeZeefZ dd Z e Z dd Z e
dZ!e rd 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+nd dlZeZe,ZefZ e-Z d+d Z$d,d Z d-d Z dod/d0Z.dpd1d2Z/d3d4 Z0d5d6 Z1d7d8 Z2d9d: Z3d;d< Z4d=d> Z5d?d Z%d@d Z&dAd" Z'dqdBd$Z(drdCd&Z)dsdDd(Z*dEd* Z+dtdFdGZ6dudJdKZ7e8edLr ej9Z:dMZ;n ej<Z:e Z;G dNdO dOeZ=dZ>dZ?dZ@dPdQ ZAdvdRdSZBe rdTZ dUdVlCmDZD dWdX ZEerdwdZd[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'dkZNe(e)e*dkZOdS )y NWeakKeyDictionary winP z\[((?:\d|;)*)([a-zA-Z])c C s t pt S N)sysgetfilesystemencodinggetdefaultencoding r r /_compat.pyget_filesystem_encoding s r
c C s, |d krt | }|d kr d}t| ||ddS )NreplaceT)line_buffering)get_best_encoding_NonClosingTextIOWrapper)streamencodingerrorsr 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)codecslookupname LookupError)r r r r is_ascii_encoding s r c C s$ t | ddpt }t|r dS |S )z1Returns the default stream encoding if not found.r Nutf-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_streamio
TextIOWrapper__init__)selfr r r Zextrar r r r$ 0 s z!_NonClosingTextIOWrapper.__init__c C sR t |tst|rDz | W n tk
r2 Y nX | jt|S tj| |S r )
isinstancestris_bytesflush Exceptionbufferwriter" r# r% xr r r r, 8 s z _NonClosingTextIOWrapper.writec C s |D ]}| | qd S r )r, )r% linesliner 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 zThe 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 )Nread1)r r! r9 readlineread)r% sizefr r r r< \ s z_FixupStream.read1c C sH t | jdd }|d k r | S z| jd W n tk
rB Y dS X dS )Nreadabler FT)r r! r> r* r- r r r rA g s z_FixupStream.readablec C sp t | jdd }|d k r | S z| jd W n> tk
rj z| jd W n tk
rd Y Y dS X Y nX dS )Nwritable 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 )NseekableFT)r r! seektellr* 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 ) iteritemsr. r r r <lambda> rD rK c C s t | ttfS 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* msvcrtZsetmodeosZO_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_researchrJ r r r isidentifier s rS c C s
t tjS r )rP r stdinr r r r get_binary_stdin s rU c C s
t tjS r )rP r stdoutr r r r get_binary_stdout s rW c C s
t tjS 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 | tr| t d} | S )Nr )r& bytesdecoder
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 )iteritemsrJ r r r rK rD c C s t | tttfS r )r& r_
memoryviewrL rJ r r r r( s Fc C s0 zt | dtW 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 | dr| S t| dd }|d k r0t |dr0|S d S NFr+ T)rj r r Zbufr r r _find_binary_reader s
rn c C s4 t | dr| S t| dd }|d k r0t |dr0|S d S rl )rk r rm r r r _find_binary_writer s
ro c C s t t| ddpdS )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 | dr| }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 | dr| }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 | tr| t d} n| dddd} | S )Nr r surrogateescape)r& r_ r` r
encodera r r r rc s
c C sB t | dr| j}n|d k r |}nt| }t|tr>|dd}|S )Nstrerrorr r )hasattrr| r' r& r_ r` )eri msgr r r
get_streerror s
r rstrictc C s | dkrTd|kr2d|kr"t dfS t||ddfS d|krDt dfS t||ddfS |s|d krnt| |dfS tj| |||ddfS d|krtdd |krtd
d|krtd d d l}|jt j
| d
d\}}|d k rtj||||d}n t ||}t
||| dfS )N-wbF)r r TazAppending 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)dirprefix)rW r] rU r\ openr"
ValueErrortempfileZmkstemprO pathdirnamefdopen _AtomicFile) filenamemoder r Zatomicr fd tmp_filenamer@ r r r open_stream s8
r r Tc @ sJ e Zd Zdd Zedd ZdddZdd Zd
d Zd d
Z dd Z
dS )r c C s || _ || _|| _d| _d S NF)_f
_tmp_filename_real_filenameclosed)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_valuetbr r r __exit__ s z_AtomicFile.__exit__c C s
t | jS 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_resubra 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 }| dS )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| |dj}|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)sZ_writeZ ansi_wrapperr r _safe_write9 s
z'auto_wrap_for_ansi.<locals>._safe_write)_ansi_stream_wrappersgetr* r coloramaZ AnsiToWin32r r, )r r cachedr r r r r r auto_wrap_for_ansi( s"
r c C s* t jt jjj} | 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 cachesrc_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)Prer" rO r r weakrefr version_infor9 platform
startswithZWINZDEFAULT_COLUMNScompiler 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 inputrangerj 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 %