o MEgNP@s^ddlmZmZmZmZmZmZmZddlm 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*krdd+l5Z5dd+l6Z6e5j7d,d-Z8e8j9d.d/d0d1e8j9d2d3d4d5d+d6d7e8j9d8d9d:d;dd?d@d+d@d=e8j9dAdBdCd5dDdEe8j9dFdGe:dHd+dIdJe8j9dKdLdMd5dNdEe8j9dOdPdQd5dRdEe8;ZdTkrIe6j?@ZAneBedUdVdW ZCeC@ZAWd+n 1sawYe3eAeeZdZdZdZdddZddd eddfd d Zdd d ZdS) TextElementz)Base class for elements that render text.noner(r)r!NcCs||j|_tdd|_dSNleftjustify) enter_style style_namestylerr/r+r#r#r&r-]szTextElement.on_enterr/cCs(|j|t|tr|jdSddSN)r/append isinstancestr current_styler0r#r#r&r1as(zTextElement.on_textcCs |dSrL) leave_styler+r#r#r&r2d zTextElement.on_leaver:) r;r<r=rBrJr-rr1r2r#r#r#r&rCXs  rCc@s`eZdZUdZdZeed<edddeddfdd Z dedd fd d Z d e de de fddZd S) Paragraphz A Paragraph.zmarkdown.paragraphrHrrr r!cCs||jpddSrErGr$r#r#r&r'nszParagraph.createNcCs ||_dSrLrG)r,rHr#r#r&__init__rs zParagraph.__init__r6r7ccs|j|j_|jVdSrL)rHr/r8r#r#r&r9us  zParagraph.__rich_console__)r;r<r=rBrJrr?r@rr'rTrrrr9r#r#r#r&rShs rScsdeZdZdZedddeddfddZdd d Zd edd ffdd Z de de de fddZ ZS)Headingz A heading.rrr r!cCs||j}|SrL)level)r%rr headingr#r#r&r' zHeading.creater(r)NcCst|_||jdSrL)rr/rIrJr+r#r#r&r-szHeading.on_enterrVcs ||_d||_tdS)Nz markdown.h)rVrJsuperrT)r,rV __class__r#r&rTs zHeading.__init__r6r7ccsL|j}d|_|jdkrt|tjddVdS|jdkr!tdV|VdS)Ncenterr zmarkdown.h1.border)r rK)r/rHrVrr DOUBLEr)r,r6r7r/r#r#r&r9s    zHeading.__rich_console__r:)r;r<r=rBr@rr'r-intrTrrrr9 __classcell__r#r#rZr&rU|s rUc@sZeZdZdZdZedddeddfddZd ed edd fd d Z de de de fddZ d S) CodeBlockz&A code block with syntax highlighting.zmarkdown.code_blockrrr r!cCs(|jpd}|dd}||pd|jS)Nr^ rdefault)info partition code_theme)r%rr node_info lexer_namer#r#r&r's zCodeBlock.createrithemeNcCs||_||_dSrL)rirj)r,rirjr#r#r&rTs zCodeBlock.__init__r6r7ccs8t|j}tt||j|jddtjd}|VdS)Nrjdim) border_styler ) rOr/rstriprrrirjr SQUARE)r,r6r7codesyntaxr#r#r&r9s zCodeBlock.__rich_console__)r;r<r=rBrJr@rr'rOrTrrrr9r#r#r#r&rbsrbc@sJeZdZdZdZdddZddd d defd d Zd ede de fddZ dS) BlockQuotezA block quote.zmarkdown.block_quoter!NcC t|_dSrLrelementsr,r#r#r&rTrRzBlockQuote.__init__r(r)r3rcC|j|dSNFrurMr4r#r#r&r5 zBlockQuote.on_child_closer6r7c csd|j|jdd}|j|j||jd}|j}td}td|}|D] }|V|EdH|Vq"dS)NwidthrK u▌ )update max_width render_linesrurKr) r,r6r7render_optionslinesrKrpaddingliner#r#r&r9s  zBlockQuote.__rich_console__r!N) r;r<r=rBrJrTr>r5rrrr9r#r#r#r&rrs$  rrc@s*eZdZdZdZdededefddZdS) HorizontalRulez%A horizontal rule to divide sections.Fr6r7r!ccs |jddd}t|dVdS)Nz markdown.hrrDrdr~) get_styler)r,r6r7rKr#r#r&r9szHorizontalRule.__rich_console__N) r;r<r=rBrrrrr9r#r#r#r&rsrc@speZdZdZedddeddfddZded ee dd fd d Z d dddde fddZ de dedefddZd S) ListElementzA list element.rrr r!cCs|j}||d|dS)Ntypestart) list_data)r%rr rr#r#r&r'szListElement.create list_type list_startNcCsg|_||_||_dSrL)itemsrr)r,rrr#r#r&rTs zListElement.__init__r(r)r3rcCst|tsJ|j|dSrx)rNListItemrrMr4r#r#r&r5s zListElement.on_child_closer6r7ccs~|jdkr|jD] }|||EdHq dS|jdurdn|j}|t|j}|jD]}|||||EdH|d7}q+dS)Nbulletr )rr render_bulletrlen render_number)r,r6r7itemnumber last_numberr#r#r&r9s    zListElement.__rich_console__)r;r<r=rBr@rr'rOrr`rTr>r5rrrr9r#r#r#r&rs& rc @sheZdZdZdZdddZddd d defd d Zd ede de fddZ d ede de de de f ddZ dS)rzAn item in a list.z markdown.itemr!NcCrsrLrtrvr#r#r&rT rRzListItem.__init__r(r)r3rcCrwrxryr4r#r#r&r5 rzzListItem.on_child_closer6r7c cs|j|jdd}|j|j||jd}|jddd}td|}td|}td }t|D]\} } | r5|n|V| EdH|Vq-dS) Nr|r~zmarkdown.item.bulletrDru • z r)rrrrurKrrr ) r,r6r7rr bullet_stylerrrfirstrr#r#r&rs   zListItem.render_bulletrrccstt|d}|j|j|d}|j|j||jd}|jddd}td} td||} t| |d d|} t |D]\} } | rG| n| V| EdH| Vq?dS) Nr]r|r~zmarkdown.item.numberrDrrrcr ) rrOrrrrurKrrrjustr )r,r6r7rr number_widthrr number_stylerrnumeralrrr#r#r&r s zListItem.render_numberr)r;r<r=rBrJrTr>r5rrrrr`rr#r#r#r&rs.   rcspeZdZdZdZedddeddfdd Zd ed e dd ffd d Z dfdd Z de de defddZZS) ImageItemz#Renders a placeholder for an image.Frrr r!rcCs||j|jSr") destination hyperlinksr$r#r#r&r'6s zImageItem.createrrNcs ||_||_d|_tdSrL)rrlinkrYrT)r,rrrZr#r&rTCszImageItem.__init__r(r)cs&|jj|_tdd|_t|dSrE)rPrrr/rYr-r+rZr#r&r-Is  zImageItem.on_enterr6r7ccsbt|jp |jp dd}|jpt|jdddd}|jr%||tj d|dddVdS) Nr/r u🌆 rcr^)end) rrrr/rstriprsplitrstylizeassemble)r,r6r7 link_styletitler#r#r&r9Ns " zImageItem.__rich_console__r:)r;r<r=rBrr@rr'rOr>rTr-rrrr9rar#r#rZr&r1s rc @seZdZdZ  ddedededeeded df d d Z e d efd d Z deded dfddZ de eefd efddZd efddZdS)r)z!Manages the console render state.Nmonokair6r7rKinline_code_lexerinline_code_themer!cCsD||_||_t||_t|_d|_|dur td||d|_dSdS)Nr^rk)r6r7r style_stackr stack_syntaxr)r,r6r7rKrrr#r#r&rT[s zMarkdownContext.__init__cCs|jjS)z>Current style which is the product of all styles on the stack.)rcurrentrvr#r#r&rPlszMarkdownContext.current_styler/ node_typecCs\|dvr$|jdur$|j|}||jj|tj||jj ddS|jj||dS)z#Called when the parser visits text.rpNr~) r highlightrnrtopr1rrrr)r,r/rhighlight_textr#r#r&r1qs zMarkdownContext.on_textrJcCs"|jj|dd}|j||jS)zEnter a style context.rDr)r6rrpushrP)r,rJrKr#r#r&rI|s zMarkdownContext.enter_stylecCs|j}|S)zLeave a style context.)rpop)r,rKr#r#r&rQrXzMarkdownContext.leave_style)Nr)r;r<r=rBrrrrrOrTpropertyrPr1rrIrQr#r#r#r&r)Xs,  r)c@seZdZUdZeeeeee e e dZ e eeeefed<hdZ      dd ed ed eed eeefd edeedeeddfddZdededefddZdS)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 "python". inline_code_theme: (Optional[str], optional): Pygments theme for inline code highlighting, or None for no highlighting. Defaults to None. ) paragraphrW code_block block_quotethematic_breaklistrimageru>rpemphstrikestrongrNrDTmarkuprgrHrKrrrr!c CsD||_t}|||_||_||_||_||_||_|p||_ dSrL) rr parseparsedrgrHrKrrr) r,rrgrHrKrrrparserr#r#r&rTs  zMarkdown.__init__r6r7ccs|j|jdd}t||||j|jd}|j}|j}d}|D]^\}} |j} | dvr8| |j dd| q | dkrE| rD| d| q | d krR| rQ| d| q | d kr| rp|jd dd} |j rj| t |jd 7} || q ||j s| d | t dd|jddd}||| |j| || d| q | |vr|r| r|d| q |q |d| |j r| |j | |q |j| pt} |r3| r| ||} |j| | |q |j} |jr |jj|| r|r tdV|| |jEdH| |n| |n| ||| |jEdH| j }q | ||} |j| | ||j rQ| ||j !|j|jj|| rw|rgtdV|| |jEdH| |n| || j }q dS)zRender markdown to the console.rDr)rrF) html_inline html_blockr/rrc linebreak softbreakrz markdown.linkrz (T) underlinezmarkdown.link_url)z markdown.N)"rrKr)rrrwalkerinlinestr1literalreplacerrrrIrQ is_containerrugetrAr'rrr-rrr5rrenderr7r2rrn)r,r6r7rKr(nodesrrrenteringrr element_classelementr#r#r&r9s                          zMarkdown.__rich_console__)rNrDTNN)r;r<r=rBrSrUrbrrrrrrrurrrOrrr?rrrrrr>rTrrrr9r#r#r#r&rsT    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)destactionrdrz-tz --code-themergrzpygments code theme)rrdrz-iz--inline-code-lexerrz-yz --hyperlinksrzenable hyperlinks)rrrz-wz--widthr}z*width of output (default will auto-detect))rrrdrz-jz --justifyrHzenable full text justifyz-pz--pagepagezuse pager to scroll output)r-rtzutf-8)encodingfullrF)rHrgrr)fileforce_terminalr})rr})Stypingrrrrrrrcommonmark.blocksr r^r _loopr _stackr r6rrrr containersrjupyterrpanelrrulersegmentrrKrrrqrr/rrrrArCrSrUrbrrrrrrr)rr;argparsesysArgumentParserr add_argumentr` parse_argsargs rich.consolerstdinread markdown_bodyopen markdown_filerHrgrrrrpydocioStringIOrr}printpagerrgetvaluer#r#r#r&s$          < $ ",'0