U
祡c @ s d dl mZmZmZmZmZmZ er0d dlmZ ddl m
Z
ddl m Z ddl
mZ erdd dlmZ dZG dd d ZG d
d d Zee ed d
dZee eddddZeeddddZdS ) )
TYPE_CHECKINGAnyDictIterableListSequence)ConsoleRenderable ) get_console)Segment)DEFAULT_TERMINAL_THEMEz{code}
c @ sF e Zd ZdZeeddddZee ee eeeef dddZ dS ) JupyterRenderablez)A shim to write html to Jupyter notebook.N)htmltextreturnc C s || _ || _d S )N)r r )selfr r r 0/tmp/pip-unpacked-wheel-qkhzhrww/rich/jupyter.py__init__ s zJupyterRenderable.__init__)includeexcludekwargsr c sF | j | jd}r(fdd| D } rB fdd| D }|S )Nz
text/plainz text/htmlc s i | ]\}}| kr||qS r r .0kvr r r
s z7JupyterRenderable._repr_mimebundle_..c s i | ]\}}| kr||qS r r r r r r r s )r r items)r r r r datar r r r _repr_mimebundle_ s z#JupyterRenderable._repr_mimebundle_)
__name__
__module__ __qualname____doc__strr r r r r# r r r r r
s
r
c @ s: e Zd ZdZdZdee ee eeeef dddZ dS ) JupyterMixinz@Add to an Rich renderable to make it render in Jupyter notebook.r r )r r r r r c sl t }t|| |j}t|}||}||d}rNfdd| D } rh fdd| D }|S )Nr c s i | ]\}}| kr||qS r r r r r r r 5 s z2JupyterMixin._repr_mimebundle_..c s i | ]\}}| kr||qS r r r r r r r 7 s )r
listrenderoptions_render_segmentsZ_render_bufferr ) r r r r consolesegmentsr r r! r r" r r# ) s
z JupyterMixin._repr_mimebundle_N)
r$ r% r& r' __slots__r r( r r r# r r r r r) $ s
r) )r/ r c C s t t ddd}g }|j}t}t| D ]b\}}}|r8q(||}|r||}|rdd| d| dn|}|jrd|j d| d }|| q(d
|} tj | d }
|
S ) N)r r c S s | dd dd ddS )z Escape html.&z&z>)replace)r r r r escape< s z _render_segments..escapez
zz z )code)
r( appendr r ZsimplifyZget_html_stylelinkjoinJUPYTER_HTML_FORMATformat) r/ r5 Z fragmentsZappend_fragmentZthemer stylecontrolZruler7 r r r r r- ; s"
r- N)r/ r r c C sD t | }t||}zddlm} || W n tk
r> Y nX dS )zRender segments to Jupyter.r )displayN)r- r
ZIPython.displayr? ModuleNotFoundError)r/ r r Zjupyter_renderableZipython_displayr r r r? T s
r? )argsr r c O s t }|j| |S )zProxy for Console print.)r
print)rA r r. r r r rB b s rB )typingr r r r r r Z rich.consoler r6 r
segmentr Zterminal_themer r; r
r) r( r- r? rB r r r r s