gewdZddlmZ ddlmZn#e$r ddlmZYnwxYwddlmZddl m Z ddl m Z m Z mZmZmZmZmZmZmZmZmZmZddlmZmZddlmZdd lmZm Z!dd l"m#Z#d d l$m%Z%m&Z&e%j'd dZ(dZ)dZ*e*e)dZ+ere,e-e.fZ/ne,e-fZ/dZ0dZ1dZ2dZ3Gdde Z4Gdde4Z5dZ6Gdde7Z8d!dZ9Gdde4Z:d S)"zqBase Widget class. Allows user to create widgets in the back-end that render in the IPython notebook front-end. )contextmanager)Iterable) get_ipython)Comm) HasTraitsUnicodeDictInstanceListIntSetBytesobservedefault Container Undefined) string_typesPY3)display)loadsdumps)standard_b64encode)__protocol_version__ __jupyter_widgets_base_version__.ct|tr fd|DSt|ttfrfd|DSt|t r d|jzS|S)Nc8i|]\}}|t|S_widget_to_json.0kvobjs 9lib/python3.11/site-packages/ipywidgets/widgets/widget.py z#_widget_to_json..!)AAAtq!?1c**AAAc0g|]}t|Srr r#r%r&s r' z#_widget_to_json..##333A3''333r* IPY_MODEL_) isinstancedictitemslisttupleWidgetmodel_idxr&s `r'r!r!s!TAAAAqwwyyAAAA Ae} % %33333333 Av  aj((r*czt|tr fd|DSt|ttfrfd|DSt|t rE|dr0|ddtjvrtj|ddS|S)Nc8i|]\}}|t|Sr_json_to_widgetr"s r'r(z#_json_to_widget..+r)r*c0g|]}t|Srr;r,s r'r-z#_json_to_widget..-r.r*r/ ) r0r1r2r3r4r startswithr5widgetsr7s `r'r<r<)s!TAAAAqwwyyAAAA Ae} % %33333333 A| $ $l)C)C"##RXR`H`H`~af%%r*) from_jsonto_jsoncvt||D]'\}}|}|ddD] }||} |||d<(dS)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_buffersrL=sg #<99&& Vss#  Cc(CC%KO &&r*ct|ttfrd}t|D]\}}t|trG|st|}d}d||<|||||gzat|t ttfr2t|||gz||}||ur|st|}d}|||<nt|t rd}|D]\}}t|trE|st |}d}||=|||||gz_t|t ttfr2t|||gz||}||ur|st |}d}|||<ntd|z|S)z!For internal, see _remove_buffersFTNz+expected state to be a list or dict, not %r) r0r3r4 enumerate _binary_typesappendr1_separate_buffersr2 ValueError) substatepathrGrH is_clonedir%vnewr$s r'rQrQIs(T5M**#S h'' ' 'DAq!]++ ' %#H~~H $I" q!!!##DA3J////AdE233 '(DA3J gNND==$)#'>>$( "&HQK ' Hd # #S NN$$ ' 'DAq!]++ ' %#H~~H $IQKq!!!##DA3J////AdE233 '(DA3J gNND==$)#'>>$( "&HQK ' FQRRR Or*c8gg}}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']], [, ]) )rQ)rFrGrHs r'_remove_buffersrYvs-'L eRw ? ?E , ''r*c t|t|krdS||krdSt||D]\}}trJt|dt|dkrdSVt |tr|jdkr|}t |tr|jdkr|}||krdSdS)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)lenrEr memoryviewcastr0formattobytes)abiaibs r'_buffer_list_equalres 1vvQuAvvta))B  "~~""3'':b>>+>+>s+C+CCCuuD"j)) "bi3.>.>ZZ\\"j)) "bi3.>.>ZZ\\Rxxuu 4r*cPeZdZdZedZeddZdS)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.Loggerlogc6ddlm}|S)Nr)rh) traitletsrh get_logger)selfrhs r' _log_defaultzLoggingHasTraits._log_defaults$!!!!!!~~r*N)__name__ __module__ __qualname____doc__r rhrrmrr*r'rgrgsQ (# $ $C WU^^  ^   r*rgc4eZdZdZeZdZddZdS)CallbackDispatcherz1A structure for registering and running callbackscd}|jD]k} ||i|}||n|}#t$rM}t}||jd||dn|Yd}~dd}~wwxYw|S)z%Call all of the registered callbacks.NzException in callback %s: %sTexc_info) callbacks Exceptionrrhwarning showtraceback)rlargskwargsvaluecallback local_valueeips r'__call__zCallbackDispatcher.__call__s J JH J&h777 (3'> E ' ' ' ]]:H$$%CXq[_$````$$&&&  ' s A3AA..A3Fc|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)rwremoverPrlr~rs r'register_callbackz$CallbackDispatcher.register_callbackst  ,h$.00 N ! !( + + + + + ,HDN:: N ! !( + + + + + , ,::r*NF)rnrorprqr rwrrrr*r'rsrssH;;I ,,,,,,r*rscfd}|S)z+decorator for showing tracebacks in IPythonc |g|Ri|S#t$rT}t}||jd|dn|Yd}~dSYd}~dSd}~wwxYw)Nz!Exception in widget method %s: %sTru)rxrrhryrz)rlr{r|rrmethods r'mz_show_traceback..ms #6$000000 1 # # #Bz  !DfaZ^ ____  """""""""`_____ #s  A-AA((A-r)rrs` r'_show_tracebackrs###### Hr*c&eZdZdZdZdZdZdS)WidgetRegistryci|_dSN) _registryrls r'__init__zWidgetRegistry.__init__s r*c|j|i}||i}||i}||i}||i} || |<dS)zRegister a valueN)r setdefault) rl model_modulemodel_module_version_range model_name view_moduleview_module_version_range view_nameklass model_version view_versions r'registerzWidgetRegistry.registers~00rBB $//0JBOO "--j"==  ++K<< "--.GLL "' Yr*c|j|}t|d}||} | |} t| d} | |} | S)z Get a valuer)rr3values) rlrmodel_module_versionrrview_module_versionrmodule_versions model_names view_modules view_versions view_names widget_classs r'getzWidgetRegistry.getsr.6 ?113344Q7 ":. $[1 -..0011!4 !), r*c #Kt|jD]\}}t|D]\}}t|D]\}}t|D]_\}}t|D]8\} } t| D]\} } ||||| | f| fV9`dSr)sortedrr2) rlrmmrmvrvmrvvrvnrwidgets r'r2zWidgetRegistry.itemssP &t~';';'='= > > B B L"%+BHHJJ%7%7 B B! r&,RXXZZ&8&8BBNJ+1"((**+=+=BB R06rxxzz0B0BBB,L"5;BHHJJ5G5GBB 1 6+7 T_amox*y|B+B%B%B%B%BBBBB B B Br*N)rnrorprrrr2rr*r'rrsU((("BBBBBr*rcd}t|tr!ddl}|dt|S||S)z?For backwards compatibility, we support @register(name) syntax.c |}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_traitsr5 widget_typesr default_value)rws r'regzregister..reg sz    ! !$$Q%7%E$%&=$>$L$%m$4$B$%n$5$C$%&<$=$K$%lO$A$*  , , , r*rNzxWidget registration using a string name has been deprecated. Widget registration now uses a plain `@register` decorator.)r0rwarningswarnDeprecationWarning)namerrs r'rr sg   $ %% QSe f f f s4yyr*ceZdZdZiZeZedZe dZ e dZ e dZ e d=dZ d>dZd Zed d d d Zeddd d Zeedd d Zedd dd Zedd dd Zeddd Zedd dd Zedd ZedZeddZ e!Z"dZ#e$Z%ee&dZ'ee&dZ(fdZ)d Z*d!Z+e,d"d#Z-e.d$Z/d%Z0d?d&Z1d@d'Z2d(Z3d)Z4d*Z5d?d+Z6d>d,Z7d>d-Z8fd.Z9fd/Z:d0Z;ed3Z?e@d4ZAd5ZBd6ZCe d7ZDe d8ZEd9ZFd?d:ZGd;ZHd<ZIxZJS)Ar5Nct|jD]}|dSr)r3r@rclose)clsrs r' close_allzWidget.close_all,s@3;--//00  F LLNNNN  r*c|t_dS)zRegisters a callback to be called when a widget is constructed. The callback must have the following signature: callback(widget)N)r5_widget_construction_callback)r~s r'on_widget_constructedzWidget.on_widget_constructed2s 08,,,r*ctj5ttjrt|dSdSdS)3Static method, called when a widget is constructed.N)r5rcallable)rs r'_call_widget_constructedzWidget._call_widget_constructed:sE  / ;Im@n@n ;  0 0 8 8 8 8 8 < ; ; ;r*c |didd}|ddtkrtd|dt|dd }|d }t j|d |d |d |d|d|d}||}d|vrt||d|d||dS)rmetadataversionrrrz8Incompatible widget protocol versions: received version z, expected version contentdatarFrrrrrr)commrGrHN) rsplitPROTOCOL_VERSION_MAJORrRrr5rrL set_state)rmsgrrrFrrs r'handle_comm_openedzWidget.handle_comm_opened@s0''*b))--i<< ==  a $: : :*pwpwpwzNzNOPP P9~f%W *..u_/E/45L/M/4]/C/4^/D/45K/L/4\/B DD 4((( T ! ! ^ 4c)n E E Er*Fci}|tj}|D] }||||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: N drop_defaultsrr) version_major version_minorrF)r5r@r_get_embed_stater6)rr@rFrs r'get_manager_statezWidget.get_manager_stateUsb ?n++--G Z ZF%+%<%<=%<%Y%YE&/ " "!"QGGGr*c|j|j|jd}t||\}}}||d<t |dkrdt ||D|d<|S)N)rrrrrFrc`g|]+\}}d|t|dd,S)base64ascii)encodingrTr)rdecode)r#pds r'r-z+Widget._get_embed_state..ms\ H H H%)Aq.6)*);A)>)>)E)Eg)N)N!P!P H H Hr*rH)rrrrY get_stater\rE)rlrrF model_staterGrHs r'rzWidget._get_embed_stateds* .$($>   .=T^^Zg^=h=h-i-i* \7$g w<.s888888r*Tr)traitsrs r' _default_keyszWidget._default_keyss$88$!7!78888r*rc |dd|_tt|jdi|t||dS)zPublic constructorr6Nr)poprsuperr5rropen)rlr| __class__s r'rzWidget.__init__s`J55$fd$..v...''--- r*c.|dS)zObject disposalN)rrs r'__del__zWidget.__del__s r*c|jft|\}}}td||d|dti}|j |j|d<t di||_dSdS)z6Open a comm to the frontend if one isn't already open.Nzjupyter.widget)rFrGr) target_namerrHrcomm_idr)rrYrr1rrr)rlrFrGrHr{s r'rz Widget.opens 9 +:4>>;K;K+L+L (E<$4',lKK '"+-A!BD ~)"&.Y t DIII  r*rc|ddS|j|_|j|j|t j|j<dS)z Called when the comm is changed.newN)r6rron_msg _handle_msgr5r@)rlchanges r' _comm_changedzWidget._comm_changedsI %= F )***(,t}%%%r*c|jjS)zmGets the model id of this widget. If a Comm doesn't exist yet, a Comm will be created automagically.)rrrs r'r6zWidget.model_ids y  r*c|jNtj|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)rr5r@rr6r_ipython_display_rs r'rz Widget.closesU 9 N  t}d 3 3 3 IOO   DI%)D " " " ! r*c$||}t|dkrf|jr-|D]\}}||jvr ||j|<t |\}}}d||d}|||dSdS)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. rKrupdate)rrFrGrHN)rr\_property_lockr2rY_send)rlrKrFrr}rGrHrs r' send_statezWidget.send_states3'' u::>>" :#(;;==::KD%t22249+D1+:5+A+A (E<%|TTC JJsGJ , , , , , >r*c||j}n@t|tr|g}n't|tr|}nt di}|}|D]}||d|j}|t|||}ts?t||tr$t|trt|}|r!| |||js|||<|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 NonerB)rr0rrrRrtrait_metadata_trait_to_jsongetattrrrbytesr]_comparer) rlrKrrrFrr$rBr}s r'rzWidget.get_states ;9DD \ * * S5DD X & & SDDQRR R ! !A))!Y8KLLGGGD!,,d33E *:fQi77 *Jue>!   q 1 1     >>!Q'' '6Mr*c L|jdi|5|5|D]M}||jvrB||d|j}||||||N dddn #1swxYwYddddS#1swxYwYdS)z3Called when a state is received from the front-end.rANr)_lock_propertyhold_trait_notificationsrr_trait_from_json set_trait)rl sync_datarrAs r'rzWidget.set_statesv !T - -9 - - K Kt/L/L/N/N K K! K K49$$ $ 3 3D+484I!K!KINN49T?D)I)IJJJ  K K K K K K K K K K K K K K K K K K K K K K K K K K K K K K K K K Ks5BAB5 BB BB BB Bc:|d|d|dS)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)rrlrrHs r'sendz Widget.send#s( h7;;W MMMMMr*c>|j||dS)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_msg/s& --hv-FFFFFr*c>|j||dS)a\(Un)Register a widget displayed callback. Parameters ---------- callback: method handler Must have a signature of:: callback(widget, **kwargs) kwargs from display are passed through without modification. remove: bool True if the callback should be unregistered.r.N)_display_callbacksrrs r' on_displayedzWidget.on_displayed=s& 11(61JJJJJr*c tt|jdi||D]I\}}|dr/|j|||JdS)z/Dynamically add trait attributes to the Widget.rNr)rr5 add_traitsr2 get_metadatarrPr)rlrrtraitrs r'r4zWidget.add_traitsLs&fd&00000!<<>> & &KD%!!&)) &   &&&%%% & &r*c|d}|jO|jjC||jvr:||t ||r||t t||dS)z#Called when a property has changed.rNr) rkernelr_should_send_propertyrrrr5 notify_change)rlr rrs r'r:zWidget.notify_changeTsf~ 9 TY%5%Aty  T%?%?gdTXFYFY%Z%Z D))) fd))&11111r*cP||Sr)_gen_repr_from_keys _repr_keysrs r'__repr__zWidget.__repr__`s ''(9(9:::r*c+HK||_ dVi|_dS#i|_wxYw)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)rl propertiess r'r$zWidget._lock_propertygs@) % EEE"$D   "D  $ $ $ $s !c#2K|jdurdVdS d|_dVd|_||j|jdS#d|_||j|jwxYw)z@Hold syncing any state until the outermost context manager exitsTNF) _holding_syncr_states_to_sendclearrs r' hold_synczWidget.hold_syncvs   % % EEEEE -%)"%*" 4555$**,,,,,&+" 4555$**,,,,s ACalled when a view has been displayed for this widget instanceN)r1)rlr|s r'_handle_displayedzWidget._handle_displayeds#///////r*c|S)zConvert a trait value to json.rr8rls r'rzWidget._trait_to_json r*c|S)zConvert json values to objects.rrUs r'r&zWidget._trait_from_jsonrVr*c t|}t|dkr |dddz}d|i}|jdd|jd|d<t |d |j|jd i|dSdS) z>Called when `IPython.display.display` is called on the widget.nNu…z text/plainrrrz(application/vnd.jupyter.widget-view+jsonT)rawr)reprr\rrrrS)rlr| plaintextrs r'rzWidget._ipython_display_sJJ y>>C  !$3$%/I )  ? &"#!" N@@D; < $ ? & "D " , ,V , , , , , ' &r*cl|j*|jj |j||dSdSdS)z.Sends a message to the model in the front-end.N)rrH)rr8r,)rlrrHs r'rz Widget._sends@ 9 TY%5%A INNWN 5 5 5 5 5 ! %A%Ar*c#nK|}t|jD]}|ddkrt||}||}|||jrCt |ttfr&|jtkr|t|dkr|VdS)Nr_) rrrrrrr0rr rr\)rlrrKr}r6s r'r=zWidget._repr_keyss$)$$  C1v}}D#&&E3KE}}UE$788 UY$566 %22=CJJ!OOIIII  r*cnjj}dfd|D}|d|dS)Nz, c3BK|]}|dt|VdS)=N)r)r#rKrls r' z-Widget._gen_repr_from_keys..sM  ssGD#... /      r*())rrnjoin)rlr class_name signatures` r'r<zWidget._gen_repr_from_keyss[^, II         &::yyy11r*)FNrr)NF)Krnrorprr@rr classmethodr staticmethodrrrrrrrtagrrrrrrrr _view_countr rr rrrr rrBr rCrsr1r/rrrrr propertyr6rrrrrrr,rr2r4r:r>rr$rEr9rr rOrSrr&rrr=r< __classcell__)rs@r'r5r5 sP%)!G">##L[ 88\899\9   \ ( H H H\ H    OOO '- !T333363D3>>G3 +t====@SdS^^#G$D O[_aaaadadjnadaoao$ """"%#4#..74D *,,,,/CTCNN"72 RTTTTWTW]aTWTbTb#dtKLLLMPMPVZMPM[M[ 8)d ; ; ;D 43 4 4 4D WV__99_9TVVNMceeO!"4b99X0"55N  % % % WV__--_-!!X! * * *----&BWWW K K K N N N N G G G G K K K K&&&&& 2 2 2 2 2;;; % %^ % - -^ -*aa_a2444000\\---26666 $2222222r*r5N)r);rq contextlibrcollections.abcr ImportError collectionsIPython.core.getipythonripykernel.commrrjrrr r r r r rrrrripython_genutils.py3compatrrIPython.displayrjsonrrHrrIrr_versionrrrrr!r<widget_serializationr] bytearrayrrOrLrQrYrergrsrobjectrrr5rr*r'r|s' &%%%%%%(((((((%%%$$$$$$$$%0/////98888888######77777777%%%%%%MMMMMMMM3-3C88;! ,E2MM+M & & &+++Z ( ( (@      y    ",",",",",)",",",H    &B&B&B&B&BV&B&B&BP*[2[2[2[2[2 [2[2[2[2[2s