o tf;@sUdZddlZddlZddlZddlmZddlmZddlm Z ddl m Z m Z m Z mZmZmZmZmZmZmZmZmZmZddlmZmZdd lmZdd lm Z d d l!m"Z"m#Z#dd l$m%Z%m&Z&m'Z'ddl(Z(e()e Z*ddZ+e%,ddZ-e&,ddZ.e+dddZ/iZ0ej1e2dfe3d<ddZ4ddZ5e5e4dZ6e7e8e9fZ:ddZ;ddZGd#d$d$e Z?Gd%d&d&e?Z@d'd(ZAGd)d*d*ZBeBZCd+d,ZDGd-d.d.eEZFGd/dde?ZGdS)0zqBase Widget class. Allows user to create widgets in the back-end that render in the Jupyter notebook front-end. N)contextmanager)Iterable) get_ipython) Any HasTraitsUnicodeDictInstanceListIntSetBytesobservedefault Container Undefined)loadsdumpscomm)standard_b64encode) deprecation _get_frame)__protocol_version____control_protocol_version__ __jupyter_widgets_base_version__cCs$|tjvrtj|dvSt|S)aBReturn True if the given environment variable is turned on, otherwise False If the environment variable is set, True will be returned if it is assigned to a value other than 'no', 'n', 'false', 'off', '0', or '0.0' (case insensitive). If the environment variable is not set, the default value is returned. )nonfalseoff0z0.0)osenvironlowerbool)namerr(b/var/www/html/software/conda/envs/catlas/lib/python3.10/site-packages/ipywidgets/widgets/widget.pyenvsets r*.JUPYTER_WIDGETS_ECHOTrWidget _instancescsXt|trfdd|DSt|ttfr fdd|DSt|tr*d|jS|S)Nci|] \}}|t|qSr(_widget_to_json.0kvobjr(r) 2z#_widget_to_json..cg|]}t|qSr(r1r4r6r7r(r) 4z#_widget_to_json.. IPY_MODEL_) isinstancedictitemslisttupler.model_idxr8r(r7r)r20s   r2csxt|trfdd|DSt|ttfr fdd|DSt|tr:|dr:|ddtvr:t|ddS|S)Ncr0r(_json_to_widgetr3r7r(r)r9<r:z#_json_to_widget..cr;r(rHr<r7r(r)r=>r>z#_json_to_widget..r? )r@rArBrCrDstr startswithr/rFr(r7r)rI:s $rI) from_jsonto_jsoncCsBt||D]\}}|}|ddD]}||}q|||d<qdS)zThe inverse of _remove_buffers, except here we modify the existing dict/lists. Modifying should be fine, since this is used when state comes from the wire. N)zip)state buffer_pathsbuffersZ buffer_pathbufferr8keyr(r(r) _put_buffersKs  rVc CsZt|ttfrUd}t|D]E\}}t|tr0|st|}d}d||<|||||gq t|tttfrRt|||g||}||urR|sNt|}d}|||<q |St|trd}|D]D\}}t|tr|sqt|}d}||=|||||gq`t|tttfrt|||g||}||ur|st|}d}|||<q`|St d|)z!For internal, see _remove_buffersFTNz+expected state to be a list or dict, not %r) r@rCrD enumerate _binary_typesappendrA_separate_buffersrB ValueError) ZsubstatepathrRrSZ is_clonedir6Zvnewr5r(r(r)rZWsN      rZcCs"gg}}t|g||}|||fS)aReturn (state_without_buffers, buffer_paths, buffers) for binary message parts A binary message part is a memoryview, bytearray, or python 3 bytes object. As an example: >>> state = {'plain': [0, 'text'], 'x': {'ar': memoryview(ar1)}, 'y': {'shape': (10,10), 'data': memoryview(ar2)}} >>> _remove_buffers(state) ({'plain': [0, 'text']}, {'x': {}, 'y': {'shape': (10, 10)}}, [['x', 'ar'], ['y', 'data']], [, ]) )rZ)rQrRrSr(r(r)_remove_bufferss  r^cCsZt|t|kr dS||krdSt||D]\}}t|dt|dkr*dSqdS)zCompare two lists of buffers for equality. Used to decide whether two sequences of buffers (memoryviews, bytearrays, or python 3 bytes) differ, such that a sync is needed. Returns True if equal, False if unequal FTB)lenrP memoryviewcast)abiaibr(r(r)_buffer_list_equalsrgc@s(eZdZdZedZedddZdS)LoggingHasTraitszA parent class for HasTraits that log. Subclasses have a log trait, and the default behavior is to get the logger from the currently running Application. zlogging.LoggerlogcCsddlm}|S)Nr)ri) traitletsriZ get_logger)selfrir(r(r) _log_defaults zLoggingHasTraits._log_defaultN)__name__ __module__ __qualname____doc__r rirrlr(r(r(r)rhs rhc@s(eZdZdZeZddZdddZdS) CallbackDispatcherz1A structure for registering and running callbacksc Osd}|jD]<}z ||i|}Wn(ty8}zt}|dur*|jjd||ddn|WYd}~qd}~ww|dur?|n|}q|S)z%Call all of the registered callbacks.NzException in callback %s: %sTexc_info) callbacks Exceptionrriwarning showtraceback)rkargskwargsvaluecallbackZ local_valueeipr(r(r)__call__s zCallbackDispatcher.__call__FcCsD|r||jvr|j|dS|s||jvr |j|dSdSdS)z(Un)Register a callback Parameters ---------- callback: method handle Method to be registered or unregistered. remove=False: bool Whether to unregister the callback.N)rtremoverYrkr{rr(r(r)register_callbacks  z$CallbackDispatcher.register_callbackNF)rmrnrorpr rtr~rr(r(r(r)rqs rqcsfdd}|S)z decorator for showing tracebacksc szz |g|Ri|WSty<}z$t}|dur&|jjd|ddn |WYd}~dSWYd}~dSd}~ww)Nz!Exception in widget method %s: %sTrr)rurrirvrw)rkrxryr|r}methodr(r)msz_show_traceback..mr()rrr(rr)_show_tracebacks  rc@s,eZdZddZddZddZddZd S) WidgetRegistrycCs i|_dSN) _registryrkr(r(r)__init__ zWidgetRegistry.__init__c CsJ|j|i}||i}||i}||i}||i} || |<dS)zRegister a valueN)r setdefault) rk model_moduleZmodel_module_version_range model_name view_moduleZview_module_version_range view_nameklass model_version view_versionr(r(r)registers      zWidgetRegistry.registerc CsF|j|}t|d}||} | |} t| d} | |} | S)z Get a valuer)rrCvalues) rkrmodel_module_versionrrZview_module_versionrZmodule_versionsZ model_namesZ view_modulesZ view_versionsZ view_names widget_classr(r(r)gets zWidgetRegistry.getc cst|jD]F\}}t|D];\}}t|D]0\}}t|D]%\}}t|D]\} } t| D]\} } ||||| | f| fVq:q0q&qqqdSr)sortedrrB) rkrmmrmvrvmrvvrZvnrwidgetr(r(r)rBszWidgetRegistry.itemsN)rmrnrorrrrBr(r(r(r)rs  rc CsF|}t|dj|dj|dj|dj|dj|dj||S)z>A decorator registering a widget class in the widget registry. _model_module_model_module_version _model_name _view_module_view_module_version _view_name)Z class_traitsrr default_value)rwr(r(r)rs rc@seZdZddZddZdS)_staticpropertycCs ||_dSrfget)rkrr(r(r)r$rz_staticproperty.__init__cCs|dusJ|Srr)rkZ owner_selfZ owner_clsr(r(r)__get__'s z_staticproperty.__get__N)rmrnrorrr(r(r(r)r#s rcseZdZdZdZeddZeddZeddZedd Z e d d Z e d d Z e ddZe ddZe ddZe ddZe dmddZdnddZddZedddd jdd!Zed"d#dd jdd!Zeed$dd jdd!Zeddd%d&jdd!Zeddd'd&jdd!Zed(d)d*jdd!Zeddd+d&jdd!Ze dd,Z!e"d-d*Z#e$d.d/d0Z%e&Z'dZ(e)Z*e+e,d1Z-fd2d3Z.d4d5Z/d6d7Z0d8d9Z1d:d;Z2e3d<d=d>Z4e5d?d@Z6dAdBZ7dodCdDZ8dpdEdFZ9dGdHZ:dIdJZ;dKdLZfdQdRZ?fdSdTZ@dUdVZAeBdWdXZCeBdYdZZDd[d\ZEeFd]d^ZGd_d`ZHe dadbZIe dcddZJdedfZKdodgdhZLdidjZMdkdlZNZOS)qr.NcC,td}|jjtkr|jjdvstdtS)Nr getmembersZsetup_instanceZ setup_classzWidget.widgets is deprecated.rf_code co_filenameTRAITLETS_FILEco_namerr/framer(r(r)widgets4zWidget.widgetscCr)Nrrz%Widget._active_widgets is deprecated.rrr(r(r)_active_widgets>rzWidget._active_widgetscCr)Nrrz#Widget._widget_types is deprecated.rrrrrrrrr(r(r) _widget_typesHrzWidget._widget_typescCr)Nrrz"Widget.widget_types is deprecated.rrr(r(r) widget_typesRrzWidget.widget_typescCsttD]}|qdSr)rCr/rclose)clsrr(r(r) close_all\s zWidget.close_allcCs |t_dS)zRegisters a callback to be called when a widget is constructed. The callback must have the following signature: callback(widget)N)r._widget_construction_callback)r{r(r(r)on_widget_constructedas zWidget.on_widget_constructedcCs*tjdurttjrt|dSdSdS)3Static method, called when a widget is constructed.N)r.rcallable)rr(r(r)_call_widget_constructediszWidget._call_widget_constructedcCsN|didd}|ddtkrtd|tf||_|j|jdS)z Class method, called when the comm-open message on the "jupyter.widget.control" comm channel is received metadataversionr+rzWIncompatible widget control protocol versions: received version %r, expected version %rN)rsplitCONTROL_PROTOCOL_VERSION_MAJORr[r _control_common_msg_handle_control_comm_msg)rrmsgrr(r(r)handle_control_comm_openedos z!Widget.handle_control_comm_openedc Cs|jdur td|dd}|d}|dkrO|t}i}d}|D]}|j|j|j|j|dd||j <q%t |\}}} |jj t d ||d | d dStd |) Nz)Control comm has not been properly openedcontentdatarZrequest_statesF drop_defaults)rrrrQZ update_states)rZstatesrRrSzAUnknown front-end to back-end widget control msg with method "%s") r RuntimeErrorget_manager_stater/rrrr get_staterEr^sendrA) rrrrrZ full_staterrrRrSr(r(r)r|s0    zWidget._handle_control_comm_msgc Cs|didd}|ddtkrtd|tf|dd}|d }t|d |d |d |d |d|d}||d}d|vrNt||d|d||dS)rrrrr+rzOIncompatible widget protocol versions: received version %r, expected version %rrrrQrrrrrrrrRrSN)rrPROTOCOL_VERSION_MAJORr[rrrV set_state)rrrrrQrrr(r(r)handle_comm_openeds    zWidget.handle_comm_openedFcCs<i}|dur t}|D] }|j|d||j<q dd|dS)zReturns the full state for a widget manager for embedding :param drop_defaults: when True, it will not include default value :param widgets: list with widgets to include in the state (or all widgets when None) :return: Nrrr) version_major version_minorrQ)r/r_get_embed_staterE)rrrQrr(r(r)rs  zWidget.get_manager_statecCsX|j|j|jd}t|j|d\}}}||d<t|dkr*ddt||D|d<|S)N)rrrrrQrcSs&g|]\}}d|t|ddqS)base64ascii)encodingr\r)rdecode)r4pdr(r(r)r=s   z+Widget._get_embed_state..rS)rrrr^rr`rP)rkrrQZ model_staterRrSr(r(r)rs  zWidget._get_embed_statecCstdd|jdS)NrrrrrE)rA _model_idrr(r(r) get_view_specszWidget.get_view_specZ WidgetModelzName of the model.T)helpZ read_onlysyncz@jupyter-widgets/basezThe namespace for the model.z@A semver requirement for namespace version containing the model.zName of the view.) allow_nonerzThe namespace for the view.rzCA semver requirement for the namespace version containing the view.)rzEXPERIMENTAL: The number of views of the model displayed in the frontend. This attribute is experimental and may change or be removed in the future. None signifies that views will not be tracked. Set this to 0 to start tracking view creation/deletion.)rzThe traits which are synced.keyscCsdd|jddDS)NcSsg|]}|qSr(r()r4r'r(r(r)r=sz(Widget._default_keys..Tr)traitsrr(r(r) _default_keysszWidget._default_keysr(c s6|dd|_tjdi|t||dS)zPublic constructorrENr()poprsuperrr.ropen)rkry __class__r(r)rs  zWidget.__init__cCtdNz8Widgets cannot be copied; custom implementation requiredNotImplementedErrorrr(r(r)__copy__zWidget.__copy__cCrrr)rkmemor(r(r) __deepcopy__rzWidget.__deepcopy__cCs |dS)zObject disposalN)rrr(r(r)__del__s zWidget.__del__cCsd|jdur0t|\}}}td||d|dtid}|jdur%|j|d<tjdi||_dSdS)z6Open a comm to the frontend if one isn't already open.Nzjupyter.widget)rQrRr) target_namerrSrcomm_idr()rr^rrArrZ create_comm)rkrQrRrSrxr(r(r)r s   z Widget.openrcCs4|ddurdS|j|_|j|j|t|j<dS)z Called when the comm is changed.newN)rErrr _handle_msgr/)rkchanger(r(r) _comm_changeds zWidget._comm_changedcCs|jjS)zmGets the model id of this widget. If a Comm doesn't exist yet, a Comm will be created automagically.)rrrr(r(r)rE#szWidget.model_idcCs6|jdurt|jd|jd|_d|_dSdS)zClose method. Closes the underlying comm. When the comm is closed, all of the widget views are automatically removed from the front-end.N)rr/rrEr_repr_mimebundle_rr(r(r)r.s   z Widget.closecCst|j|d}t|dkr8|jr"|D]\}}||jvr!||j|<qt|\}}}d||d}|j||ddSdS)aSends the widget state, or a piece of it, to the front-end, if it exists. Parameters ---------- key : unicode, or iterable (optional) A single property's name or iterable of property names to sync with the front-end. rUrupdaterrQrRrN)rr`_property_lockrBr^_send)rkrUrQr'rzrRrSrr(r(r) send_state:s     zWidget.send_statec Cs|dur|j}nt|tr|g}n t|tr|}ntdi}|}|D]!}||d|j}|t|||}|rB| |||j sF|||<q%|S)a]Gets the widget state, or a piece of it. Parameters ---------- key : unicode or iterable (optional) A single property's name or iterable of property names to get. Returns ------- state : dict of states metadata : dict metadata for each field: {key: metadata} Nz2key must be a string, an iterable of keys, or NonerN) rr@rKrr[rtrait_metadata_trait_to_jsongetattr_comparer) rkrUrrrQrr5rNrzr(r(r)rMs   zWidget.get_statecCs|jjdko |jjdkS)NZndarraynumpy)rrmrn)rkrGr(r(r) _is_numpylszWidget._is_numpycCs0||s ||rddl}|||S||kS)Nr)r r Z array_equal)rkrcrdnpr(r(r)r os zWidget._comparec Cstr4i}|D]\}}||jvr|j|dddr|||<q|r4t|\}}}d||d}|j||d|jdi|B|%|D]}||jvr^||d|j} | || |||qDWdn1siwYWddSWddS1swYdS) z3Called when a state is received from the front-end.Z echo_updateTr-rrrMNr() r,rBrrr^r_lock_propertyZhold_trait_notifications_trait_from_jsonZ set_trait) rkZ sync_dataZ echo_stateattrrzZecho_buffer_pathsZ echo_buffersrr'rMr(r(r)rvs. PzWidget.set_statecCs|jd|d|ddS)aSends a custom msg to the widget model in the front-end. Parameters ---------- content : dict Content of the message to send. buffers : list of binary buffers Binary buffers to send with message custom)rrrN)rrkrrSr(r(r)rs z Widget.sendcCs|jj||ddS)a=(Un)Register a custom msg receive callback. Parameters ---------- callback: callable callback will be passed three arguments when a message arrives:: callback(widget, content, buffers) remove: bool True if the callback should be unregistered.)rN)_msg_callbacksrrr(r(r)rs z Widget.on_msgc sHtjdi||D]\}}|dr!|j|||q dS)z/Dynamically add trait attributes to the Widget.rNr()r add_traitsrB get_metadatarrYr)rkrr'traitrr(r)rs   zWidget.add_traitscs\|d}|jdur&t|jdddur&||jvr&||t||r&|j|dt|dS)z#Called when a property has changed.r'NkernelTr)rr r_should_send_propertyrr notify_change)rkrr'rr(r)rs  zWidget.notify_changecCs||Sr)_gen_repr_from_keys _repr_keysrr(r(r)__repr__szWidget.__repr__cks$||_z dVWi|_dSi|_w)aLock a property-value pair. The value should be the JSON state of the property. NOTE: This, in addition to the single lock for all state changes, is flawed. In the future we may want to look into buffering state changes back to the front-end.N)r)rk propertiesr(r(r)rs  zWidget._lock_propertyc csd|jdur dVdSzd|_dVWd|_||j|jdSd|_||j|jw)z@Hold syncing any state until the outermost context manager exitsTNF) _holding_syncr_states_to_sendclearrr(r(r) hold_syncs     zWidget.hold_synccCs||d|j}||jvrGt||||i}t||j|i}tt|d|dkrG|d|dkrGt|d|drG|jrE|j |dS|jrR|j |dSdS)z'Check the property lock (property_lock)rNrrrFT) rr rr^ jsonloads jsondumpsrgrr discardadd)rkrUrzrNZ split_valueZ split_lockr(r(r)rs   zWidget._should_send_propertycCs|dd}|d}|dkr-d|vr+|d}d|vr$t||d|d||d Sd S|dkr7|d S|d krMd|vrK||d|dd Sd S|jd |d S) z0Called when a msg is received from the front-endrrrrrQrRrSZ request_staterz9Unknown front-end to back-end widget msg with method "%s"N)rVrr_handle_custom_msgrierror)rkrrrrQr(r(r)rs  zWidget._handle_msgcCs||||dS)z%Called when a custom msg is received.N)rrr(r(r)r'szWidget._handle_custom_msgcC|S)zConvert a trait value to json.r(rGrkr(r(r)r zWidget._trait_to_jsoncCr))zConvert json values to objects.r(r*r(r(r)rr+zWidget._trait_from_jsoncKsPt|}t|dkr|ddd}d|i}|jdur&dd|jd|d<|SdS)Nnu…z text/plainrrrz(application/vnd.jupyter.widget-view+json)reprr`rr)rkryZ plaintextrr(r(r)r"s   zWidget._repr_mimebundle_cCsB|jdurt|jdr|jjdurn |jj||ddSdSdS)z.Sends a message to the model in the front-end.NrT)rrS)rhasattrrr)rkrrSr(r(r)r6s&z Widget._sendccs|}t|jD]4}|ddkrq t||}||}|||jr$q t|ttfr;|jt kr;|dus:t |dkr;q |Vq dS)Nr_) rrrr r rr@rrrr`)rkrrUrzrr(r(r)r;s   zWidget._repr_keyscs,jj}dfdd|D}d||S)Nz, c3s"|] }d|t|VqdS)z{}={!r}N)formatr )r4rUrr(r) Os  z-Widget._gen_repr_from_keys..z{}({}))rrmjoinr0)rkr class_name signaturer(rr)rMs  zWidget._gen_repr_from_keys)FNrr)NF)Prmrnrorrrrrrr classmethodr staticmethodrrrrrrrrrtagrrrrrrrr Z _view_countrrr rrrrrrr r r rqrrrrrrrrpropertyrErrrr r rrrrrrrrr"rrrr'r rrrrr __classcell__r(r(rr)r.-s                               )Hrpr#systyping contextlibrcollections.abcrZIPythonrrjrrrrr r r r r rrrrjsonrr#rr$rrrrutilsrr_versionrrrinspectgetfilerr*rrrr,r/MutableMappingrK__annotations__r2rIZwidget_serializationra bytearraybytesrXrVrZr^rgrhrqrrrrobjectrr.r(r(r(r)sL   <         - $,