U
    祡c\7                     @   sT  U d dl Z d dlmZmZmZ d dlmZ d dlmZm	Z	m
Z
mZmZmZmZmZ ddlmZ ddlmZmZmZmZ ddlmZ dd	lmZ dd
lmZ ddlmZmZ ddl m!Z! ddl"m#Z# G dd deZ$G dd deeZ%e&dkrPd dl'Z'd dl(Z(d dl)m*Z* d dlm+Z+mZm,Z, ddl-m.Z. ddlmZ ddl/m%Z% ddl0m1Z1 ddl2m3Z3 ddl4m5Z5 ddl6m7Z7 e Ze5ddddZ4e7d d!d"Z6e68d#d$d% d&e1d'd(e6d)d*d+d,d-e4e3d.g	Z9e*e9Z:d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAgZ;e%edBHZ<i Z=e+e,e>e>f e?f e@dC< eAdDD ]ZBe;eBeCe;  ZDe;D ]ZEeEeDkrTqBe(FdE e'Gd dFdk r~eHeIe: dGe'' dH d  e=eDeEf< eCe=eCe;d kre=JeKe=L d   e7dIdJZ6e6MdK e6MdL e6MdM e=N D ]6\\ZOZPZQe68eOePe#eQdNeQdOk r dPndQdR qe<Re.Se6 qBq(W 5 Q R X dS )S    N)EventRLockThread)TracebackType)IOAnyCallableListOptionalTextIOTypecast   )get_console)ConsoleConsoleRenderableRenderableType
RenderHook)Control)	FileProxy)JupyterMixin)
LiveRenderVerticalOverflowMethod)Screen)Textc                       sF   e Zd ZdZdedd fddZdddd	Zddd
dZ  ZS )_RefreshThreadz3A thread that calls refresh() at regular intervals.LiveN)liverefresh_per_secondreturnc                    s&   || _ || _t | _t jdd d S )NT)daemon)r   r   r   donesuper__init__)selfr   r   	__class__ -/tmp/pip-unpacked-wheel-qkhzhrww/rich/live.pyr#      s    z_RefreshThread.__init__r   c                 C   s   | j   d S N)r!   setr$   r'   r'   r(   stop   s    z_RefreshThread.stopc              	   C   s@   | j d| j s<| jj | j  s0| j  W 5 Q R X q d S )Nr   )r!   waitr   r   _lockis_setrefreshr,   r'   r'   r(   run   s    

z_RefreshThread.run)	__name__
__module____qualname____doc__floatr#   r-   r2   __classcell__r'   r'   r%   r(   r      s   r   c                   @   s<  e Zd ZdZd)dddddddddd	ee ee eeeeeee	ee
g ef  ddd	d
ZeedddZedddZd*eddddZddddZd dddZeee  ee ee ddddZddddZddddZeedddZdd eedd!d"d#Zddd$d%Zee ee d&d'd(ZdS )+r   a  Renders an auto-updating live display of any given renderable.

    Args:
        renderable (RenderableType, optional): The renderable to live display. Defaults to displaying nothing.
        console (Console, optional): Optional Console instance. Default will an internal Console instance writing to stdout.
        screen (bool, optional): Enable alternate screen mode. Defaults to False.
        auto_refresh (bool, optional): Enable auto refresh. If disabled, you will need to call `refresh()` or `update()` with refresh flag. Defaults to True
        refresh_per_second (float, optional): Number of times per second to refresh the live display. Defaults to 4.
        transient (bool, optional): Clear the renderable on exit (has no effect when screen=True). Defaults to False.
        redirect_stdout (bool, optional): Enable redirection of stdout, so ``print`` may be used. Defaults to True.
        redirect_stderr (bool, optional): Enable redirection of stderr. Defaults to True.
        vertical_overflow (VerticalOverflowMethod, optional): How to handle renderable when it is too tall for the console. Defaults to "ellipsis".
        get_renderable (Callable[[], RenderableType], optional): Optional callable to get renderable. Defaults to None.
    NFT   Zellipsis)	consolescreenauto_refreshr   	transientredirect_stdoutredirect_stderrvertical_overflowget_renderable)
renderabler:   r;   r<   r   r=   r>   r?   r@   rA   r   c       	         C   s   |dkst d|| _|d k	r"|nt | _|| _d| _|| _|| _d | _d | _	t
 | _d | _|| _d| _|rpdn|| _d | _|| _|	| _|
