o Fh @sHddlmZddlmZmZmZmZmZmZm Z m Z ddl m Z m Z er*ddlmZeddGdddZeddGd d d ZerLGd d d eZGd ddZdedeeeffddZdedeeeffdeeeffddZdddedeeeffdedeedefddZdedeeeffde defddZdS) ) dataclass) TYPE_CHECKINGAnyCallableDictOptionalUnioncastoverload)AttrDict MagicFilter) HandlerObjectT)frozenc@seZdZUeed<eed<dS)FlagnamevalueN)__name__ __module__ __qualname__str__annotations__rrrK/var/www/html/venv/lib/python3.10/site-packages/aiogram/dispatcher/flags.pyr s  rc @seZdZUeed<ededdfddZdeddfddZe de defde deffd d Z e deddfd d Z e d eddfd d Z dde ed ede e defdffdd Z dS) FlagDecoratorflagreturncCs||SNr)clsrrrr _with_flagszFlagDecorator._with_flagrcCst|jj|}||Sr)rrrr)selfrnew_flagrrr _with_values zFlagDecorator._with_value.cCdSrrrrrrr__call__zFlagDecorator.__call__cCr"rrr#rrrr$ r%kwargscKr"rr)rr&rrrr$$r%NcKsl|r|rtd|dur(t|r(it||jj|jji|_ttdt f|S| |dur3t |S|S)Nz;The arguments `value` and **kwargs can not be used together.) ValueErrorcallableextract_flags_from_objectrrr aiogram_flagr rrr!r )rrr&rrrr$(s r)rrrrr classmethodrrr!r rr$rrrrrrrs& $rc @s2eZdZ   d dededededef ddZd S) _ChatActionFlagProtocol.actioninterval initial_sleepr&rcKr"rr)rr-r.r/r&rrrr$<sz _ChatActionFlagProtocol.__call__N)...)rrrrfloatrrr$rrrrr,;sr,c@s0eZdZUdedefddZereed<dSdS) FlagGeneratorrrcCs"|ddkr tdtt|dS)Nr_z(Flag name must NOT start with underscoreT)AttributeErrorrr)rrrrr __getattr__Gs zFlagGenerator.__getattr__ chat_actionN) rrrrrr4rr,rrrrrr1Fs  r1objrcCs"t|dsiSttttf|jS)Nr*)hasattrr rrrr*)r6rrrr)Ps r)handlerr cCs.t|tr d|vr |d}t|dr|jSiS)z Extract flags from handler or middleware context data :param handler: handler object or data :return: dictionary with all handler flags r8flags) isinstancedictr7r9)r8rrr extract_flagsVs  r<N)defaultrr=cCst|}|||S)z Get flag by name :param handler: handler object or data :param name: name of the flag :param default: default value (None) :return: value of the flag or default )r<get)r8rr=r9rrrget_flagds r?magiccCst|}|t|S)z Check flags via magic filter :param handler: handler object or data :param magic: instance of the magic :return: the result of magic filter check )r<resolver )r8r@r9rrr check_flagsvsrB) dataclassesrtypingrrrrrrr r magic_filterr r aiogram.dispatcher.event.handlerr rrr,r1rr)r<r?rBrrrrs2 ( (  * *