:vj9RUdZddlZddlZddlZddlmZddlmZddlm 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(jRe Z*dZ+e%jXddZ-e&jXddZ.e+ddZ/iZ0ejbe2dfe3d<dZ4dZ5e5e4dZ6e7e8e9fZ:dZ;dZGdde Z?Gdde?Z@d ZAGd!d"ZBeBZCd#ZDGd$d%eEZFGd&de?ZGy)'zqBase 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__c|tjvr#tj|jdvSt|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)namers j/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/ipywidgets/widgets/widget.pyenvsetr+s8  rzzzz$%%'/VVVG}.JUPYTER_WIDGETS_ECHOTrWidget _instancesc 6t|tr.|jDcic]\}}|t||c}}St|tt fr|Dcgc]}t||c}St|t rd|jzS|Scc}}wcc}w)N IPY_MODEL_) isinstancedictitems_widget_to_jsonlisttupler0model_idxobjkvs r*r7r70s!T78wwyAytq!?1c**yAA Ae} %123A3'33 Av ajj(( B3s BBc ht|tr.|jDcic]\}}|t||c}}St|tt fr|Dcgc]}t||c}St|t r(|jdr|ddtvr t|ddS|Scc}}wcc}w)Nr3 ) r4r5r6_json_to_widgetr8r9str startswithr1r;s r*rBrB:s!T78wwyAytq!?1c**yAA Ae} %123A3'33 As  \ :qv?S!BC&!! B3s B)B/) from_jsonto_jsonc^t||D]\}}|}|ddD]}||} |||d< y)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_pathsbuffers buffer_pathbufferr=keys r* _put_buffersrPKsI #<9 Vs#Cc(C$%KO :r,ct|ttfrd}t|D]\}}t|tr;|s t|}d}d||<|j ||j ||gzQt|t ttfsmt|||gz||}||us|s t|}d}|||<|St|t rd}|jD]\}}t|tr9|s t |}d}||=|j ||j ||gzOt|t ttfskt|||gz||}||us|s t |}d}|||<|Std|z)z!For internal, see _remove_buffersFTNz+expected state to be a list or dict, not %r) r4r8r9 enumerate _binary_typesappendr5_separate_buffersr6 ValueError) substatepathrKrL is_clonedir?vnewr>s r*rUrUWs(T5M* h'DAq!]+ #H~H $I" q!##DA3J/AdE23(DA3J gND=$#'>$( "&HQK(D O' Hd # NN$DAq!]+ #H~H $IQKq!##DA3J/AdE23(DA3J gND=$#'>$( "&HQK%" OFQRRr,c0gg}}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']], [, ]) )rU)rJrKrLs r*_remove_buffersr]s*'L eRw ?E , ''r,ct|t|k7ry||k(ryt||D]<\}}t|jdt|jdk7s<yy)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)lenrI memoryviewcast)abiaibs r*_buffer_list_equalrgse 1vQAva)B b>  s #z"~':':3'? ? r,c<eZdZdZedZeddZy)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.Loggerlogc.ddlm}|jS)Nr)rj) traitletsrj get_logger)selfrjs r* _log_defaultzLoggingHasTraits._log_defaults!~~r,N)__name__ __module__ __qualname____doc__r rjrror,r*riris* # $C U^  r,ric,eZdZdZeZdZddZy)CallbackDispatcherz1A structure for registering and running callbackscd}|jD]} ||i|}||n|}|S#t$rF}t}| |jj d||dn|j Yd}~`d}~wwxYw)z%Call all of the registered callbacks.NzException in callback %s: %sTexc_info) callbacks Exceptionrrjwarning showtraceback)rnargskwargsvaluecallback local_valueeips r*__call__zCallbackDispatcher.__call__sH J&77 (3'> E'  ' ]:HH$$%CXq[_$`$$&  's% A4.msk #$000 1 #Bz  !DfaZ^ _  ""` #s  A# L"%+BHHJ%7! r&,RXXZ&8NJ+1"((*+= R06rxxz0B,L"5;BHHJ5G 1 6+7 T_amox*y|B+B%B6H1C,>'9&8!?sCC N)rprqrrrrrr6rtr,r*rrs("Br,rc |j}tj|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) class_traitsrr default_value)rws r*rrs|A q)77!"#:!;!I!I!"=!1!?!?!">!2!@!@!"#9!:!H!H!"<!>!>!' ) Mr,ceZdZdZdZy)_staticpropertyc||_yrfget)rnrs r*rz_staticproperty.__init__$s  r,c*|J|jSrr)rn owner_self owner_clss r*__get__z_staticproperty.__get__'s!!!yy{r,N)rprqrrrrrtr,r*rr#s r,rceZdZdZdZedZedZedZedZ e dZ e dZ e dZe d Ze d Ze d Ze dBd ZdCdZdZedddj-dZedddj-dZeeddj-dZedddj-dZedddj-dZeddj-dZedddj-dZe dZ!e"d Z#e$d!d"Z%e&Z'd Z(e)Z*e+e,d#Z-fd$Z.d%Z/d&Z0d'Z1d(Z2e3d)d*Z4e5d+Z6d,Z7dDd-Z8dEd.Z9d/Z:d0Z;d1Zfd4Z?fd5Z@d6ZAeBd7ZCeBd8ZDd9ZEeFd:ZGd;ZHe d<ZIe d=ZJd>ZKdDd?ZLd@ZMdAZNxZOS)Fr0Nctd}|jjtk(r|jjdvs t dt S)Nr getmemberssetup_instance setup_classzWidget.widgets is deprecated.rf_code co_filenameTRAITLETS_FILEco_namerr1frames r*widgetszWidget.widgets4sH 1  ((N: @T@TYHAH 7 8r,ctd}|jjtk(r|jjdvs t dt S)Nrrz%Widget._active_widgets is deprecated.rrs r*_active_widgetszWidget._active_widgets>sH 1  ((N: @T@TYHAH ? @r,ctd}|jjtk(r|jjdvs t dt S)Nrrz#Widget._widget_types is deprecated.rrrrrrrrs r* _widget_typeszWidget._widget_typesHsH 1  ((N: @T@TYHAH = >r,ctd}|jjtk(r|jjdvs t dt S)Nrrz"Widget.widget_types is deprecated.rrs r* widget_typeszWidget.widget_typesRsH 1  ((N: @T@TYHAH < =r,chttjD]}|jyr)r8r1rclose)clsrs r* close_allzWidget.close_all\s#:,,./F LLN0r,c|t_y)zRegisters a callback to be called when a widget is constructed. The callback must have the following signature: callback(widget)N)r0_widget_construction_callback)rs r*on_widget_constructedzWidget.on_widget_constructedas 08,r,ctj0ttjrtj|yyy)3Static method, called when a widget is constructed.N)r0rcallable)rs r*_call_widget_constructedzWidget._call_widget_constructedis5  / / ;ImIm@n  0 0 8Ao ;r,c|jdijdd}|jddtk7rtd|dt||_|j j |jy) z Class method, called when the comm-open message on the "jupyter.widget.control" comm channel is received metadataversionr-rz@Incompatible widget control protocol versions: received version , expected version N)rsplitCONTROL_PROTOCOL_VERSION_MAJORrVr _control_common_msg_handle_control_comm_msg)rrmsgrs r*handle_control_comm_openedz!Widget.handle_control_comm_openedosy ''*b)--i< == a $B BxB^_` `    !=!=>r,c|j td|dd}|d}|dk(r|jtj }i}d}|D]D}|j |j |j|j|d||j<Ft|\}}} |jjtd || | ytd |z) Nz)Control comm has not been properly openedcontentdatarrequest_statesF drop_defaults)rrrrJ update_states)rstatesrKrLzAUnknown front-end to back-end widget control msg with method "%s") r RuntimeErrorget_manager_stater1rrrr get_stater:r]sendr5) rrrrr full_staterrrKrLs r*rzWidget._handle_control_comm_msg|s    $JK K9~f%h % %  ! ! # '')GJ!M!"("4"4$*$8$8,2,H,H#--M-J / 6??+"1@ 0K -J g    " "4&!)$ # bekkl lr,c ||jdijdd}|jddtk7rtd|dt|dd }|d }t j|d |d |d |d|d|d}||}d|vrt ||d|d|j|y)rrrrr-rz8Incompatible widget protocol versions: received version rrrrJrrrrrrrrKrLN)rrPROTOCOL_VERSION_MAJORrVrrrP set_state)rrrrrJrrs r*handle_comm_openedzWidget.handle_comm_openeds''*b)--i< == a $: :pwzNOP P9~f%W !}}U?%;/45L/M/4]/C/4^/D/45K/L/4\/B D 4( T ! ^ 4c)n Er,Fci}|tj}|D]!}|j|||j<#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: rrr) version_major version_minorrJ)r1r_get_embed_stater:)rrrJrs r*rzWidget.get_manager_statesO ? '')GF%+%<%<=%<%YE&// "!"QGGr,c :|j|j|jd}t|j |\}}}||d<t |dkDr>t ||Dcgc]#\}}d|t|jdd%c}}|d<|Scc}}w) N)rrrrrJrbase64ascii)encodingrXrrL) rrrr]rr`rIrdecode)rnrrJ model_staterKrLpds r*r zWidget._get_embed_states** ..$($>$>  .=T^^Zg^=h-i* \7$g w)E)Eg)N!P-G HE)   Hs&(Bc2tdd|jS)Nrrrr r:)r5 _model_idrs r* get_view_speczWidget.get_view_specs!1t~~NNr, WidgetModelzName of the model.T)help 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.keyscL|jdDcgc]}|c}Scc}w)NTr)traits)rnr)s r* _default_keyszWidget._default_keyss'!%$!78!7!7888s !rtc |jdd|_t| di|tj ||j y)zPublic constructorr:Nrt)poprsuperrr0ropen)rnr __class__s r*rzWidget.__init__s<J5 "6"''- r,ctdNz8Widgets cannot be copied; custom implementation requiredNotImplementedErrorrs r*__copy__zWidget.__copy__!"\]]r,ctdr'r()rnmemos r* __deepcopy__zWidget.__deepcopy__r+r,c$|jy)zObject disposalN)rrs r*__del__zWidget.__del__s  r,c|jkt|j\}}}td||d|dti}|j |j |d<tj di||_yy)z6Open a comm to the frontend if one isn't already open.Nzjupyter.widget)rJrKr) target_namerrLrcomm_idrt)rr]rr5rr create_comm)rnrJrKrLr~s r*r$z Widget.open sx 99 +:4>>;K+L (E<$4',lK '"+-A!BD ~~)"&..Y((040DI r,rc|dy|j|_|jj|j|t |j<y)z Called when the comm is changed.newN)r:rrr _handle_msgr1)rnchanges r* _comm_changedzWidget._comm_changedsB %=  ))*$( 4==!r,c.|jjS)zmGets the model id of this widget. If a Comm doesn't exist yet, a Comm will be created automagically.)rr3rs r*r:zWidget.model_id#s yy   r,c|jItj|jd|jj d|_d|_yy)zClose method. Closes the underlying comm. When the comm is closed, all of the widget views are automatically removed from the front-end.N)rr1r"r:r_repr_mimebundle_rs r*rz Widget.close.sB 99 NN4==$ / IIOO DI%)D " !r,c|j|}t|dkDrk|jr6|jD]#\}}||jvs||j|<%t |\}}}d||d}|j ||yy)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. rOrupdaterrJrKrN)rr`_property_lockr6r]_send)rnrOrJr)rrKrLrs r* send_statezWidget.send_state:s3' u:>""#(;;=KD%t22249++D1$1,;5+A (E<%|TC JJsGJ , r,ch| |j}n2t|tr|g}nt|tr|}n t di}|j }|D]Y}|j |d|j}|t|||}|r |j|||jrU|||<[|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} z2key must be a string, an iterable of keys, or NonerF) rr4rCrrVrtrait_metadata_trait_to_jsongetattr_comparer) rnrOrrrJrr>rFrs r*rzWidget.get_stateMs ;99D S !5D X &DQR RA))!Y8K8KLGGD!,d3E  eVAY=T=T(U a   r,cj|jjdk(xr|jjdk(S)Nndarraynumpy)r%rprq)rnr<s r* _is_numpyzWidget._is_numpyls+{{##y0VQ[[5K5Kw5VVr,c||j|s|j|rddl}|j||S||k(S)Nr)rLrK array_equal)rnrcrdnps r*rHzWidget._compareos6 >>! q 1 >>!Q' '6Mr,c "trmi}|jD].\}}||jvs|j|dds*|||<0|r(t |\}}}d||d}|j |||j di|5|j5|D]J}||jvs|j|d|j} |j|| |||L ddddddy#1swYxYw#1swYyxYw) z3Called when a state is received from the front-end. echo_updateTr/r@rrENrt) r.r6rrEr]rB_lock_propertyhold_trait_notifications_trait_from_json set_trait) rn sync_data echo_stateattrrecho_buffer_paths echo_buffersrr)rEs r*rzWidget.set_statevs  J(0 e499$)<)Mj>Y; -|+'$5  3 5 !T -9 -t/L/L/N!499$ $ 3 3D+484I4I!KINN49T?D)IJ "0O - -/N/N - -s*DC9,;C9(D9D >DDc0|jd|d|y)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)rBrnrrLs r*rz Widget.sends h7;W Mr,c>|jj||y)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_callbacksrrs r*rz Widget.on_msgs --hv-Fr,c t|di||jD]@\}}d|jvs|jj ||j |By)z/Dynamically add trait attributes to the Widget.rNrt)r# add_traitsr6rrrTrC)rnrr)traitr%s r*razWidget.add_traitssQ $V$!<<>KD%'   &%*r,c|d}|jSt|jdd<||jvr.|j|t||r|j |t ||y)z#Called when a property has changed.r)NkernelTr>)rrGr_should_send_propertyrCr# notify_change)rnr8r)r%s r*rfzWidget.notify_changeshf~ 99 WTYY$%G%Styy T%?%?gdTXFY%ZD) f%r,c@|j|jSr)_gen_repr_from_keys _repr_keysrs r*__repr__zWidget.__repr__s''(9::r,c+JK||_ di|_y#i|_wxYww)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)rA)rn propertiess r*rRzWidget._lock_propertys() % "$D "D s## #c#DK|jdurdy d|_dd|_|j|j|jjy#d|_|j|j|jjwxYww)z@Hold syncing any state until the outermost context manager exitsTNF) _holding_syncrC_states_to_sendclearrs r* hold_synczWidget.hold_syncs    %  -%)"%*" 4 45$$**,&+" 4 45$$**,sB  A=B >BB c|j|d|j}||jvrt||||i}t||j|i}t t |d|dk(rE|d|dk(r:t |d|dr(|jr|jj|y|jr|jj|yy)z'Check the property lock (property_lock)rFrrrFT) rErFrAr] jsonloads jsondumpsrgrnrodiscardadd)rnrOrrF split_value split_locks r*rezWidget._should_send_propertys%%c9d6I6IJ $%% %)C1E*FGK(30C0CC0H)IJJ)KN34 1 ENjm3&{1~z!}E%%((005     $ $S )r,c<|dd}|d}|dk(r3d|vr.|d}d|vrt||d|d|j|y y |dk(r|jy |d k(rd|vr|j|d|dy y |jj d |zy ) z0Called when a msg is received from the front-endrrrr?rJrKrL request_stater\z9Unknown front-end to back-end widget msg with method "%s"N)rPrrC_handle_custom_msgrjerror)rnrrrrJs r*r7zWidget._handle_msgs9~f%h X $W !T) ^(C !$3%/I )  ?? &"#!" NN@D; < K 'r,c|jMt|jdr|jj n |jj||yyy)z.Sends a message to the model in the front-end.Nrd)rrL)rhasattrrdr)rnrrLs r*rBz Widget._send6sI 99 gdiiYaFbdii&6&6&Bhl IINNWN 5'C r,c#JK|j}t|jD]w}|ddk(r t||}||}|j ||j r:t |ttfr$|j tk(r|t|dk(rt|yyw)Nr_) rrrrGrHrr4rr rr`)rnrrOrrbs r*rizWidget._repr_keys;s$))$C1v}D#&E3KE}}UE$7$78UY$56%%2=CJ!OI%sB!B#cjj}djfd|D}dj||S)Nz, c3VK|] }dj|t|"yw)z{}={!r}N)formatrG).0rOrns r* z-Widget._gen_repr_from_keys..Os-    S'$"4 5s&)z{}({}))r%rpjoinr)rnr class_name signatures` r*rhzWidget._gen_repr_from_keysMs@^^,, II   z955r,)FNrr)NF)Prprqrrrrrrrrr classmethodr staticmethodrrrrrrr rrtagrrrrrrrr _view_countrrr rrr r rArnr ror rvr_rr*r.r0r$rr9propertyr:rrCrrLrHrrrrarfrjrrRrqrerr7r{rFrTr<rBrirh __classcell__)r%s@r*r0r0-sP%)!M8899  ? ?mm>  ( H H O - !T3363D3>3 +t==@SdS^#$D O[_aadadjnadao$ ""%#4#.4D *,,/CTCN"2 RTTWTW]aTWTbdtKLMPMPVZMPM[ $ D 3 4D V_99VNMeO0"5N ^^ 1 V_))!! *-&>WK6 N G& &; % % - -.aa24(6 $6r,)Hrsr%systyping contextlibrcollections.abcrIPythonrrlrrrr r r r r rrrrrjsonrrsrrtrrr rutilsrr_versionrrrinspectgetfilerr+rrrr.r1MutableMappingrC__annotations__r7rBwidget_serializationra bytearraybytesrSrPrUr]rgrirvrrrrobjectrr0rtr,r*rsN  %$8%*kk + 4-33C8;!C!=!C!CC!H!K4dC57 V " "3= 16! Y.  &+Z (.  y  ",)",H  &B&BX    ff6 f6r,