File: //usr/local/lib/python3.10/dist-packages/rich/__pycache__/markdown.cpython-310.pyc
o
.��f7f � @ s d dl mZ d dlZd dlmZmZmZmZmZm Z m
Z
d dl m Z d dl
mZ ejdkr4d dlmZ nd dlmZ d dlmZ d d
lmZ d d lmZ d d lmZ d d
lmZm Z m Z m Z d dl m Z d dl!m"Z" d dl#m$Z$ d dl%m&Z& d dl'm(Z( d dl)m*Z*m+Z+ d dl,m-Z- d dl.m/Z/m0Z0 G dd� d�Z1G dd� de1�Z2G dd� de1�Z3G d d � d e3�Z4G d d � d e3�Z5G d d!� d!e3�Z6G d"d#� d#e3�Z7G d$d%� d%e1�Z8G d&d'� d'e1�Z9G d(d)� d)e1�Z:G d*d+� d+e1�Z;G d,d-� d-e1�Z<G d.d/� d/e1�Z=G d0d1� d1e1�Z>G d2d3� d3e3�Z?G d4d5� d5e3�Z@G d6d7� d7e3�ZAG d8d9� d9�ZBG d:d;� d;e"�ZCeDd<k�rd dlEZEd dlZeEjFd=d>�ZGeGjHd?d@dAdB� eGjHdCdDdEdFddGdH� eGjHdIdJdKdLdMdN� eGjHdOdPdQddQdN� eGjHdRdSdTdFdUdV� eGjHdWdXeIdYddZd[� eGjHd\d]d^dFd_dV� eGjHd`dadbdFdcdV� eG�J� ZKd ddlLmZ eKjMdek�r�ejN�O� ZPn eQeKjMdfdgdh�� ZReR�O� ZPW d � n 1 �s�w Y eCePeKjS�r�dindjeKjTeKjUeKjVdk�ZWeKjX�r�d dlYZYd dlZZZeY�[� Z\ee\eKj]eKj^dl�Ze�_eW� eZ�`e\�a� � dS eeKj]eKj^dmdn�Ze�_eW� dS dS )o� )� annotationsN)�ClassVar�Dict�Iterable�List�Optional�Type�Union)�
MarkdownIt)�Token)� � )�get_args)�Table� ��box)�
loop_first)�Stack)�Console�ConsoleOptions�
JustifyMethod� RenderResult)� Renderables)� JupyterMixin)�Panel)�Rule)�Segment)�Style�
StyleStack)�Syntax)�Text�TextTypec @ sZ e Zd ZU dZded< ed#d
d ��Zd$dd�Zd%dd�Zd$dd�Z d&dd�Z
d'd d!�Z d"S )(�MarkdownElementTzClassVar[bool]�new_line�markdown�
'Markdown'�tokenr �return�'MarkdownElement'c C s | � S )z�Factory to create markdown element,
Args:
markdown (Markdown): The parent Markdown object.
token (Token): A node from markdown-it.
Returns:
MarkdownElement: A new markdown element
� ��clsr% r' r* r* �8/usr/local/lib/python3.10/dist-packages/rich/markdown.py�create! s zMarkdownElement.create�context�'MarkdownContext'�Nonec C � dS )zuCalled when the node is entered.
Args:
context (MarkdownContext): The markdown context.
Nr* ��selfr/ r* r* r- �on_enter. � zMarkdownElement.on_enter�textr" c C r2 )zpCalled when text is parsed.
Args:
context (MarkdownContext): The markdown context.
Nr* �r4 r/ r7 r* r* r- �on_text5 r6 zMarkdownElement.on_textc C r2 )zwCalled when the parser leaves the element.
Args:
context (MarkdownContext): [description]
Nr* r3 r* r* r- �on_leave< r6 zMarkdownElement.on_leave�child�boolc C r2 )a| Called when a child element is closed.
This method allows a parent element to take over rendering of its children.
Args:
context (MarkdownContext): The markdown context.
child (MarkdownElement): The child markdown element.
Returns:
bool: Return True to render the element, or False to not render the element.
Tr* �r4 r/ r; r* r* r- �on_child_closeC s z MarkdownElement.on_child_close�console� 'Console'�options�'ConsoleOptions'�'RenderResult'c C r2 )Nr* r* �r4 r? rA r* r* r- �__rich_console__S s z MarkdownElement.__rich_console__N�r% r& r' r r( r) �r/ r0 r( r1 �r/ r0 r7 r" r( r1 �r/ r0 r; r) r( r< )r? r@ rA rB r( rC ) �__name__�
__module__� __qualname__r$ �__annotations__� classmethodr. r5 r9 r: r> rE r* r* r* r- r# s
r# c @ s e Zd ZdZdS )�UnknownElementz�An unknown element.
Hopefully there will be no unknown elements, and we will have a MarkdownElement for
everything in the document.
N)rJ rK rL �__doc__r* r* r* r- rO Y s rO c @ s2 e Zd ZdZdZddd�Zdd d �Zdd
d�ZdS )� TextElementz)Base class for elements that render text.�noner/ r0 r( r1 c C s |� | j�| _tdd�| _d S �N�left��justify)� enter_style�
style_name�styler! r7 r3 r* r* r- r5 g s zTextElement.on_enterr7 r" c C s( | j �|t|t�r|j� d S d � d S �N)r7 �append�
isinstance�str�
current_styler8 r* r* r- r9 k s (zTextElement.on_textc C s |� � d S rZ )� leave_styler3 r* r* r- r: n � zTextElement.on_leaveNrG rH )rJ rK rL rP rX r5 r9 r: r* r* r* r- rQ b s
rQ c @ s@ e Zd ZU dZdZded< edd d ��Zddd�Zddd�Z dS )� Paragraphz A Paragraph.zmarkdown.paragraphr rV r% r& r' r r( � 'Paragraph'c C s | |j pdd�S rS rU r+ r* r* r- r. x s zParagraph.creater1 c C s
|| _ d S rZ rU �r4 rV r* r* r- �__init__| �
zParagraph.__init__r? r rA r r c c s � | j | j_ | jV d S rZ )rV r7 rD r* r* r- rE s �
zParagraph.__rich_console__N)r% r& r' r r( rb �rV r r( r1 �r? r rA r r( r )
rJ rK rL rP rX rM rN r. rd rE r* r* r* r- ra r s
ra c sD e Zd ZdZeddd ��Zdd
d�Zd � fdd� Zd dd�Z� Z S ) �Headingz
A heading.r% r& r' r r( � 'Heading'c C s
| |j �S rZ )�tagr+ r* r* r- r. � s
zHeading.creater/ r0 r1 c C s t � | _|�| j� d S rZ )r! r7 rW rX r3 r* r* r- r5 � s zHeading.on_enterrj r] c s || _ d|� �| _t� �� d S )N� markdown.)rj rX �superrd )r4 rj �� __class__r* r- rd � s zHeading.__init__r? r rA r r c c sL � | j }d|_| jdkrt|tjdd�V d S | jdkr!td�V |V d S )N�center�h1zmarkdown.h1.border)r rY �h2� )r7 rV rj r r �HEAVYr! )r4 r? rA r7 r* r* r- rE � s �
�
zHeading.__rich_console__)r% r& r' r r( ri rG )rj r] r( r1 rg )
rJ rK rL rP rN r. r5 rd rE �
__classcell__r* r* rm r- rh � s
rh c @ s6 e Zd ZdZdZedd d
��Zddd�Zddd�ZdS ) � CodeBlockz&A code block with syntax highlighting.zmarkdown.code_blockr% r& r' r r( � 'CodeBlock'c C s( |j pd}|�d�d }| |pd|j�S )Nrr � r r7 )�info� partition�
code_theme)r, r% r' � node_info�
lexer_namer* r* r- r. � s
zCodeBlock.creater| r] �themer1 c C s || _ || _d S rZ )r| r} )r4 r| r} r* r* r- rd � �
zCodeBlock.__init__r? r rA r r c c s0 � t | j��� }t|| j| jddd�}|V d S )NTr )r} � word_wrap�padding)r] r7 �rstripr r| r} )r4 r? rA �code�syntaxr* r* r- rE � s ��
zCodeBlock.__rich_console__N)r% r& r' r r( rv )r| r] r} r] r( r1 rg ) rJ rK rL rP rX rN r. rd rE r* r* r* r- ru � s
ru c @ s2 e Zd ZdZdZddd�Zdd d
�Zddd�ZdS )�
BlockQuotezA block quote.zmarkdown.block_quoter( r1 c C � t � | _d S rZ �r �elements�r4 r* r* r- rd � r` zBlockQuote.__init__r/ r0 r; r) r< c C � | j �|� dS �NF�r� r[ r= r* r* r- r> � � zBlockQuote.on_child_closer? r rA r r c c sd � |j |jd d�}|j| j|| jd�}| j}td�}td|�}|D ]
}|V |E d H |V q"d S )N� ��width�rY �
u ▌ )�update� max_width� render_linesr� rY r ) r4 r? rA �render_options�linesrY r$ r� �liner* r* r- rE � s �
�zBlockQuote.__rich_console__N�r( r1 rI rg )rJ rK rL rP rX rd r> rE r* r* r* r- r� � s
r� c @ s e Zd ZdZdZd d d
�Zd S )
�HorizontalRulez%A horizontal rule to divide sections.Fr? r rA r r( r c c s � |j ddd�}t|d�V d S )Nz markdown.hrrR ��defaultr� )� get_styler )r4 r? rA rY r* r* r- rE � s �z HorizontalRule.__rich_console__Nrg )rJ rK rL rP r$ rE r* r* r* r- r� � s r� c @ s. e Zd ZdZddd�Zdd d �Zddd�ZdS )� TableElementz.MarkdownElement corresponding to `table_open`.r( r1 c C s d | _ d | _d S rZ )�header�bodyr� r* r* r- rd � r~ zTableElement.__init__r/ r0 r; r) r< c C s0 t |t�r
|| _dS t |t�r|| _dS td��)Nz Couldn't process markdown table.F)r\ �TableHeaderElementr� �TableBodyElementr� � RuntimeErrorr= r* r* r- r> � s
��zTableElement.on_child_closer? r rA r r c c s| � t tjd�}| jd ur | jjd ur | jjjD ]}|�|j� q| jd ur9| jj D ]}dd� |jD �}|j
|� q)|V d S )Nr c S s g | ]}|j �qS r* )�content)�.0�elementr* r* r- �
<listcomp> s z1TableElement.__rich_console__.<locals>.<listcomp>) r r � SIMPLE_HEAVYr� �row�cells�
add_columnr� r� �rows�add_row)r4 r? rA �table�columnr� � row_contentr* r* r- rE � s �
z TableElement.__rich_console__Nr� rI rg )rJ rK rL rP rd r> rE r* r* r* r- r� � s
r� c @ �$ e Zd ZdZddd�Zdd d �Zd
S )r� z@MarkdownElement corresponding to `thead_open` and `thead_close`.r( r1 c C s
d | _ d S rZ )r� r� r* r* r- rd re zTableHeaderElement.__init__r/ r0 r; r) r< c C s t |t�sJ �|| _dS r� )r\ �TableRowElementr� r= r* r* r- r> s z!TableHeaderElement.on_child_closeNr� rI �rJ rK rL rP rd r> r* r* r* r- r�
�
r� c @ r� )r� z@MarkdownElement corresponding to `tbody_open` and `tbody_close`.r( r1 c C �
g | _ d S rZ )r� r� r* r* r- rd re zTableBodyElement.__init__r/ r0 r; r) r< c C � t |t�sJ �| j�|� dS r� )r\ r� r� r[ r= r* r* r- r> ! � z TableBodyElement.on_child_closeNr� rI r� r* r* r* r- r� r� r� c @ r� )r� z:MarkdownElement corresponding to `tr_open` and `tr_close`.r( r1 c C r� rZ )r� r� r* r* r- rd , re zTableRowElement.__init__r/ r0 r; r) r< c C r� r� )r\ �TableDataElementr� r[ r= r* r* r- r> / r� z TableRowElement.on_child_closeNr� rI r� r* r* r* r- r� ) r� r� c @ s2 e Zd ZdZeddd ��Zdd
d�Zddd�ZdS )r� z[MarkdownElement corresponding to `td_open` and `td_close`
and `th_open` and `th_close`.r% r& r' r r( r) c C s\ t |j�d��p d}d|v rd}nd|v rd}n d|v r d}nd }|tt�v s)J �| |d
�S ) NrY rr ztext-align:right�rightztext-align:centerro ztext-align:leftrT r� rU )r] �attrs�getr r )r, r% r' rY rV r* r* r- r. ; s
zTableDataElement.createrV r r1 c C s t d|d�| _|| _d S )Nrr rU )r! r� rV rc r* r* r- rd L s
zTableDataElement.__init__r/ r0 r7 r" c C s2 t |t�r t|�n|}|�|j� | j�|� d S rZ )r\ r] r! �stylizer^ r� � append_textr8 r* r* r- r9 P s zTableDataElement.on_textNrF rf rH )rJ rK rL rP rN r. rd r9 r* r* r* r- r� 7 s
r� c @ s< e Zd ZdZed dd ��Zd!dd�Zd"dd�Zd#d d �Zd S )$� ListElementzA list element.r% r& r' r r( �
'ListElement'c C s | |j t|j�dd���S )N�startr )�type�intr� r� r+ r* r* r- r. Y s zListElement.create� list_typer] �
list_start�
int | Noner1 c C s g | _ || _|| _d S rZ )�itemsr� r� )r4 r� r� r* r* r- rd ] s
zListElement.__init__r/ r0 r; r) r< c C r� r� )r\ �ListItemr� r[ r= r* r* r- r> b r� zListElement.on_child_closer? r rA r r c c s� � | j dkr| jD ] }|�||�E d H q d S | jd u r dn| j}|t| j� }t| j�D ]\}}|�|||| |�E d H q-d S )N�bullet_list_openr )r� r� �
render_bulletr� �len� enumerate�
render_number)r4 r? rA �item�number� last_number�indexr* r* r- rE i s �
� ��z ListElement.__rich_console__N)r% r& r' r r( r� )r� r] r� r� r( r1 rI rg ) rJ rK rL rP rN r. rd r> rE r* r* r* r- r� V s
r� c @ s< e Zd ZdZdZddd�Zd d d
�Zd dd�Zd dd�ZdS ) r� zAn item in a list.z
markdown.itemr( r1 c C r� rZ r� r� r* r* r- rd } r` zListItem.__init__r/ r0 r; r) r< c C r� r� r� r= r* r* r- r> � r� zListItem.on_child_closer? r rA r r c c s� � |j |jd d�}|j| j|| jd�}|jddd�}td|�}td|�}td �}t|�D ]\} }
| r5|n|V |
E d H |V q-d S )
Nr r� r� zmarkdown.item.bulletrR r� u • z r� )r� r� r� r� rY r� r r ) r4 r? rA r� r� � bullet_style�bulletr� r$ �firstr� r* r* r- r� � s �
�zListItem.render_bulletr� r� r� c c s� � t t|��d }|j|j| d�}|j| j|| jd�}|jddd�}td�} td| |�}
t|� � |d �d |�} t
|�D ]\} }
| rG| n|
V |
E d H | V q?d S )
N� r� r� zmarkdown.item.numberrR r� r� rw r ) r� r] r� r� r� r� rY r� r �rjustr )r4 r? rA r� r� � number_widthr� r� � number_styler$ r� �numeralr� r� r* r* r- r� � s �
�zListItem.render_numberNr� rI rg )
r? r rA r r� r� r� r� r( r ) rJ rK rL rP rX rd r> r� r� r* r* r* r- r� x s
r� c @ s$ e Zd Zeddd��Zdd d
�ZdS )�Linkr% r& r' r r( r) c C s |j �dd�}| |jt|��S )N�href�#)r� r� r� r] )r, r% r' �urlr* r* r- r. � s z Link.creater7 r] r� c C s t |�| _|| _d S rZ )r! r7 r� )r4 r7 r� r* r* r- rd � s
z
Link.__init__NrF )r7 r] r� r] )rJ rK rL rN r. rd r* r* r* r- r� � s r� c sL e Zd ZdZdZed d d
��Zd � fdd� Zd � fdd� Zd dd �Z � Z
S )!� ImageItemz#Renders a placeholder for an image.Fr% r& r' r r( r) c C s | t |j�dd��|j�S )z�Factory to create markdown element,
Args:
markdown (Markdown): The parent Markdown object.
token (Any): A token from markdown-it.
Returns:
MarkdownElement: A new markdown element
�srcrr )r] r� r� �
hyperlinksr+ r* r* r- r. � s zImageItem.create� destinationr] r� r< r1 c s || _ || _d | _t� �� d S rZ )r� r� �linkrl rd )r4 r� r� rm r* r- rd � s zImageItem.__init__r/ r0 c s& |j j| _tdd�| _t� �|� d S rS )r^ r� r! r7 rl r5 r3 rm r* r- r5 � s
zImageItem.on_enterr? r rA r r c c sf � t | jp | jp d d�}| jp t| j�d��dd�d �}| jr%|�|� tj d|ddd�}|V d S ) N�r� �/r ���u 🌆 rw rr )�end)
r r� r� r7 r! �strip�rsplitr� r� �assemble)r4 r? rA �
link_style�titler7 r* r* r- rE � s �"
zImageItem.__rich_console__rF )r� r] r� r< r( r1 rG rg ) rJ rK rL rP r$ rN r. rd r5 rE rt r* r* rm r- r� � s r� c @ sL e Zd ZdZ d d dd�Zed dd��Zd!dd�Zd"dd�Zd d d �Z dS )#�MarkdownContextz!Manages the console render state.N�monokair? r rA r rY r �inline_code_lexer�
Optional[str]�inline_code_themer] r( r1 c C sD || _ || _t|�| _t� | _d | _|d ur td||d�| _d S d S )Nrr )r} )r? rA r � style_stackr �stack�_syntaxr )r4 r? rA rY r� r� r* r* r- rd � s
�zMarkdownContext.__init__c C s | j jS )z>Current style which is the product of all styles on the stack.)r� �currentr� r* r* r- r^ � s z MarkdownContext.current_styler7 � node_typec C s\ |dv r$| j dur$| j �|�}|�� | jj�| tj|| jj d�� dS | jj�| |� dS )z#Called when the parser visits text.> �fence� code_inlineNr� )
r� � highlightr� r� �topr9 r! r� r� r� )r4 r7 r� �highlight_textr* r* r- r9 � s �zMarkdownContext.on_textrX �Union[str, Style]c C s" | j j|dd�}| j�|� | jS )zEnter a style context.rR r� )r? r� r� �pushr^ )r4 rX rY r* r* r- rW � s zMarkdownContext.enter_stylec C s | j �� }|S )zLeave a style context.)r� �pop)r4 rY r* r* r- r_ s
zMarkdownContext.leave_style)Nr� ) r? r rA r rY r r� r� r� r] r( r1 )r( r )r7 r] r� r] r( r1 )rX r� r( r )
rJ rK rL rP rd �propertyr^ r9 rW r_ r* r* r* r- r� � s �
r� c @ s� e Zd ZU dZi de�de�de�de�de�de�de �d e �d
e
�d e �d e �d
e
�de�de�de�de�Zded< h d�Z d4d5d'd(�Zd6d+d,�Zd7d2d3�ZdS )8�Markdowna� A Markdown renderable.
Args:
markup (str): A string containing markdown.
code_theme (str, optional): Pygments theme for code blocks. Defaults to "monokai".
justify (JustifyMethod, optional): Justify value for paragraphs. Defaults to None.
style (Union[str, Style], optional): Optional style to apply to markdown.
hyperlinks (bool, optional): Enable hyperlinks. Defaults to ``True``.
inline_code_lexer: (str, optional): Lexer to use if inline code highlighting is
enabled. Defaults to None.
inline_code_theme: (Optional[str], optional): Pygments theme for inline code
highlighting, or None for no highlighting. Defaults to None.
�paragraph_open� heading_openr� �
code_block�blockquote_open�hrr� �ordered_list_open�list_item_open�image�
table_open�
tbody_open�
thead_open�tr_open�td_open�th_openz*ClassVar[Dict[str, Type[MarkdownElement]]]r� > �s�emr� �strongr� NrR T�markupr] rz rV �Optional[JustifyMethod]rY r� r� r< r� r� r� r( r1 c C sP t � �d��d�}|| _|�|�| _|| _|| _|| _|| _|| _ |p$|| _
d S )N�
strikethroughr� ) r
�enabler �parse�parsedrz rV rY r� r� r� ) r4 r rz rV rY r� r� r� �parserr* r* r- rd + s
zMarkdown.__init__�tokens�Iterable[Token]c c sL � |D ] }|j dk}|jdk}|jr |s |s | �|j�E dH q|V qdS )zFlattens the token stream.r� �imgN)r� rj �children�_flatten_tokens)r4 r r' �is_fence�is_imager* r* r- r ? s �
�zMarkdown._flatten_tokensr? r rA r r c c sx � |j | jdd�}|jdd�}t|||| j| jd�}| j}| j}d}t� � }| �
|�D �]�} | j }
| j } | j
dk} | j
dk}
| j
d k}|
d
krO|�| j|
� q+|
d krZ|�d |
� q+|
d
kre|�d|
� q+|
dkr�t| j�dd��}| jr�|j ddd�}|t|d�7 }|�|� q+|j�t�| | �� q+|
dkr�| jr�|�� q+|j�� }t |t�s�J �|j ddd�}|�|� |�|j j |
� |�� |�d|
� |j ddd�}|�|� |�|j |
� |�� |�d|
� q+| |v �r#|
dk�r#|
dk�r#| �r|�d| � �� q+|
�r |�� q+|�d| � �� | j�r |�| j|
� |�� q+| j �| j ��p,t!}|�| | �}| �s9|�rD|j�|� |�"|� |
�rq|j�� }|j �p\|j�o\|jj#�$||�}|�rp|�rf|V |�%||j&�E dH n:|�r�|j�� | j}|du�r�|�||� |j �p�|j�o�|jj#�$||�}|�r�|�r�|V |�%||j&�E dH |
�s�|�r�|�'|� |j(}q+dS )z Render markdown to the console.rR r� N)�height)r� r� Fr r� r r7 � hardbreakr� � softbreakrw � link_openr� rr zmarkdown.link_urlr� �
link_closez
markdown.linkz (�)r� r rk ))r� rY r� r� r� r� r �inlinesr r� r r� rj �nestingr9 r� r] r� r� r� r rW r� r� r� r. r_ r� r\ r7 �plainr� r� rO r5 r� r> �renderrA r: r$ )r4 r? rA rY r/ r �inline_style_tagsr$ �_new_line_segmentr' r� rj �entering�exiting� self_closingr� r� r� �link_url_style�
element_class�
should_renderr7 r* r* r- rE I s� � �
�
��
�
��zMarkdown.__rich_console__)r� NrR TNN)r r] rz r] rV r rY r� r� r< r� r� r� r� r( r1 )r r r( r rg )rJ rK rL rP ra rh ru r� r� r� r� r� r� r� r� r� r� r� rM r# rd r rE r* r* r* r- r� sZ
�������� �
� � �
��� � �
r� �__main__z(Render Markdown to the console with Rich)� description�path�PATHz%path to markdown file, or - for stdin)�metavar�helpz-cz
--force-color� force_color�
store_truez force color for non-terminals)�dest�actionr� r4 z-tz --code-themerz r� zpygments code theme)r7 r� r4 z-iz--inline-code-lexerr� z-yz --hyperlinksr� zenable hyperlinks)r7 r8 r4 z-wz--widthr� z*width of output (default will auto-detect))r� r7 r� r4 z-jz --justifyrV zenable full text justifyz-pz--page�pagezuse pager to scroll output)r �-�rtzutf-8)�encoding�fullrT )rV rz r� r� )�file�force_terminalr� T)r? r� �record)b�
__future__r �sys�typingr r r r r r r � markdown_itr
�markdown_it.tokenr � version_infor �typing_extensions�
rich.tabler rr r �_loopr �_stackr r? r r r r �
containersr �jupyterr �panelr �ruler �segmentr rY r r r� r r7 r! r" r# rO rQ ra rh ru r� r� r� r� r� r� r� r� r� r� r� r� r� rJ �argparse�ArgumentParserr � add_argumentr� �
parse_args�args� rich.consoler1 �stdin�read�
markdown_body�open�
markdown_filerV rz r� r� r% r9 �io�pydoc�StringIO�fileior5 r� �print�pager�getvaluer* r* r* r- �<module> s $
; # # ", (0
;���������
��
�
��