:vjwVUdZddlmZddlZddlZddlZddlZddlZddlm Z m Z ddl m Z m Z mZmZddlmZmZddlmZmZmZmZmZmZmZmZdd lmZe rDdd lmZm Z m!Z!dd l m"Z"m#Z#ddl$Z$ddl%Z%ddl&Z&ddl'Z'ddl(Z(dd l)m*Z*dd l+m,Z,Gdde*Z-e Z.de/d<ddgZ0dZ1dZ2dZ3dZ4d/dZ5d0dZ6ef d1dZ7iZ8de/d< d2dZ9GddZ:GddZ;d3d Zd6d#Z?d7d$Z@d8d%ZAd9d&ZBd'ZC d:d(ZDd8d)ZEd;d*ZF dd.ZJy)?aDescriptor for _repr_mimebundle_ attribute that manages the comm channel. `MimeBundleDescriptor()` takes the place of a `_repr_mimebundle_` method on a class. - `MimeBundleDescriptor` is a [descriptor](https://docs.python.org/3/howto/descriptor.html). When the `_repr_mimebundle_` attribute is accessed on an instance of the decorated class, a `ReprMimeBundle` instance is created and returned. - A `ReprMimeBundle` is first and foremost a callable object that implements the `_repr_mimebundle_` protocol that jupyter expects. However, it also manages a Comm instance that is used to send the state of the python model to the javascript view. This is done lazily, so that the Comm is only created when the `_repr_mimebundle_` is first accessed. - `ReprMimeBundle` has the logic to get/set the state of the python model, and will keep the two in sync ("bind"/"unbind_instance" methods can be used to control this). ) annotationsN)asdict is_dataclass) TYPE_CHECKINGAnycastoverload) FileContentsVirtualFileContents)_ANYWIDGET_ID_KEY _DEFAULT_ESM_ESM_KEY_PROTOCOL_VERSION put_buffersremove_buffersrepr_mimebundletry_file_contents)_ANYWIDGET_SEMVER_VERSION)CallableIterableSequence) TypeAlias TypeGuard)Protocol) CommMessageceZdZddZy) _GetStatecyN)selfobjincludes f/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/anywidget/_descriptor.py__call__z_GetState.__call__<sN)r#rr$set[str] | Nonereturndict)__name__ __module__ __qualname__r&r!r'r%rr;sKr'rr SerializableMimeBundleDescriptorReprMimeBundle_repr_mimebundle__get_anywidget_state_set_anywidget_statez anywidget:ct|dr&t|jtr |jSt |t d}t|t r|j|t|}t|tr |jSy)uGet model_id from an anywidget object, or None if not an anywidget. .. note:: For protocol objects backed by a ``MimeBundleDescriptor`` (e.g. dataclasses), this forces the descriptor's ``__get__`` to run, which opens the object's comm. ``ipywidgets``-based widgets open their comm eagerly in ``__init__``, so the ``hasattr(obj, "model_id")`` branch short-circuits without side effects for them. The descriptor-driven side effect only matters if a protocol-based widget is assigned to a ``WidgetTrait`` (or otherwise inspected) and its parent is never displayed — otherwise the comm would have opened at display time anyway. model_idN) hasattr isinstancer5strgetattr _REPR_ATTRr/__get__typer0)r#bundles r%_try_get_model_idr>IsisJJs||S$A|| S*d +F&./T#Y/&.) r'cpdfd |jDcic]\}}||c}}Scc}}w)zJRecursively replace anywidget objects with 'anywidget:' strings.cTt|}| t|St|tr*|j Dcic]\}}||c}}St|t r|Dcgc] }| c}St|t rt fd|DS|Scc}}wcc}w)Nc3.K|] }|ywr r!).0item_replaces r% z9_replace_widget_refs.._replace..ps6AD$As)r>_WIDGET_REF_PREFIXr7r*itemslisttuple)vr5kvalrCrDs r%rDz&_replace_widget_refs.._replacegs$Q'  ()(4 4 a 34779=9CAx}$9= = a /01qtHTNq1 1 a 6A66 6 >1s B'B%)rJobjectr)rM)rG)r#rKrJrDs @r%_replace_widget_refsrNds4 (+yy{ 3{tq!Ax{N{ 33 3s2c ddl}t|}t|\}}}|jdd|iddtddtdd||d| S) Nrzjupyter.widgetversion anywidgetAnyModelAnyView) _model_module _model_name_model_module_version _view_module _view_name_view_module_version _view_count)state buffer_paths) target_namemetadatadatabuffers)commrNr create_commr) initial_staterPrar[r\r`s r% open_commrdvsw(7M#1-#@ E< 4  $W%"-))B +'(A#  )  ! r'z"dict[int, comm.base_comm.BaseComm]_COMMSc2t|}|tvrht|t|<tjt 5t j|tj|dddt|St|S#1swYt|SxYw)zGet or create a communication channel for a given object. Comms are cached by object id, so that if the same object is used in multiple places, the same comm will be used. Comms are deleted when the object is garbage collected. )rcN) idrerd contextlibsuppress TypeErrorweakreffinalizepop)r# get_stateobj_ids r%_get_or_create_commrpswWF V"=v   +   S&**f 5, &>6&> , &>s &BBcteZdZdZdddd 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_stater:_name_follow_changes_autodetect_observer_no_viewrGr)r"rrrsrt extra_staterKrJ file_contentss r%__init__zMimeBundleDescriptor.__init__sy x6' -$7! %%++-DAq.a0M('4!!!$ .r'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)rx)r"ownernames r% __set_name__z!MimeBundleDescriptor.__set_name__s  r'cyr r!r"instancers r%r;zMimeBundleDescriptor.__get__sLOr'cyr r!rs r%r;zMimeBundleDescriptor.__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)rsr|rtzError in Anywidget repr: r  stacklevel)r0rzrwr{rysync_object_with_view ExceptionwarningswarnrhriAttributeError ValueErrorsetattrrx)r"rrrepr_objes r%r;zMimeBundleDescriptor.__get__s  K %$($=$= -- H ##..0  < Hdjj( 3 =#  MM6qc:q I   =s$A B-B; B8B33B8;CN) rrboolrsrrtrr|rMr)None)rr<rr8r)r)rrrr<r)r/)rrMrr<r)r0)rz object | Nonerr<r)z%ReprMimeBundle | MimeBundleDescriptor)r+r,r-__doc__r~rr r;r!r'r%r/r/s-d $$( 55" 5  5  5 5,OO KK111 / 1r'ceZdZdZ d d dZd ddZd ddZddZeddZ ddZ d dd Z 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%_jj)D]j\}}t+|t,t.fr9t1|j|<|j2j4|fd fd }t7fd_ly#t$r(fd_ tjdddYwxYw) NcSr r!r#sr%z)ReprMimeBundle.__init__..isr'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.rcF|j|<j|yr )rw send_state) new_contentskeyr"s r% _on_changez+ReprMimeBundle.__init__.._on_change}s-9D%%c*OOC(r'cFijdjS)Nr$) _get_staterw)r#r"sr%rz)ReprMimeBundle.__init__..s+#ooc4o8#''#r')r#rn)rr8rr8r)r)rzcopyrwrvr _anywidget_idr{rkref_on_obj_deleted_objrjrrset_disconnectorsdetermine_state_getterrdetermine_state_setter _set_staterGr7r r r8changedconnectrp_comm)r"r#rsr|rtrvaluers`` r%r~zReprMimeBundle.__init__Ys;%8!(.B446 $$%6 c8JK  .5kk#t?S?S.TDI.1U1505++113JC%"5|!DE),U!!#&&&=@)')- DJ4% #DI MMcU#KK   s%D00.E! E!cX|j|jjy)z)Called when the python object is deleted.N)unsync_object_with_viewrclose)r"rs r%rzReprMimeBundle._on_obj_deleteds $$& r'c|j}|y|t|tr|hn t|}i|j |||j }|(|j Dcic] \}}||vs ||}}}|syt|}t|\}}}t|jddr$d||d}|jj||yycc}}w)zSend state update to the front-end view. Parameters ---------- include : set of str, optional If provided, only send the state for the keys in this set. Otherwise, send all state. Nrkernelupdate)methodr[r\)r_r`) rr7r8rrrwrGrNrr9rsend) r"r$r#r[rKrJr\r`msgs r%rzReprMimeBundle.send_statesiik ;   #-gs#;wiWGN4??3?8NDX %$W !T)tN';S^LU+  (^ . OO (X'78<< W% %r'c.|jjS)z,The unique model identifier for this widget.)rcomm_idr"s r%r5zReprMimeBundle.model_idszz!!!r'c |jryt|jjt |j S)z=Called when _repr_mimebundle_ is called on the python object.N)r5 repr_text)r{rrrreprr)r"kwargss r%r&zReprMimeBundle.__call__s0 == (:(:d499;FWXXr'c|r5|jj|j|j|r|jr|j }| d}t ||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. Raises ------ RuntimeError If the object has been deleted. NzCannot sync a deleted objectz$Refusing to re-sync a synced object.rrzCould not find a notifier on z^ (e.g. psygnal, traitlets). Changes to the python object will not be reflected in the JS view.) ron_msgrrrzr RuntimeErrorrrr_connect_psygnal_connect_traitletsadd)r"py_to_jsjs_to_pyr#r connector disconnects r%rz$ReprMimeBundle.sync_object_with_views*  JJ  d.. / OO  11))+C{4"3''"" DQRS/0BC &sDOO< ''++J7 D  3C59 ! /28r'c|jjd|jrOtjt 5|jj ddd|jrNyy#1swYxYw)zEDisconnect the view from changes in a model instance, and vice versa.N)rrrrhrirrmrs r%rz&ReprMimeBundle.unsync_object_with_view"sY $!!$$Y/)##'')+0!!//s  A88B)TNF) r#rMrsrr|zdict[str, object] | Nonertrr)rr )rzweakref.ReferenceType | Noner)r)r$zstr | Iterable[str] | Noner)r)rrr)r)r)r8)rz Sequence[str]r)ztuple[dict, dict] | None)TT)rrrrr)rr)r) r+r,r-rr~rrrpropertyr5r&rrr!r'r%r0r09sD%)04 1 1"1. 1  1  1f D"&R""Y666  6p,r'c\t|jdt|jS)z9Return a unique id for an object, to send to the JS side..)r<r,r+rs r%rr.s)3i"" #1T#Y%7%7$8 99r'c.tt|trtt|tSt |rdSt |rt St|rt|drtStSt|rtSd|d}t|)aQAutodetect 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. Raises ------ TypeError If no state-getting method can be determined. ct|Sr )rr#r$s r%rz(determine_state_getter..PsF3Kr' model_dumpz,Cannot determine a state-getting method for zI. Please implement a `_get_anywidget_state()` method that returns a dict.) r6r<_STATE_GETTER_NAMEr9r_is_traitlets_object_get_traitlets_state_is_pydantic_model_get_pydantic_state_v2_get_pydantic_state_v1_is_msgspec_struct_get_msgspec_staterj)r#rs r%rr3s*tCy,-tCy"455C0/C ### 3 %) )%%#!! 7sg>R R C.r'cN|jD]\}}t|||y)z?A default state setter that just sets attributes on the object.N)rGr)r#r[rrLs r%_default_set_stateris"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. )r6r<_STATE_SETTER_NAMEr9rrs r%rros+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)rrsysmodulesgetr9r7 SignalGrouprhrirRecursionErrorrjvarsvalues)r#rrattrs r%_get_psygnal_signal_grouprs++//),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 )signalr)eventrs r%_on_psygnal_eventz+_connect_psygnal.._on_psygnal_events  ))* +r'c(jyr )r)rrsr% _disconnectz%_connect_psygnal.._disconnects   / 0r')rzpsygnal.EmissionInfor)rr)rr)r#rrrrs ` @@r%rrs;'s +F   ,  , 1 r'cttjjd}|t||jSdS)zAReturn `True` if an object is an instance of traitlets.HasTraits. traitletsF)rrrr7 HasTraits)r#rs r%rrs1  ,I3<3H:c9.. /SeSr'syncc6tdi}|jdi|S)zGet the state of a traitlets.HasTraits instance. Returns ------- state : dict A dictionary of the state of the traitlets.HasTraits instance. Tr!)_TRAITLETS_SYNC_FLAG trait_values)r#r$rs r%rrs$#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)Nrr!)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)rr)robserverHtraitsrkr)r#rrrrs ` @@r%rrsS  $%KK SZZTZ-B(CKDkk#G, r'cttjjd}|t||jSdS)zWhether an object is an instance of pydantic.BaseModel. Returns ------- `True` if the object is an instance of pydantic.BaseModel, `False` otherwise. pydanticF)rrrr7 BaseModel)r#r s r%rrs3{{z*H2:2F:c8-- .QEQr'cLtj|j|S)aSGet 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). Returns ------- state : dict A dictionary copy of state from the pydantic BaseModel 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'cBddl}td|j|S)z+Get the state of a msgspec.Struct instance.rNr*)rr to_builtins)r#r$rs r%rr&s" +++C0 11r')r#rMr)z str | None)r#r*r)r*)rcr*rPr8r)comm.base_comm.BaseComm)r#rMrnzCallable[[], dict]r)r)r#rMr)r8)r#rMr)r)r#rMr[r*r)r)r#rMr)zCallable[[object, dict], None])r#rMr)zpsygnal.SignalGroup | None)r#rMrrr)zCallable | None)r#rMr)zTypeGuard[traitlets.HasTraits])r#ztraitlets.HasTraitsr$r(r)r.)r#rMr)zTypeGuard[pydantic.BaseModel])r#zpydantic.BaseModelr$r(r)r.)r#rMr)zTypeGuard[msgspec.Struct])r#zmsgspec.Structr$r(r)r.)Kr __future__rrhrrrrk dataclassesrrtypingrrrr _file_contentsr r _utilr rrrrrrr_versionrcollections.abcrrrrrrarrr rtyping_extensionsr _protocolsrrr.__annotations____all__r:rrrFr>rNrdrerpr/r0rrrrrrrrrrrrrrrr!r'r%r#s"# ,>   0<<+*'LHL"L)! !#3 4 ++!64(% @.0*/ ..FFRo,o,j: 3l (66T &  &  & &@R1 1 11$8 8 88M 2r'