a 9g @s:ddlmZddlmZmZmZmZmZmZm Z m Z ddl m Z m Z erTddlmZeddGdddZeddGd d d ZerGd d d eZGd ddZeeeefdddZedeeeffeeefdddZddedeeeffeeeedddZedeeeffe edddZdS)) dataclass) TYPE_CHECKINGAnyCallableDictOptionalUnioncastoverload)AttrDict MagicFilter) HandlerObjectT)frozenc@seZdZUeed<eed<dS)FlagnamevalueN)__name__ __module__ __qualname__str__annotations__rrru/var/www/vosh/data/www/fastworkle.ru/webshop-tgbot-v.1.0/venv/lib/python3.9/site-packages/aiogram/dispatcher/flags.pyr s rc@seZdZUeed<eeddddZeddddZe e defe defdd d Z e eddd d Z e edd d d Z de eee e defdfddd Z dS) FlagDecoratorflag)rreturncCs||SNr)clsrrrr _with_flagszFlagDecorator._with_flag)rrcCst|jj|}||Sr)rrrr)selfrZnew_flagrrr _with_valueszFlagDecorator._with_value.cCsdSrrrrrrr__call__szFlagDecorator.__call__cCsdSrrr!rrrr" s)kwargsrcKsdSrr)rr#rrrr"$sN)rr#rcKsj|r|rtd|durPt|rPit||jj|jji|_ttdt f|S| |durdt |n|S)Nz;The arguments `value` and **kwargs can not be used together.) ValueErrorcallableextract_flags_from_objectrrr aiogram_flagr rrr r )rrr#rrrr"(s )N)rrrrr classmethodrrr r rr"rrrrrrrs "rc@s$eZdZdeeeeedddZdS)_ChatActionFlagProtocol.)actioninterval initial_sleepr#rcKsdSrr)rr*r+r,r#rrrr"<sz _ChatActionFlagProtocol.__call__N)...)rrrrfloatrrr"rrrrr);sr)c@s*eZdZUeedddZer&eed<dS) FlagGenerator)rrcCs"|ddkrtdtt|dS)Nr_z(Flag name must NOT start with underscoreT)AttributeErrorrr)rrrrr __getattr__Gs zFlagGenerator.__getattr__Z chat_actionN) rrrrrr1rr)rrrrrr.Fs r.)objrcCs"t|dsiSttttf|jS)Nr')hasattrr rrrr')r2rrrr&Ps r&r )handlerrcCs.t|trd|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 r4flags) isinstancedictr3r5)r4rrr extract_flagsVs  r8N)default)r4rr9rcCst|}|||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 )r8get)r4rr9r5rrrget_flagdsr;)r4magicrcCst|}|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 )r8resolver )r4r<r5rrr check_flagsvsr>)Z dataclassesrtypingrrrrrrr r Z magic_filterr r Z aiogram.dispatcher.event.handlerr rrr)r.rr&r8r;r>rrrrs* ( (  (