U /eF @sdZddlmZddlZeeZddlZddlZddl Z ddl m Z ddl m Z mZddlmZddlmZmZmZmZmZmZddlmZerdd lmZd d lmZd d lmZd d l m!Z!d dl"m#Z#m$Z$ddl%m&Z&erd dl'm(Z(d dl)m*Z*dZ+ddddddddddddddddZ,ddddddddddddddd d!Z-dddddddd"ddddddd#d$Z.dde#ddd%dddd&ddd'd(d)d*Z/dde#ddd%dddd&dddd(d+d,Z0dde#ddd%dddd&dddd(d-d.Z1e#ddd/dd&dddd0d1d2Z2d3dd4d5d6d7Z3e d8ddd9d:d;d<Z4ddddd=d>d?Z5d3d4d@dAdBZ6d3dCdDdEdFZ7dGZ8dHZ9dIZ:GdJdKdKZ;dKdLdMdNZtdt|}||t|S)a Export the ``LayoutDOM`` object or document as a PNG. If the filename is not given, it is derived from the script name (e.g. ``/foo/myplot.py`` will create ``/foo/myplot.png``) Args: obj (LayoutDOM or Document) : a Layout (Row/Column), Plot or Widget object or Document to export. filename (PathLike, e.g. str, Path, optional) : filename to save document under (default: None) If None, infer from the filename. width (int) : the desired width of the exported layout obj only if it's a Plot instance. Otherwise the width kwarg is ignored. height (int) : the desired height of the exported layout obj only if it's a Plot instance. Otherwise the height kwarg is ignored. webdriver (selenium.webdriver) : a selenium webdriver instance to use to export the image. timeout (int) : the maximum amount of time (in seconds) to wait for Bokeh to initialize (default: 5) (Added in 1.1.1). Returns: filename (str) : the filename where the static file is saved. If you would like to access an Image object directly, rather than save a file to disk, use the lower-level :func:`~bokeh.io.export.get_screenshot_as_png` function. .. warning:: Responsive sizing_modes may generate layouts with unexpected size and aspect ratios. It is recommended to use the default ``fixed`` sizing mode. r#r$driverr&Npngrzunable to save an empty image) rrr#r$ ValueErrorosfspathsaver)r)r"r#r$r%r&imager33/tmp/pip-unpacked-wheel-f5fndrjf/bokeh/io/export.pyrIs&  rz List[str]cCst|||||d}t||dS)a Export a layout as SVG file or a document as a set of SVG files. If the filename is not given, it is derived from the script name (e.g. ``/foo/myplot.py`` will create ``/foo/myplot.svg``) Args: obj (LayoutDOM object) : a Layout (Row/Column), Plot or Widget object to display filename (PathLike, e.g. str, Path, optional) : filename to save document under (default: None) If None, infer from the filename. width (int) : the desired width of the exported layout obj only if it's a Plot instance. Otherwise the width kwarg is ignored. height (int) : the desired height of the exported layout obj only if it's a Plot instance. Otherwise the height kwarg is ignored. webdriver (selenium.webdriver) : a selenium webdriver instance to use to export the image. timeout (int) : the maximum amount of time (in seconds) to wait for Bokeh to initialize (default: 5) Returns: filenames (list(str)) : the list of filenames where the SVGs files are saved. .. warning:: Responsive sizing_modes may generate layouts with unexpected size and aspect ratios. It is recommended to use the default ``fixed`` sizing mode. r+svg)get_svg_write_collectionr)r"r#r$r%r&svgsr3r3r4r|s!rz str | NonecCs8t|||||d}t|dkr,tdgSt||dS)a Export the SVG-enabled plots within a layout. Each plot will result in a distinct SVG file. If the filename is not given, it is derived from the script name (e.g. ``/foo/myplot.py`` will create ``/foo/myplot.svg``) Args: obj (LayoutDOM object) : a Layout (Row/Column), Plot or Widget object to display filename (str, optional) : filename to save document under (default: None) If None, infer from the filename. width (int) : the desired width of the exported layout obj only if it's a Plot instance. Otherwise the width kwarg is ignored. height (int) : the desired height of the exported layout obj only if it's a Plot instance. Otherwise the height kwarg is ignored. webdriver (selenium.webdriver) : a selenium webdriver instance to use to export the image. timeout (int) : the maximum amount of time (in seconds) to wait for Bokeh to initialize (default: 5) (Added in 1.1.1). Returns: filenames (list(str)) : the list of filenames where the SVGs files are saved. .. warning:: Responsive sizing_modes may generate layouts with unexpected size and aspect ratios. It is recommended to use the default ``fixed`` sizing mode. r+rzNo SVG Plots were found.r5)r lenlogwarningr7r8r3r3r4rs "  r)r,r& resourcesr#r$rz Image.Image)r)r,r&r=r#r$r*c Csddlm}t}t||||d}t|jddd} | |W5QRX|dk rV|n|} | | d|jt | |t | \}}} | } W5QRXt t | d d d || || f||fS) a Get a screenshot of a ``LayoutDOM`` object. Args: obj (LayoutDOM or Document) : a Layout (Row/Column), Plot or Widget object or Document to export. driver (selenium.webdriver) : a selenium webdriver instance to use to export the image. timeout (int) : the maximum amount of time to wait for initialization. It will be used as a timeout for loading Bokeh, then when waiting for the layout to be rendered. Returns: image (PIL.Image.Image) : a pillow image loaded from PNG. .. warning:: Responsive sizing_modes may generate layouts with unexpected size and aspect ratios. It is recommended to use the default ``fixed`` sizing mode. rwebdriver_controlr=r#r$wutf-8modeencodingNfile://ZRGBAr)r%r? _tmp_htmlropenpathwritegetZmaximize_windowwait_until_render_complete_maximize_viewportrr ioBytesIOconvertZcropresize) r)r,r&r=r#r$r?tmphtmlfile web_driverZdprr-r3r3r4rs$  rc Csddlm}t}t||||d}t|jddd} | |W5QRX|dk rV|n|} | d|jt| |t t t | t } W5QRX| SNrr>r@rArBrCrF)r%r?rGrrHrIrJrKrLr r r(execute_script _SVG_SCRIPT r)r,r&r=r#r$r?rRrSrTrUr9r3r3r4r6s  r6c Csddlm}t}t||||d}t|jddd} | |W5QRX|dk rV|n|} | d|jt| |t t t | t } W5QRX| SrV)r%r?rGrrHrIrJrKrLr r r(rW _SVGS_SCRIPTrYr3r3r4r s  r r@)r)r=r#r$r*c sxdddfdd }|dk s*|dk rrddlm}t|sLtd n&t|||W5QRSQRX|S)  a%\ {% block preamble %} {% endblock %} r(r*cstddddS)NT)titletemplateZsuppress_callback_warningZ _always_newrr3r)r=r_r3r4rS-szget_layout_html..htmlNrrzhExport method called with width or height argument on a non-Plot model. The size values will be ignored.) models.plotsr isinstancewarningswarn_resized)r)r=r#r$rSrr3r`r4rs   rrNone)r,r&r*c Csddlm}ddlm}ddddd}z|||d d |Wn4|k rt}zt|td |W5d }~XYnX|tdddd d}z.is_bokeh_loadedg?)Zpoll_frequencyz.is_bokeh_render_completezThe webdriver raised a TimeoutException while waiting for a 'bokeh:idle' event to signify that the layout has rendered. Something may have gone wrong.) Zselenium.common.exceptionsrgZselenium.webdriver.support.waitrhZuntil _log_console RuntimeErrorrW _WAIT_SCRIPTr;r<)r,r&rgrhrmernr3r3r4rL;s   rLrzIterator[None])r)r#r$r*ccs>|j}|j}|dk r||_|dk r(||_dV||_||_dSN)r#r$)r)r#r$Z old_widthZ old_heightr3r3r4re_sre)itemsr"extr*c Cs|dkrt|}t|}g}dddddd}t|D]L\}}|dkrL|n|||}t|ddd }||W5QRX||q8|S) Nr(r')nameir*cSst|\}}|d||S)N_)r)rvrwbasenamerur3r3r4_indexedus z#_write_collection.._indexedrrArBrC)rr/r0 enumeraterHrJappend) rtr"ru filenamesrzrwitemfnamefr3r3r4r7ns  r7rjcspdddhz|d}Wntk r.YdSXfdd|D}t|dkrltd|D]}t|q\dS) NWARNINGERRORZSEVEREZbrowsercs$g|]}|dkr|dqS)levelmessage)rK).0r;levelsr3r4 sz _log_console..rzLThere were browser warnings and/or errors that may have affected your export)Zget_log Exceptionr:r;r<)r,Zlogsmessagesrr3rr4ros   rozTuple[int, int, int])rUr*cCsBd}||}d}|j|f|\}}d}||||||S)Nz const root = document.getElementsByClassName("bk-root")[0] const {width, height} = root.children[0].getBoundingClientRect() return [width, height, window.devicePixelRatio] aI const [width, height, dpr] = arguments return [ // XXX: outer{Width,Height} can be 0 in headless mode under certain window managers Math.max(0, window.outerWidth - window.innerWidth) + width*dpr, Math.max(0, window.outerHeight - window.innerHeight) + height*dpr, ] d)rWZset_window_size)rUZcalculate_viewport_sizeZ viewport_sizeZcalculate_window_sizer#r$Zepsr3r3r4rMs rMa const {LayoutDOMView} = Bokeh.require("models/layouts/layout_dom") const {PlotView} = Bokeh.require("models/plots/plot") function* collect_svgs(views) { for (const view of views) { if (view instanceof LayoutDOMView) { yield* collect_svgs(view.child_views.values()) } if (view instanceof PlotView && view.model.output_backend == "svg") { const {ctx} = view.canvas_view.compose() yield ctx.get_serialized_svg(true) } } } const root_views = Object.values(Bokeh.index) return [...collect_svgs(root_views)] a function* export_svgs(views) { for (const view of views) { // TODO: use to_blob() API in future const {ctx} = view.export("svg") yield ctx.get_serialized_svg(true) } } const root_views = Object.values(Bokeh.index) return [...export_svgs(root_views)] z // add private window prop to check that render is complete window._bokeh_render_complete = false; function done() { window._bokeh_render_complete = true; } const doc = Bokeh.documents[0]; if (doc.is_idle) done(); else doc.idle.connect(done); c@seZdZUdZded<ded<ded<dd d ddd d d dZddddZdddd dddZd dddZd dddZ dS) _TempFileFri_closedr'fdr(rIrRr]prefixsuffixrf)rrr*c CsPzt||td\|_|_Wn,ttfk rJt||d\|_|_YnXdS)N)rrdirr)rr/getcwdrrIOSErrorIOError)selfrrr3r3r4__init__sz_TempFile.__init__r\cCs|Srsr3rr3r3r4 __enter__sz_TempFile.__enter__r)excvaluetbr*cCs |dSrsclose)rrrrr3r3r4__exit__sz_TempFile.__exit__cCs |dSrsrrr3r3r4__del__sz_TempFile.__del__cCs`|jr dSzt|jWntk r.YnXzt|jWntk rTYnXd|_dS)NT)rr/rrrunlinkrIrr3r3r4rsz_TempFile.closeN) __name__ __module__ __qualname__r__annotations__rrrrrr3r3r3r4rs   rr\cCs tdddS)NZbokehz.htmlr)rr3r3r3r4rGsrG)=__doc__ __future__rlogging getLoggerrr;rNr/rc contextlibros.pathrrtempfilertypingrrr r r r ZPILr Z#selenium.webdriver.remote.webdriverrZ core.typesrdocumentrembedrr=rrutilrZmodels.layoutsrrar__all__rrrrr6r rrLrer7rorMrZrXrqrrGr3r3r3r4s             3$.*#$  (