}tfgLUdZddlmZddlZddlZddlZddlZddlZddlm Z m Z ddl m Z m Z mZmZmZmZmZddlmZmZddlmZmZmZmZmZmZmZmZdd lm Z e r6ddl!Z!ddl"Z"ddl#Z#ddl$Z$ddl%Z%dd l&m'Z'm(Z(m)Z)dd l*m+Z+Gd d e'Z,e Z-de.d<ddgZ/dZ0dZ1dZ2dZ3dZ4dZ5dZ6e6e4e e6e5e ddZ7e3ef d/dZ8iZ9de.d<d0dZ:GddZ;GddZd3d"Z?d4d#Z@d5d$ZAd6d%ZBd7d&ZCd'ZD d8d(ZEd6d)ZFd9d*ZG d:d+ZH d:d,ZId;d-ZJd6&>  6 &>s &A>>BcteZdZdZdddd d dZd dZed dZeddZ dd Zy )r/aDescriptor that builds a ReprMimeBundle when accessed on an instance. The `__get__` method is called when the descriptor's name is accessed on a class or instance. It returns a `ReprMimeBundle` instance, which is a callable that implements the `_repr_mimebundle_` protocol that jupyter expects, but also manages the comm channel between the python model and the javascript view. For more on descriptors, see: Parameters ---------- follow_changes : bool, optional If `True` (default), the state of the python model will be updated whenever the state of the javascript view changes (and vice versa). autodetect_observer : bool, optional If `True` (default), an attempt will be made to find a known observer-pattern API on the object (such as a psygnal.SignalGroup or traitlets.HasTraits) and use it to automatically send state changes to the javascript view. If `False`, the javascript view will only be updated when the `send_state()` method is explicitly called. no_view : bool, optional If `True`, the callable will return `None` instead of a mimebundle. This is useful for cases where you want to use the comm channel to send state updates to the front end, but don't want to display anything in the notebook (i.e., A DOM-less widget). Defaults to `False`. **extra_state : Any, optional Any extra state that should be sent to the javascript view (for example, for the `_esm` anywidget field.) By default, `{'_esm': _DEFAULT_ESM}` is added to the state. Examples -------- Note that *technically* you could name the attribute anything you want but it probably only makes sense to call it '_repr_mimebundle_'. >>> class Foo: ... _repr_mimebundle_ = MimeBundleDescriptor() >>> foo = Foo() in a jupyter notebook, this line will now access `_repr_mimebundle_`, and turn the descriptor into an instance of `ReprMimeBundle`, which spins up a comm channel, sets up state synchronization, and, when called, returns a mimebundle dict that includes the comm id. >>> foo TF)follow_changesautodetect_observerno_viewc |jtt||_t|_||_||_||_|jjD]"\}}t|}|||j|<$yr ) setdefaultrr _extra_state _REPR_ATTR_name_follow_changes_autodetect_observer_no_viewitemsr)r"rTrUrV extra_statekv file_contentss r%__init__zMimeBundleDescriptor.__init__s| x6' -$7! %%++- 5DAq-a0M('4!!!$  5r'c||_y)zCalled when this descriptor is assigned to an attribute on a class. In most cases, we won't *want* `name` to be anything other than `'_repr_mimebundle_'`. N)r[)r"ownernames r% __set_name__z!MimeBundleDescriptor.__set_name__s  r'cyr r!r"instancerfs r%__get__zMimeBundleDescriptor.__get__sLOr'cyr r!rjs r%rlzMimeBundleDescriptor.__get__sHKr'c||S t||j|j|j}|jr|j tjtt5t||j|ddd|S#t $r }tjd|dd}~wwxYw#1swY|SxYw)aCalled when this descriptor's name is accessed on a class or instance. Examples -------- >>> class Foo: ... _repr_mimebundle_ = MimeBundleDescriptor() ... >>> Foo._repr_mimebundle_ # same as Foo._repr_mimebundle_.__get__(None, Foo) >>> foo = Foo() >>> foo._repr_mimebundle_ # same as Foo._repr_mimebundle_.__get__(foo, Foo) N)rUr`rVzError in Anywidget repr: r  stacklevel)r0r]rYr^r\sync_object_with_view ExceptionwarningswarnrKrLAttributeError ValueErrorsetattrr[)r"rkrfrepr_objes r%rlzMimeBundleDescriptor.__get__s  K %$($=$= -- H ##..0  < 4 Hdjj( 3  4#  MM6qc:q I    4s$A B-B; B8B33B8;CN) rTboolrUrzrVrzr`rr)None)rftypergstrr)r{)rkr{rfr|r)r/)rkobjectrfr|r)r0)rkz object | Nonerfr|r)z%ReprMimeBundle | MimeBundleDescriptor)r+r,r-__doc__rdrhr rlr!r'r%r/r/~s-d $$( 55" 5  5  5 5*OO KK/%/.2/ ./r'cpeZdZdZ d d dZd d dZd ddZddZddZ d ddZ dd Z y)r0aOCallable object that behaves like a `_repr_mimebundle_` method... which is to say, it returns a mimebundle (mapping of mimetypes to data) when called. This object *also* controls an Comm channel between the front-end js view and some python model object (`obj`), Parameters ---------- obj : object The python model object which is being represented by the view. Most likely this will be a dataclass instance that has been made "evented" by the anywidget decorator... but we type it as `object` to allow for other use cases, to make it clearer what protocols we expect from the object. autodetect_observer : bool, optional If `True` (default), an attempt will be made to find a known observer-pattern API on the object (such as a psygnal.SignalGroup or traitlets.HasTraits) and use it to automatically send state changes to the javascript view. If `False`, the javascript view will only be updated when the `send_state()` method is explicitly called. no_view : bool, optional If `True`, the callable will return `None` instead of a mimebundle. This is useful for cases where you want to use the comm channel to send state updates to the front end, but don't want to display anything in the notebook (i.e., A DOM-less widget). Defaults to `False`. extra_state : dict, optional Any extra state that should be sent to the javascript view (for example, for the `_esm` anywidget field.) By default, `{'_esm': DEFAULT_ESM}` is added to the state. Nc|_|xsij_jjtt |_ tjj_ t_t!_t%_t)_jj-D]U\}}t/|t0t2fst5|j|<|j6j8|fdfd }Wy#t$r(fd_ tjdddYwxYw)NcSr r!r#sr%z)ReprMimeBundle.__init__..4sr'z Anywidget: z is not weakrefable, so it will not be garbage collected until the view is closed. Please consider adding `__slots__ = ('__weakref__',)` to your class definition.rocF|j|<j|yr )rY send_state) new_contentskeyr"s r% _on_changez+ReprMimeBundle.__init__.._on_changeJs-9D%%c*OOC(r')rr}rr}r)r{)r]copyrYrXr _anywidget_idr^rNref_on_obj_deleted_objrMrsrtrR_commset_disconnectorsdetermine_state_getter _get_statedetermine_state_setter _set_stater_ isinstancerrr}changedconnect)r"r#rUr`rVrvaluers`` r%rdzReprMimeBundle.__init__$s7%8!(.B446 $$%6 c8JK  +2;;sDX %$W !T)tN';S^LU+  (^ . OO 'X'78<< r'c |jryt|jjt |j S)z=Called when _repr_mimebundle_ is called on the python object.N)model_id repr_text)r^rrcomm_idreprr)r"kwargss r%r&zReprMimeBundle.__call__s0 == (:(:d499;FWXXr'c|r5|jj|j|j|r|jr|j }| t d|jrtjddyttfD]4}|||j}|s|jj|ytjd|ddyyy)aConnect the front-end to changes in the model, and vice versa. Parameters ---------- py_to_js : bool, optional If True (the default), changes in the python model will be reflected in the front-end. js_to_py : bool, optional If True (the default), changes in the front-end will be reflected in the python model. NzCannot sync a deleted objectz$Refusing to re-sync a synced object.rrozCould not find a notifier on z^ (e.g. psygnal, traitlets). Changes to the python object will not be reflected in the JS view.) ron_msgrrr]r RuntimeErrorrrsrt_connect_psygnal_connect_traitletsadd)r"py_to_jsjs_to_pyr# connector disconnects r%rqz$ReprMimeBundle.sync_object_with_views  JJ  d.. / OO  11))+C{"#ABB"" DQRS/0BC  &sDOO< ''++J7   3C59 ! -28r'c|jjd|jrOtjt 5|jj ddd|jrNyy#1swYxYw)zEDisconnect the view from changes in a model instance, and vice versa.N)rrrrKrLrrrP)r"s r%rz&ReprMimeBundle.unsync_object_with_viewsb $!!$$Y/ ,)##'')+ ,!! , ,s  A88B)TNF)r#r~rUrzr`zdict[str, Any] | NonerVrzr )rzweakref.ReferenceType | Noner)r{)r$zstr | Iterable[str] | Noner)r{)rrr)r{)rz Sequence[str]r)ztuple[dict, dict] | None)TT)rrzrrzr)r{r)r{) r+r,r-rrdrrrr&rqrr!r'r%r0r0s}D%)-1 )) ))"))+ ))  ))V 7>FY7;../3. .`,r'c\t|jdt|jS)z9Return a unique id for an object, to send to the JS side..)r|r,r+rs r%rrs)3i"" #1T#Y%7%7$8 99r'c*tt|trtt|tSt |rdSt |rt St|rt|drtStSt|rtStd|d)aAutodetect how `obj` can be serialized to a dict. This looks for various special methods and patterns on the object (e.g. dataclass, pydantic, etc...), and returns a callable that can be used to get the state of the object as a dict. As an escape hatch it first looks for a special method on the object called `_get_anywidget_state`. Returns ------- state_getter : Callable[[object], dict] A callable that takes an object and returns a dict of its state. ct|Sr )rr#r$s r%rz(determine_state_getter..s F3Kr' model_dumpz,Cannot determine a state-getting method for zI. Please implement a `_get_anywidget_state()` method that returns a dict.) hasattrr|_STATE_GETTER_NAMErr_is_traitlets_object_get_traitlets_state_is_pydantic_model_get_pydantic_state_v2_get_pydantic_state_v1_is_msgspec_struct_get_msgspec_staterMrs r%rrs tCy,-tCy"455C0/C ### 3 %) )%%#!!  6sg>R R r'cN|jD]\}}t|||y)z?A default state setter that just sets attributes on the object.N)r_rw)r#r@rvals r%_default_set_staters'KKMSS#r'crtt|trtt|tStS)a7Autodetect how `obj` can update its state from a dict. The default implementation just sets attributes on the object. Returns ------- state_setter : Callable[[object, dict], None] A callable that takes an object and a dict of its state, and updates the object accordingly. )rr|_STATE_SETTER_NAMErrrs r%rrs+tCy,-tCy"455 r'ctrddl}ntjj d}|yt |dd}t ||jr|Stjttt5t|jD]%}t ||js|ccdddS dddy#1swYyxYw)z*Look for a psygnal.SignalGroup on the obj.rNpsygnalevents)rrsysmodulesgetrr SignalGrouprKrLruRecursionErrorrMvarsvalues)r#rrattrs r%_get_psygnal_signal_groupr*s++//),S(D )F&'--.      I$$& D$ 3 34    s63C*C7CC c`t|jdfd dfd }|Sy)a Check if an object has a psygnal.SignalGroup, and connect it to send_state. Returns ------- disconnect : Callable | None A callable that disconnects the psygnal.SignalGroup from send_state, or None if no psygnal.SignalGroup was found. Nc@|jjhyr )signalrg)eventrs r%_on_psygnal_eventz+_connect_psygnal.._on_psygnal_eventPs  ))* +r'c(jyr )r)rrsr% _disconnectz%_connect_psygnal.._disconnectTs   / 0r')rzpsygnal.EmissionInfor)r{r)rr)r#rrrrs ` @@r%rrCs;'s +F   ,  , 1 r'cttjjd}|t||jSdS)zAReturn `True` if an object is an instance of traitlets.HasTraits. traitletsF)rrrr HasTraits)r#rs r%rr^s1  ,I3<3H:c9.. /SeSr'syncc6tdi}|jdi|S)z0Get the state of a traitlets.HasTraits instance.Tr!)_TRAITLETS_SYNC_FLAG trait_values)r#r$rs r%rrns$#D )F 3   %f %%r'ct|sydfd |jt|jdt j |dfd }|S) aICheck if an object is a traitlets.HasTraits, and connect it to send_state. Only traits with tagged with `sync=True` will be synced. Returns ------- disconnect : Callable | None A callable that disconnects the traitlets.HasTraits from send_state, or None if no traitlets.HasTraits was found. Nc|dhy)Nrgr!)changers r%_on_trait_changez,_connect_traitlets.._on_trait_changesF6N#$r'T)r)namesc<}||jyyr ) unobserve)r#robj_refs r%rz'_connect_traitlets.._disconnects!i ? MM* + r')rr*r)r{r)robservelisttraitsrNr)r#rrrrs ` @@r%rrvsS  $%KK SZZTZ-B(CKDkk#G, r'cttjjd}|t||jSdS)z@Return `True` if an object is an instance of pydantic.BaseModel.pydanticF)rrrr BaseModel)r#rs r%rrs1{{z*H2:2F:c8-- .QEQr'cLtj|j|S)zGet the state of a pydantic BaseModel instance. To take advantage of pydantic's support for custom encoders (with json_encoders) we call obj.json() here, and then cast back to a dict (which is what the comm expects). r)jsonloadsrs r%rrs ::chhwh/ 00r'c(|jd|S)z4Get the state of a pydantic (v2) BaseModel instance.r)moder$)rrs r%rrs >>vw> 77r'cttjjd}|t||jSdS)z> *LuLr'cJddl}tt|j|S)z+Get the state of a msgspec.Struct instance.rN)rr r* to_builtins)r#r$rs r%rrs" )g))#. //r')rAr}r?r}r)comm.base_comm.BaseComm)r#r~r)r)r#r~r)r})r#r~r)r)r#r~r@r*r)r{)r#r~r)zCallable[[object, dict], None])r#r~r)zpsygnal.SignalGroup | None)r#r~rrr)zCallable | None)r#rr)zTypeGuard[traitlets.HasTraits])r#ztraitlets.HasTraitsr$r(r)r.)r#rr)zTypeGuard[pydantic.BaseModel])r#zpydantic.BaseModelr$r(r)r.)r#rr)zTypeGuard[msgspec.Struct])r#zmsgspec.Structr$r(r)r*)Lr __future__rrKrrrsrN dataclassesrrtypingrrrr r r r _file_contentsrr_utilrrrrrrrr_versionrrDrrrrtyping_extensionsrrr _protocolsrrr.__annotations____all__rZrr _TARGET_NAME_ANYWIDGET_MODEL_NAME_ANYWIDGET_VIEW_NAME_ANYWIDGET_JS_MODULErFrGrHrRr/r0rrrrrrrrrrrrrrrr!r'r%rs"# ,>   0@@'LHL"L)! !#3 4 ++ " "*(6(&5$4E  .1  .0*/*CCLQ,Q,n: -` (26T& &'6&&@R 1  1&5 1 18 8&588M 0r'