U
    }Ha{                     @   st   d dl Z d dlZd dlZddlmZ dadadadada	dd Z
ddd	Zd
d Zejdd Zdd Zdd ZdS )    N   )AnsiToWin32Fc                   C   s   t d k	rt t  d S N)r   orig_stdout	reset_all r   r   7/tmp/pip-unpacked-wheel-25st9mmh/colorama/initialise.pyr      s    r   Tc                 C   s   |st | ||grtdtjatjatjd kr6d antt||| | t_atjd kr\d a	ntt||| | t_a	t
stt da
d S )Nz,wrap=False conflicts with any other arg=TrueT)any
ValueErrorsysstdoutr   stderrorig_stderrwrapped_stdoutwrap_streamwrapped_stderratexit_doneatexitregisterr   )	autoresetconvertstripwrapr   r   r   init   s    


r   c                   C   s    t d k	rt t_td k	rtt_d S r   )r   r   r   r   r   r   r   r   r   deinit3   s    r   c                  o   s"   t | | z
d V  W 5 t  X d S r   )r   r   )argskwargsr   r   r   colorama_text:   s    

r   c                   C   s    t d k	rt t_td k	rtt_d S r   )r   r   r   r   r   r   r   r   r   reinitC   s    r   c                 C   s&   |r"t | |||d}| r"|j} | S )N)r   r   r   )r   should_wrapstream)r    r   r   r   r   wrapperr   r   r   r   J   s      r   )FNNT)r   
contextlibr   ansitowin32r   r   r   r   r   r   r   r   r   contextmanagerr   r   r   r   r   r   r   <module>   s   

