�
"T�e� � �� � d dl mZmZmZmZmZmZ erd dlmZ ddl m
Z
ddl m Z ddl
mZ erd dlmZ dZ G d� d � Z G d
� d � Zd ee d
efd�Zd ee ded
dfd�Zdeded
dfd�Zy)� )�
TYPE_CHECKING�Any�Dict�Iterable�List�Sequence)�ConsoleRenderable� )� get_console)�Segment)�DEFAULT_TERMINAL_THEMEz�<pre style="white-space:pre;overflow-x:auto;line-height:normal;font-family:Menlo,'DejaVu Sans Mono',consolas,'Courier New',monospace">{code}</pre>
c
�R � e Zd ZdZdededdfd�Zdee dee d edeeef fd
�Z y) �JupyterRenderablez)A shim to write html to Jupyter notebook.�html�text�returnNc � � || _ || _ y )N)r r )�selfr r s �./usr/lib/python3/dist-packages/rich/jupyter.py�__init__zJupyterRenderable.__init__ s � ��� ��� � �include�exclude�kwargsc � � | j | j d�}|r(|j � D ��ci c]
\ }}||v s� ||�� }}}|r(|j � D ��ci c]
\ }}||vs� ||�� }}}|S c c}}w c c}}w �N)z
text/plainz text/html)r r �items)r r r r �data�k�vs r �_repr_mimebundle_z#JupyterRenderable._repr_mimebundle_ sy � � #�i�i�d�i�i�@�� �'+�z�z�|�D�V�a��q�G�|�A�q�D�D�D�D� �'+�z�z�|�H�V�a��q��7G�A�q�D�H�D�H�� �� E��Hs � A0�A0� A6�&A6)
�__name__�
__module__� __qualname__�__doc__�strr r r r r! � r r r r sT � �3��S� �� �� �� ��}��/7��}��HK��
�c�3�h� �r r c �F � e Zd ZdZdZdddee dee dedeeef f
d �Z y
) � JupyterMixinz@Add to an Rich renderable to make it render in Jupyter notebook.r' r r r r r r c �f � t � }t |j | |j � � }t |� }|j |� }||d�}|r(|j
� D � �
ci c]
\ } }
| |v s� | |
�� }} }
|r(|j
� D � �
ci c]
\ } }
| |vs� | |
�� }} }
|S c c}
} w c c}
} w r )r �list�render�options�_render_segments�_render_bufferr ) r r r r �console�segmentsr r r r r s r r! z JupyterMixin._repr_mimebundle_) s� � � �-������t�W�_�_�=�>�� ��)���%�%�h�/�� "��6�� �'+�z�z�|�D�V�a��q�G�|�A�q�D�D�D�D� �'+�z�z�|�H�V�a��q��7G�A�q�D�H�D�H�� �� E��Hs �& B'�3B'� B-� B-N)
r" r# r$ r% � __slots__r r& r r r! r' r r r) r) $ sM � �J��I��!���#� �� �#� �� � �
�c�3�h� � r r) r1 r c �| � dt dt fd�}g }|j }t }t j | � D ]X \ }}}|r�
||� }|r=|j |� }|r d|� d|� d�n|}|j rd|j � d|� d �} ||� �Z d
j |� } t j | � � }
|
S ) Nr r c �f � | j dd� j dd� j dd� S )z Escape html.�&z&�<z<�>z>)�replace)r s r �escapez _render_segments.<locals>.escape< s- � ��|�|�C��)�1�1�#�v�>�F�F�s�F�S�Sr z
<span style="z">z</span>z <a href="z" target="_blank">z</a>� )�code)
r& �appendr
r �simplify�get_html_style�link�join�JUPYTER_HTML_FORMAT�format) r1 r9 � fragments�append_fragment�themer �style�control�ruler; r s r r. r. ; s� � �T�S� T�S� T� �I� �&�&�O� "�E� '� 0� 0�� :� � ��e�W� � ��d�|�� ��'�'��.�D�<@�]�4�&��4�&��8�d�D��z�z�"�5�:�:�,�.@���d�K���� � � �7�7�9� �D� � %� %�4� %� 0�D� �Kr r Nc �n � t | � }t ||� } ddlm} ||� y# t $ r Y yw xY w)zRender segments to Jupyter.r )�displayN)r. r �IPython.displayrJ �ModuleNotFoundError)r1 r r �jupyter_renderable�ipython_displays r rJ rJ T s> � � �H� %�D�*�4��6��
�>��*�+�� �
�
�
�s �( � 4�4�argsr c �: � t � } |j | i |��S )zProxy for Console print.)r �print)rO r r0 s r rQ rQ b s � ��m�G� �7�=�=�$� )�&� )�)r )�typingr r r r r r � rich.consoler r: r �segmentr �terminal_themer
rA r r) r& r. rJ rQ r' r r �<module>rV s� �� E� E��.� � � 2��.�� �
� �$� �.�x�� 0� �S� �2
�h�w�'�
�s�
�t�
� *�� *�� *�� *r |