U 祡cNP @sDddlmZmZmZmZmZmZmZddlm Z ddl m Z ddl m Z ddlmZddlmZmZmZmZddlmZdd lmZdd lmZdd lmZdd lmZdd lm Z m!Z!ddl"m#Z#ddl$m%Z%m&Z&GdddZ'Gddde'Z(Gddde'Z)Gddde)Z*Gddde)Z+Gddde)Z,Gddde)Z-Gddde'Z.Gd d!d!e'Z/Gd"d#d#e)Z0Gd$d%d%e)Z1Gd&d'd'Z2Gd(d)d)eZ3e4d*kr@dd+l5Z5dd+l6Z6e5j7d,d-Z8e8j9d.d/d0d1e8j9d2d3d4d5d+d6d7e8j9d8d9d:d;dd?d@d+d@d=e8j9dAdBdCd5dDdEe8j9dFdGe:dHd+dIdJe8j9dKdLdMd5dNdEe8j9dOdPdQd5dRdEe8;ZdTkre6j?@ZAn$eBedUdVdWZCeC@ZAW5QRXe3eAer?__doc__r$r$r$r'rCOsrCc@sFeZdZdZdZdddddZdeddd d Zdddd d ZdS) TextElementz)Base class for elements that render text.noner)Nr*cCs||j|_tdd|_dSNleftjustify) enter_style style_namestylerr0r,r$r$r'r.]szTextElement.on_enterr/cCs"|j|t|tr|jnddSN)r0append isinstancestr current_styler1r$r$r'r2aszTextElement.on_textcCs |dSrN) leave_styler,r$r$r'r3dszTextElement.on_leave) r=r>r?rDrLr.rr2r3r$r$r$r'rEXs rEc@sVeZdZUdZdZeed<ededdddZ edd d d Z e e e d d dZdS) Paragraphz A Paragraph.zmarkdown.paragraphrJrrcCs||jp ddSrGrIr%r$r$r'r(nszParagraph.createN)rJr"cCs ||_dSrNrI)r-rJr$r$r'__init__rszParagraph.__init__r8ccs|j|j_|jVdSrN)rJr0r;r$r$r'r<us zParagraph.__rich_console__)r=r>r?rDrLrrArBrr(rUrrrr<r$r$r$r'rThs rTcs`eZdZdZededdddZdddd d Zedd fd d Z e e e dddZ ZS)Headingz A heading.rrcCs||j}|SrN)level)r&r r!headingr$r$r'r(s zHeading.creater)Nr*cCst|_||jdSrN)rr0rKrLr,r$r$r'r.szHeading.on_enter)rWr"cs ||_d||_tdS)Nz markdown.h)rWrLsuperrU)r-rW __class__r$r'rUs zHeading.__init__r8ccsH|j}d|_|jdkr*t|tjddVn|jdkr>tdV|VdS)Ncenterr zmarkdown.h1.border)r rM)r0rJrWrr ZDOUBLEr)r-r9r:r0r$r$r'r<s    zHeading.__rich_console__)r=r>r?rDrBrr(r.intrUrrrr< __classcell__r$r$rZr'rV|srVc@sNeZdZdZdZededdddZeeddd d Z e e e d d d Z dS) CodeBlockz&A code block with syntax highlighting.zmarkdown.code_blockrrcCs(|jpd}|dd}||p d|jS)Nr^ rr0)info partition code_theme)r&r r!Z node_info lexer_namer$r$r'r(s zCodeBlock.createN)rfthemer"cCs||_||_dSrN)rfrg)r-rfrgr$r$r'rUszCodeBlock.__init__r8ccs.t|j}t||j|jddd}|VdS)NTr)rgZ word_wrappadding)rQr0rstriprrfrg)r-r9r:codesyntaxr$r$r'r<szCodeBlock.__rich_console__)r=r>r?rDrLrBrr(rQrUrrrr<r$r$r$r'rasrac@sFeZdZdZdZddddZdded d d Zee e d d dZ dS) BlockQuotezA block quote.zmarkdown.block_quoteNr"cCs t|_dSrNrelementsr-r$r$r'rUszBlockQuote.__init__r)rr4cCs|j|dSNFrorOr6r$r$r'r7s zBlockQuote.on_child_closer8c csb|j|jdd}|j|j||jd}|j}td}td|}|D]}|V|EdH|VqBdS)NwidthrM u▌ )update max_width render_linesrorMr) r-r9r:render_optionslinesrMrrhliner$r$r'r<s  zBlockQuote.__rich_console__) r=r>r?rDrLrUr@r7rrrr<r$r$r$r'rls rlc@s&eZdZdZdZeeedddZdS)HorizontalRulez%A horizontal rule to divide sections.Fr8ccs|jddd}t|dVdS)Nz markdown.hrrFdefaultrv) get_styler)r-r9r:rMr$r$r'r<szHorizontalRule.__rich_console__N) r=r>r?rDrrrrr<r$r$r$r'r~s r~c@s`eZdZdZededdddZeee dddd Z d d e d d dZ e eedddZdS) ListElementzA list element.rrcCs|j}||d|dS)Ntypestart) list_data)r&r r!rr$r$r'r(szListElement.createN) list_type list_startr"cCsg|_||_||_dSrN)itemsrr)r-rrr$r$r'rUszListElement.__init__r)rr4cCst|tst|j|dSrq)rPListItemAssertionErrorrrOr6r$r$r'r7s zListElement.on_child_closer8ccsz|jdkr*|jD]}|||EdHqnL|jdkr8dn|j}|t|j}|jD]"}|||||EdH|d7}qRdS)Nbulletr )rr render_bulletrlen render_number)r-r9r:itemnumber last_numberr$r$r'r<s   zListElement.__rich_console__)r=r>r?rDrBrr(rQrr_rUr@r7rrrr<r$r$r$r'rs rc@s\eZdZdZdZddddZdded d d Zee e d d dZ ee e e e dddZ dS)rzAn item in a list.z markdown.itemNrmcCs t|_dSrNrnrpr$r$r'rUszListItem.__init__r)rr4cCs|j|dSrqrrr6r$r$r'r7 s zListItem.on_child_closer8c cs|j|jdd}|j|j||jd}|jddd}td|}td|}td }t|D]&\} } | rh|n|V| EdH|VqXdS) Nrtrvzmarkdown.item.bulletrFru • z rw)rxryrzrorMrrr ) r-r9r:r{r|Z bullet_stylerrhrfirstr}r$r$r'rs   zListItem.render_bullet)r9r:rrr"ccstt|d}|j|j|d}|j|j||jd}|jddd}td} td||} t| |d d|} t |D]&\} } | r| n| V| EdH| Vq|dS) Nr]rtrvzmarkdown.item.numberrFrrwrbr ) rrQrxryrzrorMrrrjustr )r-r9r:rrZ number_widthr{r|Z number_stylerrhZnumeralrr}r$r$r'rs zListItem.render_number)r=r>r?rDrLrUr@r7rrrrr_rr$r$r$r'rs rcsjeZdZdZdZededdddZee dd fd d Z d dd fdd Z e e edddZZS) ImageItemz#Renders a placeholder for an image.FrrrcCs||j|jSr#) destination hyperlinksr%r$r$r'r(4s zImageItem.createN)rrr"cs ||_||_d|_tdSrN)rrlinkrYrU)r-rrrZr$r'rUAszImageItem.__init__r)r*cs&|jj|_tdd|_t|dSrG)rRrrr0rYr.r,rZr$r'r.Gs  zImageItem.on_enterr8ccs`t|jp|jpdd}|jp6t|jdddd}|jrH||tj d|dddVdS) Nr/r u🌆 rbr^)end) rrrr0rstriprsplitrZstylizeassemble)r-r9r: link_styletitler$r$r'r<Ls " zImageItem.__rich_console__)r=r>r?rDrrBrr(rQr@rUr.rrrr<r`r$r$rZr'r/s rc@sxeZdZdZdeeeeeeddddZ e eddd Z eedd d d Z e eefed ddZedddZdS)r)z!Manages the console render state.Nmonokai)r9r:rMinline_code_lexerinline_code_themer"cCs@||_||_t||_t|_d|_|dk rCurrent style which is the product of all styles on the stack.)rcurrentrpr$r$r'rRjszMarkdownContext.current_style)r0 node_typer"cCsZ|dkrF|jdk rF|j|}||jj|tj||jj dn|jj||dS)z#Called when the parser visits text.rjNrv) rZ highlightrirtopr2rrrr)r-r0rZhighlight_textr$r$r'r2os zMarkdownContext.on_text)rLr"cCs"|jj|dd}|j||jS)zEnter a style context.rFr)r9rrpushrR)r-rLrMr$r$r'rKzs zMarkdownContext.enter_stylecCs|j}|S)zLeave a style context.)rpop)r-rMr$r$r'rSs zMarkdownContext.leave_style)Nr)r=r>r?rDrrrrrQrUpropertyrRr2rrKrSr$r$r$r'r)Vs   r)c @seZdZUdZeeeeee e e dZ e eeeefed<ddddhZdeeeeeeefeeeeed d d dZeeedddZd S)raA 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. )Z paragraphrXZ code_blockZ block_quoteZthematic_breaklistrimageroZemphstrongrjstrikerNrFT)markuprerJrMrrrr"c CsD||_t}|||_||_||_||_||_||_|p<||_ dSrN) rr parseparsedrerJrMrrr) r-rrerJrMrrrparserr$r$r'rUs  zMarkdown.__init__r8ccs|j|jdd}|jdd}t||||j|jd}|j}|j}d}|D]\}} |j } | dkrz| |j dd | qJ| d kr| r| d| qJ| d kr| r| d | qJ| d krN| r|jd dd} |j r| t|jd7} || n`||j s| d| tdd|jddd}||| |j| || d| qJ| |kr|r| rz|d| n|n.|d| |j r| |j | |qJ|j| pt} |r| r| ||} |j| | |n|j} |jrZ|jj|| rN|r.tdV|| |jEdH| |n | |n| ||| |jEdH| j!}qJ| ||} |j| | ||j r| ||j "|j|jj|| r|rtdV|| |jEdH| |n | || j!}qJdS)zRender markdown to the console.rFrN)height)rrF)Z html_inlineZ html_blockr0rwrbZ linebreakZ softbreakrz markdown.linkrz (T)Z underlinezmarkdown.link_url)z markdown.)#rrMrxr)rrrwalkerinlinestr2literalreplacerrrrKrS is_containerrogetrCr(rrr.rrr7rrenderr:r3rri)r-r9r:rMr+ZnodesrrrZenteringrrZ element_classelementr$r$r'r<s                           zMarkdown.__rich_console__)rNrFTNN)r=r>r?rDrTrVrarlr~rrrrorrrQrrrArrrrrr@rUrrrr<r$r$r$r'rs>    r__main__Nz(Render Markdown to the console with Rich) descriptionpathPATHz%path to markdown file, or - for stdin)metavarhelpz-cz --force-color force_color store_truezforce color for non-terminals)destactionrrz-tz --code-themererzpygments code theme)rrrz-iz--inline-code-lexerrz-yz --hyperlinksrzenable hyperlinks)rrrz-wz--widthruz*width of output (default will auto-detect))rrrrz-jz --justifyrJzenable full text justifyz-pz--pagepagezuse pager to scroll output)r-rtzutf-8)encodingfullrH)rJrerr)fileforce_terminalru)rru)StypingrrrrrrrZcommonmark.blocksr r^r Z_loopr _stackr r9rrrrZ containersrZjupyterrZpanelrZrulersegmentrrMrrrkrr0rrrrCrErTrVrarlr~rrrr)rr=argparsesysArgumentParserr add_argumentr_ parse_argsargsZ rich.consolerstdinreadZ markdown_bodyopenZ markdown_filerJrerrr riopydocStringIOZfileiorruprintZpagergetvaluer$r$r$r's$          < $ ",'0