| _t|  |	d| _d S )Nr   zrefresh_per_second must be > 0FT)r@   )AssertionError_renderabler   r:   _screen_alt_screen_redirect_stdout_redirect_stderr_restore_stdout_restore_stderrr   r/   
ipy_widgetr<   _startedr=   _refresh_threadr   r@   _get_renderabler   rA   _live_render)r$   rB   r:   r;   r<   r   r=   r>   r?   r@   rA   r'   r'   r(   r#   3   s,     zLive.__init__r)   c                 C   s   | j S )z'Check if live display has been started.)rL   r,   r'   r'   r(   
is_started[   s    zLive.is_startedc                 C   s    | j d k	r|   n| j}|pdS )N )rN   rD   r$   rB   r'   r'   r(   rA   `   s
    
zLive.get_renderable)r1   r   c              	   C   s   | j  | jrW 5 Q R  dS | j|  d| _| jrB| jd| _| jd |   | j	|  |rz| 
  W n tk
r   |    Y nX | jrt| | j| _| j  W 5 Q R X dS )z|Start live rendering display.

        Args:
            refresh (bool, optional): Also refresh. Defaults to False.
        NTF)r/   rL   r:   Zset_liverE   set_alt_screenrF   show_cursor_enable_redirect_ioZpush_render_hookr1   	Exceptionr-   r<   r   r   rM   start)r$   r1   r'   r'   r(   rW   h   s&    z
Live.startc                 C   s  | j   | jsW 5 Q R  dS | j  d| _| jrN| jdk	rN| j  d| _d| _| j z| j
st| jjst|   W 5 |   | j	  | j
s| jjr| j  | jd | j
r| jd | jr| j
s| j| j  | jdk	r| jr| j  X W 5 Q R X W 5 Q R X dS )zStop live rendering display.NFZvisibleT)r/   rL   r:   Z
clear_liver<   rM   r-   r@   _disable_redirect_ioZpop_render_hookrF   is_terminallinerT   rS   r=   controlrO   Zrestore_cursorrK   close
is_jupyterr1   r,   r'   r'   r(   r-      s0    




z	Live.stopc                 C   s   | j | jd k	d | S )Nr1   )rW   rD   r,   r'   r'   r(   	__enter__   s    zLive.__enter__)exc_typeexc_valexc_tbr   c                 C   s   |    d S r*   )r-   )r$   r`   ra   rb   r'   r'   r(   __exit__   s    zLive.__exit__c                 C   st   | j js| j jrp| jr@ttjts@tj| _t	dt| j tjt_| j
rpttjtsptj| _t	dt| j tjt_dS )z&Enable redirecting of stdout / stderr.r   N)r:   rY   r]   rG   
isinstancesysstdoutr   rI   r   rH   stderrrJ   r,   r'   r'   r(   rU      s    zLive._enable_redirect_ioc                 C   s8   | j rtd| j t_d| _ | jr4td| jt_d| _dS )z'Disable redirecting of stdout / stderr.r   N)rI   r   re   rf   rJ   rg   r,   r'   r'   r(   rX      s    zLive._disable_redirect_ioc                 C   s   |   }| jrt|S |S )zwGet the renderable that is being displayed

        Returns:
            RenderableType: Displayed renderable.
        )rA   rF   r   rR   r'   r'   r(   rB      s    zLive.renderabler^   )rB   r1   r   c             	   C   s(   | j  || _|r|   W 5 Q R X dS )zUpdate the renderable that is being displayed

        Args:
            renderable (RenderableType): New renderable to use.
            refresh (bool, optional): Refresh the display. Defaults to False.
        N)r/   rD   r1   )r$   rB   r1   r'   r'   r(   update   s    zLive.updatec              
   C   s"  | j  | j| j | jjrzddlm} ddlm	} W n& t
k
rb   ddl}|d Y nNX | jdkr| | _|| j | j$ | jjdd | j| jj W 5 Q R X nb| jjr| jjs| j | jt  W 5 Q R X n0| js| js| j | jt  W 5 Q R X W 5 Q R X dS )z&Update the display of the Live Render.r   )display)OutputNz(install "ipywidgets" for Jupyter supportT)r.   )r/   rO   Zset_renderablerB   r:   r]   ZIPython.displayri   Z
ipywidgetsrj   ImportErrorwarningswarnrK   Zclear_outputprintrY   Zis_dumb_terminalr   rL   r=   )r$   ri   rj   rl   r'   r'   r(   r1      s0    


zLive.refresh)renderablesr   c              	   C   sj   | j | j_ | jjrN| j. | jr(t n| j }|f|| jf}W 5 Q R X n| j	sf| j
sf|| jf}|S )z;Process renderables to restore cursor and display progress.)r@   rO   r:   Zis_interactiver/   rF   r   homeZposition_cursorrL   r=   )r$   ro   resetr'   r'   r(   process_renderables   s    

zLive.process_renderables)N)F)r3   r4   r5   r6   r
   r   r   boolr7   r   r   r#   propertyrP   rA   rW   r-   r_   r   BaseExceptionr   rc   rU   rX   rB   rh   r1   r	   r   rr   r'   r'   r'   r(   r   #   sZ    (

		r   __main__)cycle)Dictr	   Tuple)Align)r   )r   )Panel)Rule)Syntax)Tablea~  def loop_last(values: Iterable[T]) -> Iterable[Tuple[bool, T]]:
    """Iterate and generate a tuple with a flag for last value."""
    iter_values = iter(values)
    try:
        previous_value = next(iter_values)
    except StopIteration:
        return
    for value in iter_values:
        yield False, previous_value
        previous_value = value
    yield True, previous_valuepythonT)Zline_numbersZfoobarZbaz123zYou can make the terminal shorter and taller to see the live table hideText may be printed while the progress bars are rendering.z(In fact, [i]any[/i] renderable will workzSuch as [magenta]tables[/]...zPretty printed structures...ZexamplezPretty printed)typetextz	Syntax...zGive it a try!ZSGDZMYRZEURZUSDZAUDZJPYZCNHZHKDZCADZINRZDKKZGBPZRUBZNZDZMXNZIDRZTWDZTHBZVND)r:   exchange_rate_dictd   g?
      i@  zExchange Rates)titlezSource CurrencyzDestination CurrencyzExchange Ratez.4fg      ?ZredZgreen)style)Tre   	threadingr   r   r   typesr   typingr   r   r   r	   r
   r   r   r   rQ   r   r:   r   r   r   r   r[   r   Z
file_proxyr   Zjupyterr   Zlive_renderr   r   r;   r   r   r   r   r   r3   randomtime	itertoolsrw   rx   ry   Zalignrz   r   Zpanelr{   Zruler|   Zsyntaxr}   tabler~   Zadd_rowZprogress_renderablesZexamplesZ	exchangesZ
live_tabler   strr7   __annotations__rangeindexlenZselect_exchangeZexchangesleeprandintlognextpoplistkeysZ
add_columnitemssourcedestZexchange_raterh   centerr'   r'   r'   r(   <module>   s    ( k






	