U
Id @ 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 Q/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/pip/_vendor/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_segments_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) )r0 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 simplifyget_html_stylelinkjoinJUPYTER_HTML_FORMATformat) r0 r6 fragmentsappend_fragmentthemer stylecontrolruler8 r r r r r- ; s"
r- N)r0 r r c C sD t | }t||}zddlm} || W n tk
r> Y nX dS )zRender segments to Jupyter.r )displayN)r- r
IPython.displayrF ModuleNotFoundError)r0 r r jupyter_renderableipython_displayr r r rF T s
rF )argsr r c O s t }|j| |S )zProxy for Console print.)r
print)rK r r/ r r r rL b s rL )typingr r r r r r pip._vendor.rich.consoler r7 r
segmentr terminal_themer r> r
r) r( r- rF rL r r r r s