6OneMddlZddlZddlZddlZddlZ ddlmZn #ddlmZYnxYwddlZddlm Z ej ej e Zeej edd5ZejeZdddn #1swxYwYdZ en #e$rdZYnwxYwej ej e Zeej ed5Zd ezZdddn #1swxYwYGd d ejZd Zd ZdZdZ Gdde!Z"Gddej#Z$Gdde$Z%Gdde%Z&Gdde'Z(Gdde(Z)dS)N)StringIO) __version__ labextensionz package.jsonc$dtddgS)Nrname)srcdest)data4lib/python3.11/site-packages/pyviz_comms/__init__.py_jupyter_labextension_pathsrs V    r z notebook.jsz cjeZdZdZgZgZdZdZdZe dZ e dZ e dZ dS) extensionz] Base class for pyviz extensions, which allow defining shared cleanup utilities. NFc tj}||jkt_|t_n#t $rYnwxYwt jj|g|Ri|SN) get_ipythonexecution_count_last_execution_countr_repeat_execution_in_cell ExceptionparamParameterizedFunction__new__)clsargskwargs exec_counts rrzextension.__new__7st $6J3=AZ3ZI //9I + +    D *23HHHHHHHs47 AAc:|j|dSr)_delete_actionsappendractions radd_delete_actionzextension.add_delete_actionBs ""6*****r c:|j|dSr)_server_delete_actionsr"r#s radd_server_delete_actionz"extension.add_server_delete_actionFs "))&11111r c|d}|dkr|jD]}||ddS|dkr|jD]}||ddSdS)ze Processes comm messages to handle global actions such as cleaning up plots. event_typedeleteid server_deleteN)r!r')rmsgr*r$s r_process_comm_msgzextension._process_comm_msgJs &  ! !- " "s4y!!!! " " ? * *4 " "s4y!!!!+ * " "r ) __name__ __module__ __qualname____doc__r!r'rrr classmethodr%r(r/r r rrr's O ! % I I I++[+22[2 " "[ " " "r rz if ((window.PyViz === undefined) || (window.PyViz instanceof HTMLElement)) { window.PyViz = {comms: {}, comm_status:{}, kernels:{}, receivers: {}, plot_index: []} } a* if (!window.PyViz) {{ return; }} var events = []; var receiver = window.PyViz.receivers['{plot_id}']; if (receiver && receiver._partial && receiver._partial.content && receiver._partial.content.events) {{ events = receiver._partial.content.events; }} var value = cb_obj['{change}']; {transform} for (var event of events) {{ if ((event.kind === 'ModelChanged') && (event.attr === '{change}') && (cb_obj.id === event.model.id) && (JSON.stringify(value) === JSON.stringify(event.new))) {{ return; }} }} a9 var plot_id = "{plot_id}"; if ((plot_id in window.PyViz.plot_index) && (window.PyViz.plot_index[plot_id] != null)) {{ var plot = window.PyViz.plot_index[plot_id]; }} else if ((Bokeh !== undefined) && (plot_id in Bokeh.index)) {{ var plot = Bokeh.index[plot_id]; }} if (plot == null) {{ return }} if (plot_id in window.PyViz.receivers) {{ var receiver = window.PyViz.receivers[plot_id]; }} else {{ var receiver = new Bokeh.protocol.Receiver(); window.PyViz.receivers[plot_id] = receiver; }} if ((buffers != undefined) && (buffers.length > 0)) {{ receiver.consume(buffers[0].buffer) }} else {{ receiver.consume(msg) }} const comm_msg = receiver.message; if ((comm_msg != null) && (Object.keys(comm_msg.content).length > 0)) {{ plot.model.document.apply_json_patch(comm_msg.content, comm_msg.buffers) }} aZ function unique_events(events) {{ // Processes the event queue ignoring duplicate events // of the same type var unique = []; var unique_events = []; for (var i=0; i>> with StandardOutput() as stdout: ... print('This gets captured') >>> print(stdout[0]) This gets captured cftj|_txt_|_|Sr)sysstdout_stdoutr _stringioselfs r __enter__zStandardOutput.__enter__s"z &.jj0 T^ r c||j|jt _dSr)extendr;getvalue splitlinesr:r8r9)r=rs r__exit__zStandardOutput.__exit__s; DN++--88::;;;\ r N)r0r1r2r3r>rCr r rr6r6s< """""r r6ceZdZdZdZejdZdZdfd Z ddZ d Z ddgfd Z e d Zed Zd ZxZS)Comma Comm encompasses any uni- or bi-directional connection between a python process and a frontend allowing passing of messages between the two. A Comms class must implement methods send data and handle received message events. If the Comm has to be set up on the frontend a template to handle the creation of the comms channel along with a message handler to process incoming messages must be supplied. The template must accept three arguments: * id - A unique id to register to register the comm under. * msg_handler - JS code which has the msg variable in scope and performs appropriate action for the supplied message. * init_frame - The initial frame to render on the frontend. z@
{init_frame}
z'Unique identifier of this Comm instance)docNc||_||_||_||_d|_t t ||r|ntj j dS), Initializes a Comms object N)r,) _on_msg _on_error _on_stdout_on_open_commsuperrE__init__uuiduuid4hex)r=r,on_msgon_error on_stdouton_open __class__s rrPz Comm.__init__sa !#   dD""b(Fdjll6F"GGGGGr cdS)z, Initializes comms channel. Nr )r=rTs rinitz Comm.init*r cdS, Closes the comm connection Nr r<s rclosez Comm.close/r[r cdS)z, Sends data to the frontend Nr r=r metadatabufferss rsendz Comm.send4r[r c|S)z@ Decode incoming message, e.g. by parsing json. r rr.s rdecodez Comm.decode9s  r c<|jstd|jS)NzComm has not been initialized)rN ValueErrorr<s rcommz Comm.comm@s#z ><== =zr cd} g}||}|dd}|jr[t5}||dddn #1swxYwY|r ||n#YnxYw|rdd|znd}d|d}n#t $r} ||n#YnxYwd}tj tj d}|d dD]+}|\} } } } t| | | } |d j di| z },|d t|jt!| z }|r/dd|z}d||g}d|d}Yd}~nd}~wwxYw|r||d<||dS)zn Decode received message before passing it to on_msg callback if it has been defined. Ncomm_idz rGReady)msg_typecontent i)fnfnamelinez{fname} {fn} L{line} z {type}: {error})typeerrorError)rn traceback)rbr )rgpoprJr6rLjoinrrKrx extract_tbr8exc_infodictformatrur0strrd)r=r.rlr9replyervframesframerslinenorrtext error_kwargss r _handle_msgzComm._handle_msgFs  =F++c""Cggi..G| $%%&LL%%%&&&&&&&&&&&&&&&////$4:AVFKK////rF!(V< > > q!!!! E)#,..*;<,  '&E)  5 !!!!!sgAB6A( B6(A,,B6/A,0B66B B6 BB66 F:CF5CCF55F:NNNNNr)r0r1r2r3 html_templaterStringr, js_templaterPrZr_rdr4rgpropertyrjr __classcell__rXs@rrErEs$M C D D DBK H H H H H H       tR    [ X ,",",",",",","r rEcFeZdZdZdZdZedZdZddgfdZ dS) JupyterCommz~ JupyterComm provides a Comm for the notebook which is initialized the first time data is pushed to the frontend. a function msg_handler(msg) {{ var metadata = msg.metadata; var buffers = msg.buffers; var msg = msg.content.data; if ((metadata.msg_type == "Ready")) {{ if (metadata.content) {{ console.log("Python callback returned following output:", metadata.content); }} }} else if (metadata.msg_type == "Error") {{ console.log("Python failed with the following traceback:", metadata.traceback) }} else {{ {msg_handler} }} }} if ((window.PyViz == undefined) || (!window.PyViz.comm_manager)) {{ console.log("Could not find comm manager") }} else {{ window.PyViz.comm_manager.register_target('{plot_id}', '{comm_id}', msg_handler); }} cddlm}|jrdS||ji|_|j|j|jr|idSdS)Nr)rE) target_namer )ipykernel.commrErNr,rTrrM)r=IPyComms rrZzJupyterComm.inits222222 :  FWr:::  $*+++ =  MM"       r c|ddS)z Decodes messages following Jupyter messaging protocol. If JSON decoding fails data is assumed to be a regular string. ror r rfs rrgzJupyterComm.decodes 9~f%%r cJ|jr|jdSdSr])rNr_r<s rr_zJupyterComm.closes2 :  J         r Ncv|js||j|||dSz1 Pushes data across comm socket. )rbrcN)rNrZrjrdras rrdzJupyterComm.sends;z  IIKKK th@@@@@r ) r0r1r2r3rrZr4rgr_rdr r rrrus K,&&[&tRAAAAAAr rcXeZdZdZdZedZd fd ZdZdZ ddgfdZ xZ S) JupyterCommJSz JupyterCommJS provides a comms channel for the Jupyter notebook, which is initialized on the frontend. This allows sending events initiated on the frontend to python. a ct|dd}d|vr"dt|dD|d<|S)Nror rcci|]\}}|| Sr r ).0ivs r z(JupyterCommJS.decode..s"N"N"NDAq1a"N"N"Nr _buffers)r} enumerate)rr.decodeds rrgzJupyterCommJS.decodesLs9~f-..   "N"NIc)n4M4M"N"N"NGJ r Ncddlm}tt|||||||jj|_|j|j |j dS)rIr)rN) IPythonrrOrrPkernel comm_managermanagerregister_targetr, _handle_open)r=r,rTrUrVrWrrXs rrPzJupyterCommJS.__init__ss (''''' mT""++B)WUUU"{}}+8  $$TWd.?@@@@@r c|jr|jdS|j|jjvr|jj|j=dSt d|jz)r^z"JupyterCommJS %s is already closedN)rNr_r,rtargetsAssertionErrorr<s rr_zJupyterCommJS.closesf : U J       w$,...L(111$%IDG%STTTr c||_|j|j|jr||dSdSr)rNrTrrM)r=rjr.s rrzJupyterCommJS._handle_opensN  $*+++ =  MM#       r c@|j|||dSr)rjrdras rrdzJupyterCommJS.sends$ th@@@@@r r) r0r1r2r3rr4rgrPr_rrdrrs@rrrs K[ AAAAAA U U U tRAAAAAAAAr rcReZdZdZdZiZeZeZe ddZ e ddZ dS) CommManagerzu The CommManager is an abstract baseclass for establishing websocket comms on the client and the server. z function CommManager() { } CommManager.prototype.register_target = function() { } CommManager.prototype.get_client_comm = function() { } window.PyViz.comm_manager = CommManager() NcV||||||}||j|j<|Sr) server_comm_commsr,rrTr,rUrVrWrjs rget_server_commzCommManager.get_server_comm.r68YHH" 47 r cV||||||}||j|j<|Sr) client_commrr,rs rget_client_commzCommManager.get_client_commrr r) r0r1r2r3 js_managerrrErrr4rrr r rrrst JFKK[ [r rceZdZdZdZeZeZdS)JupyterCommManagerak The JupyterCommManager is used to establishing websocket comms on the client and the server via the Jupyter comms interface. There are two cases for both the register_target and get_client_comm methods: one to handle the classic notebook frontend and one to handle JupyterLab. The latter case uses the globally available PyViz object which is made available by each PyViz project requiring the use of comms. This object is handled in turn by the JupyterLab extension which keeps track of the kernels associated with each plot, ensuring the corresponding comms can be accessed. a function JupyterCommManager() { } JupyterCommManager.prototype.register_target = function(plot_id, comm_id, msg_handler) { if (window.comm_manager || ((window.Jupyter !== undefined) && (Jupyter.notebook.kernel != null))) { var comm_manager = window.comm_manager || Jupyter.notebook.kernel.comm_manager; comm_manager.register_target(comm_id, function(comm) { comm.on_msg(msg_handler); }); } else if ((plot_id in window.PyViz.kernels) && (window.PyViz.kernels[plot_id])) { window.PyViz.kernels[plot_id].registerCommTarget(comm_id, function(comm) { comm.onMsg = msg_handler; }); } else if (typeof google != 'undefined' && google.colab.kernel != null) { google.colab.kernel.comms.registerTarget(comm_id, (comm) => { var messages = comm.messages[Symbol.asyncIterator](); function processIteratorResult(result) { var message = result.value; console.log(message) var content = {data: message.data, comm_id}; var buffers = [] for (var buffer of message.buffers || []) { buffers.push(new DataView(buffer)) } var metadata = message.metadata || {}; var msg = {content, buffers, metadata} msg_handler(msg); return messages.next().then(processIteratorResult); } return messages.next().then(processIteratorResult); }) } } JupyterCommManager.prototype.get_client_comm = function(plot_id, comm_id, msg_handler) { if (comm_id in window.PyViz.comms) { return window.PyViz.comms[comm_id]; } else if (window.comm_manager || ((window.Jupyter !== undefined) && (Jupyter.notebook.kernel != null))) { var comm_manager = window.comm_manager || Jupyter.notebook.kernel.comm_manager; var comm = comm_manager.new_comm(comm_id, {}, {}, {}, comm_id); if (msg_handler) { comm.on_msg(msg_handler); } } else if ((plot_id in window.PyViz.kernels) && (window.PyViz.kernels[plot_id])) { var comm = window.PyViz.kernels[plot_id].connectToComm(comm_id); comm.open(); if (msg_handler) { comm.onMsg = msg_handler; } } else if (typeof google != 'undefined' && google.colab.kernel != null) { var comm_promise = google.colab.kernel.comms.open(comm_id) comm_promise.then((comm) => { window.PyViz.comms[comm_id] = comm; if (msg_handler) { var messages = comm.messages[Symbol.asyncIterator](); function processIteratorResult(result) { var message = result.value; var content = {data: message.data}; var metadata = message.metadata || {comm_id}; var msg = {content, metadata} msg_handler(msg); return messages.next().then(processIteratorResult); } return messages.next().then(processIteratorResult); } }) var sendClosure = (data, metadata, buffers, disposeOnDone) => { return comm_promise.then((comm) => { comm.send(data, metadata, buffers, disposeOnDone); }); }; var comm = { send: sendClosure }; } window.PyViz.comms[comm_id] = comm; return comm; } window.PyViz.comm_manager = new JupyterCommManager(); N) r0r1r2r3rrrrrr r rrrs-  PJdKKKKr r)*osr8rQrxjsonrior_versionrpathabspathdirname__file__HEREopenrzfidloadr rr NameError comm_pathfread nb_mime_jsrr PYVIZ_PROXYABORT_JSbokeh_msg_handler JS_CALLBACKlistr6 ParameterizedrErrobjectrrr r rrs  !!!!!!! !!!!!! wrwx0011 T"',,t^^ < <== 49S>>D +----+++ GOOBGOOH55 6 6  T"',,y- 0 011#Q!&&(("J###############/"/"/"/"/"+/"/"/"d  6@P f"""""T"""*p"p"p"p"p"5 p"p"p"f:A:A:A:A:A$:A:A:Az9A9A9A9A9AK9A9A9Az#####&###Nb b b b b b b b b b s>' B((B,/B,7 CC  C %E  E E