[A1e.H ddlmZddlZddlZddlmZddlmZmZm Z ddl m Z ddl m Z ddlmZmZmZdd lmZdd lmZ dd lmZdd ZGd deZGddeZGddeZGddeZy#e$rdZY:wxYw)) b64decodeN)warn)QtCoreQtGui QtWidgets)Bool) ClassNotFound)save_svgsvg_to_clipboard svg_to_image) JupyterWidget) get_colors) latex_to_pngc6tjj|s tj||ytjj|std|zy#t$r(}|j t j k7rYd}~yd}~wwxYw)zensure that a directory exists If it doesn't exists, try to create it and protect against a race condition if another process is doing the same. The default permissions are 755, which differ from os.makedirs default of 777. )modeNz %r exists but is not a directory) ospathexistsmakedirsOSErrorerrnoEEXISTisdirIOError)rres ?/usr/lib/python3/dist-packages/qtconsole/rich_jupyter_widget.py_ensure_dir_existsrsy 77>>$   KK4 (WW]]4 84?@@! ww%,,&' sA'' B0BBceZdZdZy) LatexErrorzException for Latex errorsN__name__ __module__ __qualname____doc__rr r )s$r'r ceZdZdZy)RichIPythonWidgetz4Dummy class for config inheritance. Destroyed below.Nr!r&r'rr)r)-s>r'r)ceZdZdZdZedZdZfdZfdZ fdZ dZ fdZ fd Z d Zd Zdd Zdd ZddZddZdZdZdZddZddZddZddZdZddZxZS)RichJupyterWidgetz An JupyterWidget that supports rich text, including lists, images, and tables. Note that raw performance will be reduced compared to the plain text version. z0ipykernel.pylab.backend_payload.add_plot_payloadFcd|d<t||i||j|j_i|_dt jjv|_ y)z% Create a RichJupyterWidget. richkindjpegN) super__init___get_image_tag_html_exporter image_tag_name_to_svg_mapr QImageReadersupportedImageFormats_jpg_supported)selfargskw __class__s rr1zRichJupyterWidget.__init__Cs^6  $%"%)-(;(;%!# %(:(:(P(P(RRr'c0d|_t| y)z Shows a dialog to export HTML/XML in various formats. Overridden in order to reset the _svg_warning_displayed flag prior to the export running. FN)_svg_warning_displayedr0 export_html)r9r<s rr?zRichJupyterWidget.export_htmlYs ',# r'c,jj|j}|jtj j rtj}|jdfd|jdfd|jjjd;|j|jdfd|jdfd |St9|}|S) zC Reimplemented to return a custom context menu for images. z Copy Imagec&jSN) _copy_imagenamer9srz6RichJupyterWidget._context_menu_make..os1A1A$1Gr'zSave Image As...c&jSrB) _save_imagerDsrrFz6RichJupyterWidget._context_menu_make..pst7G7G7Mr'NzCopy SVGctSrB)r )svgsrrFz6RichJupyterWidget._context_menu_make..vs 3CC3Hr'zSave SVG As...c0tjSrB)r _control)r9rJsrrFz6RichJupyterWidget._context_menu_make..xsxT]]'Cr')rLcursorForPosition charFormatstringPropertyr QTextFormat ImageNamerQMenu addAction addSeparatorr5getr0_context_menu_make)r9posformatmenurErJr<s` @@rrVz$RichJupyterWidget._context_menu_makegs005@@B$$U%6%6%@%@A ??4(D NN<)G H NN-/M N    ''++D$7C!!#z+HI/CE 7-c2D r'c|j|jd|j|j|d|jddy)ztAppend the Out[] prompt and make the output nicer Shared code for some the following if statement T N)_append_plain_text output_sep _append_html_make_out_prompt)r9msg prompt_numbers r_pre_image_appendz#RichJupyterWidget._pre_image_appendsC 6 $// >E d+r'cN|jjd|jdd|j|r|j |d}|jdd}|d}|dd}d|vrD|j |||j |dd |j|jd y d |vrp|j ||t|d jd }|j|d |jd d  |j|jd y d|vr||jrp|j ||t|djd }|j|d |jdd  |j|jd y d|vrE|j || |j|dd |j|jd y t |I|Sy #t$rt |E|cYSwxYw)/Overridden to handle rich data types, like SVG.zexecute_result: %scontentexecution_countrdatametadata image/svg+xmlT image/pngasciiNri image/jpeg text/latex)logdebugrUinclude_outputflush_clearoutputrb _append_svgr^ output_sep2rencode _append_pngr8 _append_jpg _append_latexr r0_handle_display_data_handle_execute_result) r9r`rerarhripngjpgr<s rr{z(RichJupyterWidget._handle_execute_results +SWWY-CD   s #  " " $)nG#KK(91=M6?D9~j1H$&&&sM:  o!6=!!$"2"2D9$&&sM:[ 1 8 8 AB  dX\\+BF6H I!!$"2"2D9%$*=*=&&sM:\ 2 9 9' BC  dX\\,BF6H I!!$"2"2D9%&&sM:=&&tL'94@!!$"2"2D9w5c::? $4"= 77<<=sH H$#H$c|jjd|jdd|j|r0|j |dd}|dd}|jjd|jddd|vr|d}|j |dy d |vrBt |d jd }|j|d|jd d  y d |vrN|jrBt |d jd }|j|d|jd d  y d|vrtr |j|ddy t|=|Sy #t$rt|=|cYSwxYw)rdzdisplay_data: %srerfrhriz display: %srjTrkrlNrmrnro)rprqrUrrrsrtrrvrwr8rxrryr r0rz)r9r`rhrirJr|r}r<s rrzz&RichJupyterWidget._handle_display_datas )3779b+AB   s #  " " $y>&)D9~j1H HHNN=#'')R*@ A$&?+  d+$ [ 1 8 8 AB  dX\\+t5T U%$*=*=\ 2 9 9' BC  dX\\,PT5U V%,=&&tL'94@ w3C883 $*"= 77<<=s E11F  F cdddg}gd}|Dcgc]}|dz }}||z|z}|Dcgc]!}dj|j#}}gd}||zD]L\} } |t| t|  } | | vs| | vry|j| s:|j | sLyycc}wcc}w) zs Determine if a Latex string is in math mode This is the only mode supported by qtconsole math displaymath)equationeqnarraymultlinegatheralignflalignalignat*z\begin{{{0}}} \end{{{0}}}))z\[z\])z\(z\))$$r)$rFT)rXsplitlen startswithendswith) r9latex basic_envs starable_envsenv star_envsenvs env_syntax math_syntaxstartendinners r_is_latex_mathz RichJupyterWidget._is_latex_maths m, / *783S3Y8 8M)I5RVW3299#>DDFW W & 2 JE3#e*c#hY/E~&5>>#+>   !9Xs B(&B-cn t|j|S#t$rtd|cYSwxYw)z4Get color from the current syntax style if loadable.default)r syntax_styler )r9colors r _get_colorzRichJupyterWidget._get_colors> 0d//07 7 0i(/ / 0s 44c@d}|j|rt|dd|jd}|A|jdr0|j dr t|dd|jd}|r|j |||yt#t $rY'wxYw)z! Append latex data to the widget.NFdvipngfgcolor)wrapbackendrr matplotlib)rrrrr Exceptionrwr )r9r before_promptrir|s rryzRichJupyterWidget._append_latexs   u %u5(%)__Y%?AC ;5++C0U^^C5H "5ul)-)CE    S- :    sB BBcB|j|j|||y)z# Append raw JPG data to the widget.rmN)_append_custom _insert_jpg)r9r}rris rrxzRichJupyterWidget._append_jpg s D,,c=8Tr'cB|j|j|||y)z, Append raw PNG data to the widget. rmN)r _insert_png)r9r|rris rrwzRichJupyterWidget._append_png s! D,,c=8Tr'c>|j|j||y)z, Append raw SVG data to the widget. N)r _insert_svg)r9rJrs rrtzRichJupyterWidget._append_svgs D,,c=Ar'c2|jj}t|j}|j t j jtj||t j}|j||S)zr Adds the specified QImage to the document and returns a QTextImageFormat that references it. ) rLdocumentstrcacheKey addResourcer QTextDocument ImageResourcerQUrlQTextImageFormatsetName)r9imagerrErXs r _add_imagezRichJupyterWidget._add_imagesp==))+5>>#$U00>>#[[. 7'')t r'c|j|}tjjj |y)z@ Copies the ImageResource with 'name' to the clipboard. N) _get_imager QApplication clipboardsetImage)r9rErs rrCzRichJupyterWidget._copy_image#s/%((*33E:r'c|jj}|jtjj t j|}|S)zE Returns the QImage stored as the ImageResource with 'name'. )rLrresourcerrrrr)r9rErrs rrzRichJupyterWidget._get_image)sE==))+!!%"5"5"C"C"(++d"35 r'c j|dvre |j|jd}|ot|tj j |}|j|d|jdd|drd|d|jdd|d Sy tj}tj|}|jtjj|j||j|jd |d t!j"d dt%|j'j)j+dS|dk(rE t%|j,|jd}|j9d} | dkDsJ|| dSy#t$rd|jdzcYSwxYw#t$rP|j.sAt0j2j5|ddt0j2j6d|_YywxYw)ag Return (X)HTML mark-up for the image-tag given by match. Parameters ---------- match : re.SRE_Match A match to an HTML image tag as exported by Qt, with match.group("Name") containing the matched image ID. path : string|None, optional [default None] If not None, specifies a path to which supporting files may be written (e.g., for linked images). If None, all images are to be included inline. format : "png"|"svg"|"jpg", optional [default "png"] Format for returned or referenced images. )r|r}rEzCouldn't find image %sNz/qt_img.PNGz zCouldn't save image!zrJzError converting PNG to SVG.zCannot convert PNG images to SVG, export with PNG figures instead. If you want to export matplotlib figures as SVG, add to your ipython config: c.InlineBackend.figure_format = 'svg' And regenerate the figures.TzCannot convert PNG images to SVG. You must export this session with PNG images. If you want to export matplotlib figures as SVG, add to your config c.InlineBackend.figure_format = 'svg' and regenerate the figures.zUnrecognized image format)rgroupKeyErrorrrrbasenamesaver QByteArrayQBufferopen QIODevice WriteOnlyuppercloseresubrtoBase64rhdecoder5r>r QMessageBoxwarningOkfind) r9matchrrXrrelpathbabuffer_rJoffsets rr2z RichJupyterWidget._get_image_tag1s" ] " M F(;<"4(''**40::$ F8KVT#%;BE..FFc t|}|j|}||j|j<|j |j ||j y#t $r|j|dYywxYw)z- Insert raw SVG data into the widet. zReceived invalid SVG data.N)r rr5rErrrr)r9rrJrrXs rrzRichJupyterWidget._insert_svgs ! %E__U+F36D ! !&++- 0      v &     J  # #F,H I Js A--B  B ctj|jd}|jtjj|j |j |j|d|j d|jr7|jd}|j|}|j||yy)z@ Shows a save dialog for the ImageResource with 'name'. z Save Imagez file (*.)rN) r QFileDialogrL setAcceptMode AcceptSavesetDefaultSuffixlower setNameFilterexec_ selectedFilesrr)r9rErXdialogfilenamers rrHzRichJupyterWidget._save_images&&t}}lCY22==> /HI <<>++-a0HOOD)E JJx ( r')FN)F)Nr|rB)r)r"r#r$r%_payload_source_plotrr8r>r1r?rVrbr{rzrrryrxrwrtrrCrr2rrrrrH __classcell__r<s@rr+r+1s N%[N# S,2,";H9B60.UU B ; G6R@@ !4 ! )r'r+c"eZdZdZfdZxZS)r)z(Deprecated class. Use RichJupyterWidget.cDtdtt| |i|y)Nz6RichIPythonWidget is deprecated, use RichJupyterWidget)rDeprecationWarningr0r1)r9ar;r<s rr1zRichIPythonWidget.__init__s# E  ! !"r"r')r"r#r$r%r1rrs@rr)r)s2##r')i)base64rrrwarningsrqtpyrrr traitletsr pygments.utilr qtconsole.svgr r r jupyter_widgetrstylesrIPython.lib.latextoolsr ImportErrorrrr r)r+r&r'rrs ))'BB)3 A"%%? ?D))D)R #)#M LsA99BB