o tfL@sdZddlmZddlZddlmZddlmZmZm Z m Z m Z m Z m Z mZmZmZmZddlmZmZmZmZddlmZerTddlZdd lmZmZdd lmZmZd d gZGd d d eZ GdddeZ!eddGdd d Z"dddZ#dS)aA SignalGroup class that allows connecting to all SignalInstances on the class. Note that unlike a slot/callback connected to SignalInstance.connect, a slot connected to SignalGroup.connect does *not* receive the direct arguments that were emitted by a given SignalInstance. Instead, the slot/callback will receive an EmissionInfo named tuple, which contains `.signal`: the SignalInstance doing the emitting, and `.args`: the args that were emitted. ) annotationsN)MappingProxyType) TYPE_CHECKINGAnyCallableClassVarContextManagerIterableIteratorLiteralMapping NamedTupleoverload)_NULLSignalSignalInstance_SignalBlocker) mypyc_attr)F ReducerFunc)RefErrorChoice WeakCallback EmissionInfo SignalGroupc@s"eZdZUdZded<ded<dS)rzTuple containing information about an emission event. Attributes ---------- signal : SignalInstance args: tuple rsignalztuple[Any, ...]argsN)__name__ __module__ __qualname____doc____annotations__r"r"W/var/www/html/software/conda/envs/catlas/lib/python3.10/site-packages/psygnal/_group.pyr*s  cseZdZdZ d5d6fd d Zd7fd d Zd8ddZd9fdd Zd8ddZd:ddZ d5dddddd;d$d%Z dd.d/Z d?d@fd3d4 ZZS)A SignalRelayaSpecial SignalInstance that can be used to connect to all signals in a group. This class will rarely be instantiated by a user (or anything other than a SignalGroup). Parameters ---------- signals : Mapping[str, SignalInstance] A mapping of signal names to SignalInstance instances. instance : Any, optional An object to which this `SignalRelay` is bound, by default None NsignalsMapping[str, SignalInstance]instancerreturnNonecs&tjtf|dt||_i|_dS)N) signaturer')super__init__rr_signals_sig_was_blocked)selfr%r' __class__r"r#r,Es  zSignalRelay.__init__slotrcs*t|t|jdkr|dSdS)Nr)r+ _append_slotlen_slots_connect_relayr/r2r0r"r#r3Ls  zSignalRelay._append_slotcCs\t td|jD] }|j|jddddqWddS1s'wYdS)NignoreFT) check_nargs check_typesunique)warningscatch_warnings simplefilterr-valuesconnect _slot_relayr/sigr"r"r#r6Qs   "zSignalRelay._connect_relay#int | WeakCallback | Literal['all']cs"t||js|dSdSN)r+ _remove_slotr5_disconnect_relayr7r0r"r#rFZs  zSignalRelay._remove_slotcCs |jD]}||jqdSrE)r-r? disconnectrArBr"r"r#rG_szSignalRelay._disconnect_relayrcGs*t}rt||}||fdSdSrE)rZcurrent_emitterrZ_run_emit_loop)r/rZemitterinfor"r"r#rAcs  zSignalRelay._slot_relayFr9r:r;max_argsCallable | Noner9 bool | Noner:r; bool | strrK int | None)Callable[[Callable], Callable] | Callablecs*dfdd }|dur|S||S)aAConnect `slot` to be called whenever *any* Signal in this group is emitted. Params are the same as `psygnal.SignalInstance.connect`. It's probably best to check whether all signals are uniform (i.e. have the same signature). Parameters ---------- slot : Callable A callable to connect to this signal. If the callable accepts less arguments than the signature of this slot, then they will be discarded when calling the slot. check_nargs : bool | None If `True` and the provided `slot` requires more positional arguments than the signature of this Signal, raise `TypeError`. by default `True`. check_types : bool | None If `True`, An additional check will be performed to make sure that types declared in the slot signature are compatible with the signature declared by this signal, by default `False`. unique : bool | str If `True`, returns without connecting if the slot has already been connected. If the literal string "raise" is passed to `unique`, then a `ValueError` will be raised if the slot is already connected. By default `False`. max_args : int, optional If provided, `slot` will be called with no more more than `max_args` when this SignalInstance is emitted. (regardless of how many arguments are emitted). Returns ------- Union[Callable[[Callable], Callable], Callable] [description] r2rr(cs(jD] }|j|dq|SNrJ)r-r?r@)r2rCr9r:rKr/r;r"r#_innersz*SignalRelay.connect_direct.._innerN)r2rr(rr")r/r2r9r:r;rKrSr"rRr#connect_direcths+ zSignalRelay.connect_directr"excludeIterable[str | SignalInstance]csLt|jD]\}}|r||vs||vrq |j|j|<|q dS)z1Block this signal and all emitters from emitting.N)r+blockr-itemsZ _is_blockedr.)r/rUnamerCr0r"r#rWs   zSignalRelay.blockcs8t|jD]\}}|j|ds|q dS)zs  z(SignalGroup.__init__..)typehasattr TypeErrorrjrXrnr$_psygnal_relay)r/r'r"rqr#r, s  zSignalGroup.__init__strictrasignal_aliasesMapping[str, str | None]c sTddt|ddD}|rtd|i}t|D]}t||d}t|tr+|||<qit|di||_tttfdd|jD|_}|r|D]}tt||tr]t ||qNt |d krfd nd } t |d krpd nd } t j | dt |d| dtddt|di} i| ||_t|j|_|r|jstdtdS)zHCollects all Signal instances on the class under `cls._psygnal_signals`.cSsh|] }|dr|qS)_psygnal startswithrpkr"r"r# $s  z0SignalGroup.__init_subclass__..__dict__r"zMSignalGroup subclass cannot have attributes starting with '_psygnal'. Found: Nrjcs"h|] }|vs |dr|qS))r{Zpsygnalr|r~reservedr"r#r=s rNamesNameZareis z reserved. You cannot use these names to access SignalInstances as attributes on a SignalGroup. (You may still access them as keys to __getitem__: `group['name']`). stacklevelrmz@All Signals in a strict SignalGroup must have the same signature)getattrrvdir isinstancerrjsetrrldelattrr4r<warnsorted UserWarningrm _is_uniformr?rkr+__init_subclass__) rrrxry forbiddenrjrval conflictsrYrZArealiasesr0rr#rsX           zSignalGroup.__init_subclass__r$cC|jS)aSignalInstance that can be used to connect to all signals in this group. Examples -------- ```python from psygnal import Signal, SignalGroup class MySignals(SignalGroup): sig1 = Signal() sig2 = Signal() group = MySignals() group.sig2.connect(...) # connect to a single signal by name group.all.connect(...) # connect to all signals in the group ``` )rwr/r"r"r#allYszSignalGroup.allr&cC t|jS)z6A mapping of signal names to SignalInstance instances.)rrnrr"r"r#r%os zSignalGroup.signalsintcCr)zDReturn the number of signals in the group (not including the relay).)r4rnrr"r"r#__len__| zSignalGroup.__len__itemstrrcCs |j|S)Get a signal instance by name.rnr/rr"r"r# __getitem__rzSignalGroup.__getitem___SignalGroup__namecCstt|jd|)rz object has no attribute )AttributeErrorrtr)r/rr"r"r# __getattr__szSignalGroup.__getattr__ Iterator[str]cCr)z,Yield the names of all signals in the group.)iterrnrr"r"r#__iter__rzSignalGroup.__iter__cCs ||jvS)z?Return True if the group contains a signal with the given name.rrr"r"r# __contains__s zSignalGroup.__contains__cCs|jj}d|dt|dS)zReturn repr(self).z )r1rr4)r/rYr"r"r#__repr__szSignalGroup.__repr__cCr)@Return true if all signals in the group have the same signature.)rkrrr"r"r#psygnals_uniformszSignalGroup.psygnals_uniformcCstjdtdd|jS)rzuThe `is_uniform` method on SignalGroup is deprecated. Use `psygnals_uniform` instead. This will be an error in v0.11.rr)r<r FutureWarningrkrr"r"r# is_uniforms zSignalGroup.is_uniformmemodict[int, Any]cCst||jjdS)N)r')rtrwr')r/rr"r"r# __deepcopy__szSignalGroup.__deepcopy__.threadr9r:r;rK on_ref_errorpriorityr4threading.Thread | Literal['main', 'current'] | Noner9rMr:r;rNrKrOrrrCallable[[F], F]cCdSrEr")r/rr9r:r;rKrrr"r"r#r@s zSignalGroup.connectr2rc CrrEr" r/r2rr9r:r;rKrrr"r"r#r@s rrF | NoneCallable[[F], F] | Fc Cs>|dur|jj|||||||dS|jj||||||||dS)Nr)rwr@rr"r"r#r@s(  rJrLrPcCs|jj|||||dSrQ)rwrT)r/r2r9r:r;rKr"r"r#rTs zSignalGroup.connect_directTr`cC|jj||dS)N)r2r`)rwrH)r/r2r`r"r"r#rHzSignalGroup.disconnectr"rUrVcC|jj|dSNr])rwrWr^r"r"r#rW szSignalGroup.blockcC |jSrE)rwrZrr"r"r#rZ  zSignalGroup.unblockr\cCrr)rwr_r^r"r"r#r_szSignalGroup.blockedcCrrE)rwpauserr"r"r#rrzSignalGroup.pausereducerReducerFunc | NoneinitialcCrN)rr)rwresumer/rrr"r"r#rrzSignalGroup.resumecCrr)rwpausedrr"r"r#rszSignalGroup.pausedrE)r'rr(r))rxraryrzr(r))r(r$)r(r&)r(r)rrr(r)rrr(r)r(r)rrr(ra)r(r)r(ra)rrr(r)rrr9rMr:rMr;rNrKrOrrrrr(r)r2rrrr9rMr:rMr;rNrKrOrrrrr(r)r2rrrr9rMr:rMr;rNrKrOrrrrr(rrcrgrhrdrerbrf)rrrrr(r))rrrrr(r\)!rrrr r!rkr,rpropertyrr%rrrrrr classmethodrrrrr@rTrHrWrZr_rrrrrir"r"r0r#rs +  =            $    r%Iterable[Signal]r(racCsJt}|D]}tdd|jjD}|r||vrdS||qdS)z3Return True if all signals have the same signature.css|]}t|jVqdSrE)r annotation)rppr"r"r# $sz_is_uniform..FT)rtupler* parametersr?add)r%seensvr"r"r#r s  r)r%rr(ra)$r __future__rr<typesrtypingrrrrrr r r r r rZpsygnal._signalrrrrZ_mypycr threadingrrZpsygnal._weak_callbackrr__all__rr$rrr"r"r"r#s(  4  !J