3 ˆd‚Aã@s†ddlZddlZddlmZddlmZddlmZddlm Z ddl m Z Gdd„de ƒZ ejGd d „d eƒƒZGd d „d eƒZdS) éN)Úchain)Úref)Ú IObservable)Ú_validate_everything)Ú TraitErrorc@s(eZdZdZdddœdd„Zdd„ZdS)Ú TraitSetEventa‚ An object reporting in-place changes to a traits sets. Parameters ---------- removed : set, optional Old values that were removed from the set. added : set, optional New values added to the set. Attributes ---------- removed : set Old values that were removed from the set. added : set New values added to the set. N)ÚremovedÚaddedcCs,|dkrtƒ}||_|dkr"tƒ}||_dS)N)Úsetrr )Úselfrr ©r ú9/tmp/pip-build-7vycvbft/traits/traits/trait_set_object.pyÚ__init__'s zTraitSetEvent.__init__cCs|jj›d|j›d|j›dS)Nz (removed=z, added=ú))Ú __class__Ú__name__rr )r r r r Ú__repr__1szTraitSetEvent.__repr__)rÚ __module__Ú __qualname__Ú__doc__rrr r r r rs rcsüeZdZdZ‡fdd„Zffdddœ‡fdd„ Zdd „Z‡fd d „Z‡fd d „Z‡fdd„Z ‡fdd„Z ‡fdd„Z ‡fdd„Z ‡fdd„Z ‡fdd„Z‡fdd„Z‡fdd„Z‡fdd„Z‡fd d!„Z‡fd"d#„Zd$d%„Zd&d'„Zd(d)„Zd*d+„Z‡ZS),ÚTraitSetax A subclass of set that validates and notifies listeners of changes. Parameters ---------- value : iterable, optional Iterable providing the items for the set. item_validator : callable, optional Called to validate and/or transform items added to the set. The callable should accept a single item and return the transformed item, raising TraitError for invalid items. If not given, no item validation is performed. notifiers : list of callable, optional A list of callables with the signature:: notifier(trait_set, removed, added) Where 'added' is a set containing new values that have been added. And 'removed' is a set containing old values that have been removed. If this argument is not given, the list of notifiers is initially empty. Attributes ---------- item_validator : callable Called to validate and/or transform items added to the set. The callable should accept a single item and return the transformed item, raising TraitError for invalid items. notifiers : list of callable A list of callables with the signature:: notifier(trait_set, removed, added) where 'added' is a set containing new values that have been added and 'removed' is a set containing old values that have been removed. cstƒj|ƒ}t|_g|_|S)N)ÚsuperÚ__new__rÚitem_validatorÚ notifiers)ÚclsÚargsÚkwargsr )rr r r`s zTraitSet.__new__N)rrcs:|dk r|ˆ_tƒj‡fdd„|Dƒƒ|dk r6|ˆ_dS)Nc3s|]}ˆj|ƒVqdS)N)r)Ú.0Úitem)r r r ú isz$TraitSet.__init__..)rrrr)r Úvaluerr)r)r r rfs zTraitSet.__init__cCs x|jD]}||||ƒqWdS)a9 Call all notifiers. This simply calls all notifiers provided by the class, if any. The notifiers are expected to have the signature:: notifier(trait_set, removed, added) Any return values are ignored. Any exceptions raised are not handled. Notifiers are therefore expected not to raise any exceptions under normal use. Parameters ---------- removed : set The items that have been removed. added : set The new items that have been added to the set. N)r)r rr Únotifierr r r Únotifyms zTraitSet.notifycs<|jƒ}tƒj|ƒ}|j|ƒ}t|ƒdkr8|j|tƒƒ|S)zÓ Return self &= value. Parameters ---------- value : set or frozenset A value. Returns ------- self : TraitSet The updated set. r)ÚcopyrÚ__iand__Ú differenceÚlenr#r )r r!Úold_setÚretvalr)rr r r%…s    zTraitSet.__iand__cs\ˆjƒ}t|ttfƒr(‡fdd„|Dƒ}tƒj|ƒ}ˆj|ƒ}t|ƒdkrXˆjtƒ|ƒ|S)zÒ Return self |= value. Parameters ---------- value : set or frozenset A value. Returns ------- self : TraitSet The updated set. csh|]}ˆj|ƒ’qSr )r)rr)r r r ú °sz#TraitSet.__ior__..r) r$Ú isinstancer Ú frozensetrÚ__ior__r&r'r#)r r!r(r)r )r)r r r-œs     zTraitSet.__ior__cs<|jƒ}tƒj|ƒ}|j|ƒ}t|ƒdkr8|j|tƒƒ|S)zÐ Return self-=value. Parameters ---------- value : set or frozenset A value. Returns ------- self : TraitSet The updated set. r)r$rÚ__isub__r&r'r#r )r r!r(r)r)rr r r.¼s    zTraitSet.__isub__cs~tƒ}tƒ}t|ttfƒrZt|ƒ}ˆj|ƒ}|j|ƒ}‡fdd„|Dƒ}|jˆƒ}||B}tƒj|ƒ}|sn|rzˆj||ƒ|S)zÒ Return self ^= value. Parameters ---------- value : set or frozenset A value. Returns ------- self : TraitSet The updated set. csh|]}ˆj|ƒ’qSr )r)rr)r r r r*ìsz$TraitSet.__ixor__..)r r+r,Ú intersectionr&rÚ__ixor__r#)r r!rr ÚvaluesZ raw_addedZvalidated_addedr))r)r r r0Ós      zTraitSet.__ixor__cs6|j|ƒ}||k}tƒj|ƒ|s2|jtƒ|hƒdS)zÅ Add an element to a set. This has no effect if the element is already present. Parameters ---------- value : any The value to add to the set. N)rrÚaddr#r )r r!Ú value_in_self)rr r r2øs  z TraitSet.addcs(t|ƒ}tƒjƒ|r$|j|tƒƒdS)z$ Remove all elements from this set. N)r rÚclearr#)r r)rr r r4 s zTraitSet.clearcs,||k}tƒj|ƒ|r(|j|htƒƒdS)zÊ Remove an element from the set if it is a member. If the element is not a member, do nothing. Parameters ---------- value : any An item in the set N)rÚdiscardr#r )r r!r3)rr r r5s  zTraitSet.discardcs<|jƒ}tƒj|Ž|j|ƒ}t|ƒdkr8|j|tƒƒdS)z Remove all elements of another set from this set. Parameters ---------- args : iterables The other iterables. rN)r$rÚdifference_updater&r'r#r )r rr(r)rr r r6"s    zTraitSet.difference_updatecs<|jƒ}tƒj|Ž|j|ƒ}t|ƒdkr8|j|tƒƒdS)z« Update the set with the intersection of itself and another set. Parameters ---------- args : iterables The other iterables. rN)r$rÚintersection_updater&r'r#r )r rr(r)rr r r72s    zTraitSet.intersection_updatecstƒjƒ}|j|htƒƒ|S)a  Remove and return an arbitrary set element. Raises KeyError if the set is empty. Returns ------- item : any An element from the set. Raises ------ KeyError If the set is empty. )rÚpopr#r )r r)rr r r8Bs z TraitSet.popcs tƒj|ƒ|j|htƒƒdS)a2 Remove an element that is a member of the set. If the element is not a member, raise a KeyError. Parameters ---------- value : any An element in the set Raises ------ KeyError If the value is not found in the set. N)rÚremover#r )r r!)rr r r9Vs zTraitSet.removecs`t|ƒ}ˆj|ƒ}|j|ƒ}‡fdd„|Dƒ}|jˆƒ}tƒj||Bƒ|sP|r\ˆj||ƒdS)z Update the set with the symmetric difference of itself and another. Parameters ---------- value : iterable csh|]}ˆj|ƒ’qSr )r)rr)r r r r*tsz7TraitSet.symmetric_difference_update..N)r r/r&rÚsymmetric_difference_updater#)r r!r1rZ raw_resultZvalidated_resultr )r)r r r:is   z$TraitSet.symmetric_difference_updatecsL‡fdd„tj|ƒDƒ}|jˆƒ}tƒj|ƒt|ƒdkrHˆjtƒ|ƒdS)zž Update the set with the union of itself and others. Parameters ---------- args : iterables The other iterables. csh|]}ˆj|ƒ’qSr )r)rr)r r r r*„sz"TraitSet.update..rN)rÚ from_iterabler&rÚupdater'r#r )r rZvalidated_valuesr )r)r r r<{s    zTraitSet.updatecs*t‡fdd„|Dƒtj|jˆƒgd}|S)zb Perform a deepcopy operation. Notifiers are transient and should not be copied. csg|]}tj|ˆƒ‘qSr )r$Údeepcopy)rÚx)Úmemor r ú •sz)TraitSet.__deepcopy__..)rr)rr$r=Z validator)r r?Úresultr )r?r Ú __deepcopy__Žs  zTraitSet.__deepcopy__cCs|jjƒ}|d=|S)zw Get the state of the object for serialization. Notifiers are transient and should not be serialized. r)Ú__dict__r$)r rAr r r Ú __getstate__œs zTraitSet.__getstate__cCsg|d<|jj|ƒdS)zƒ Restore the state of the object after serialization. Notifiers are transient and are restored to the empty list. rN)rCr<)r Ústater r r Ú __setstate__¦szTraitSet.__setstate__cCs|jS)a Return a list of callables where each callable is a notifier. The list is expected to be mutated for contributing or removing notifiers from the object. Parameters ---------- force_create: boolean Not used here. )r)r Z force_creater r r Ú _notifiers°s zTraitSet._notifiers)rrrrrrr#r%r-r.r0r2r4r5r6r7r8r9r:r<rBrDrFrGÚ __classcell__r r )rr r9s*%    %           rcsVeZdZdZ‡fdd„Zdd„Zdd„Zdd „Z‡fd d „Zd d „Z ddd„Z ‡Z S)ÚTraitSetObjectaŠ A specialization of TraitSet with a default validator and notifier for compatibility with Traits versions before 6.0. Parameters ---------- trait : CTrait The trait that the set has been assigned to. object : HasTraits The object this set belongs to. Can also be None in cases where the set has been disconnected from its HasTraits parent. name : str The name of the trait on the object. value : iterable The initial value of the set. Attributes ---------- trait : CTrait The trait that the set has been assigned to. object : callable A callable that when called with no arguments returns the HasTraits object that this set belongs to, or None if there is no such object. name : str The name of the trait on the object. value : iterable The initial value of the set. csX||_|dkrdd„nt|ƒ|_||_d|_|jr<|d|_tƒj||j|j gddS)NcSsdS)Nr r r r r ÚÝsz)TraitSetObject.__init__..Z_items)rr) ÚtraitrÚobjectÚnameÚ name_itemsZ has_itemsrrÚ _validatorr")r rKrLrMr!)rr r rÚs  zTraitSetObject.__init__cCsŠt|ddƒ}t|ddƒ}|dks(|dkr,|S|ƒ}|jj}|dkrF|Sy|||j|ƒStk r„}z|jdƒ|‚WYdd}~XnXdS)al Validates the value by calling the inner trait's validate method. Parameters ---------- value : any The value to be validated. Returns ------- value : any The validated value. Raises ------ TraitError On validation failure for the inner trait. rLNrKzEach element of the)ÚgetattrZ item_traitÚvalidaterMrZ set_prefix)r r!Z object_refrKrLrQZexcpr r r rOæs   zTraitSetObject._validatorcCs`|jdkrdS|jƒ}|dkr"dSt||jƒ|k r6dSt||d}|jjƒ}|j|j||ƒdS)aE Converts and consolidates the parameters to a TraitSetEvent and then fires the event. Parameters ---------- trait_set : set The complete set removed : set Set of values that were removed. added : set Set of values that were added. N)rr )rNrLrPrMrrKÚ items_eventZtrait_items_event)r Z trait_setrr rLÚeventrRr r r r" s   zTraitSetObject.notifiercs$t|jd|j‡fdd„|Dƒƒ}|S)zb Perform a deepcopy operation. Notifiers are transient and should not be copied. Ncsh|]}tj|ˆƒ’qSr )r$r=)rr>)r?r r r*7sz.TraitSetObject.__deepcopy__..)rIrKrM)r r?rAr )r?r rB-s zTraitSetObject.__deepcopy__cstƒjƒ}|d=|d=|S)zw Get the state of the object for serialization. Notifiers are transient and should not be serialized. rLrK)rrD)r rA)rr r rD<s zTraitSetObject.__getstate__cCs<|jddƒ|jg|d<dd„|d<d|d<|jj|ƒdS) zƒ Restore the state of the object after serialization. Notifiers are transient and are restored to the empty list. rMÚrcSsdS)Nr r r r r rJOsz-TraitSetObject.__setstate__..rLNrK)Ú setdefaultr"rCr<)r rEr r r rFGs    zTraitSetObject.__setstate__NcCstjt|ƒtt|ƒf|jƒfS)zB Overridden to make sure we call our custom __getstate__. )ÚcopyregÚ_reconstructorÚtyper ÚlistrD)r Úprotocolr r r Ú __reduce_ex__SszTraitSetObject.__reduce_ex__)N) rrrrrrOr"rBrDrFr[rHr r )rr rI½s '   rI)r$rVÚ itertoolsrÚweakrefrZtraits.observation.i_observablerZtraits.trait_baserZtraits.trait_errorsrrLrÚregisterr rrIr r r r Ú s     $