o 4j @sddlmZddlmZddlmZddlmZmZm Z m Z ddl m Z m Z er.ddlmZeddGd d d ZeddGd d d ZerPGd ddeZGdddZd'ddZd(ddZddd)d!d"Zd*d%d&ZdS)+) annotations)Callable) dataclass) TYPE_CHECKINGAnycastoverload)AttrDict MagicFilter) HandlerObjectT)frozenc@seZdZUded<ded<dS)FlagstrnamervalueN)__name__ __module__ __qualname____annotations__rrU/var/www/kodo/Anonymous/send/lib/python3.10/site-packages/aiogram/dispatcher/flags.pyr s  r c@sfeZdZUded<edddZddd Zedd d Zedd d Zeddd Z dddd ZdS) FlagDecoratorr flagreturncCs||SNr)clsrrrr _with_flagszFlagDecorator._with_flagrrcCst|jj|}||Sr)r rrr)selfrnew_flagrrr _with_values zFlagDecorator._with_valueCallable[..., Any]cCdSrrrrrrr__call__zFlagDecorator.__call__cCr!rrr"rrrr##r$kwargscKr!rr)rr%rrrr#'r$N Any | None"Callable[..., Any] | FlagDecoratorcKsp|r |r d}t||dur*t|r*it||jj|jji|_ttdt f|S| |dur5t |S|S)Nz;The arguments `value` and **kwargs can not be used together.) ValueErrorcallableextract_flags_from_objectrrr aiogram_flagrrrrr )rrr%msgrrrr#+s )rr rr)rrrr)rr rr )r%rrrr)rr&r%rrr') rrrr classmethodrrrr#rrrrrs      rc@seZdZ   ddd d Zd S)_ChatActionFlagProtocol.actionrintervalfloat initial_sleepr%rrrcKr!rr)rr/r0r2r%rrrr#@sz _ChatActionFlagProtocol.__call__N)...) r/rr0r1r2r1r%rrr)rrrr#rrrrr.?s r.c@s(eZdZUd ddZerded<d Sd S) FlagGeneratorrrrrcCs&|ddkr d}t|tt|dS)Nr_z(Flag name must NOT start with underscoreT)AttributeErrorrr )rrr,rrr __getattr__Ks zFlagGenerator.__getattr__r. chat_actionN)rrrr)rrrr6rrrrrrr3Js  r3objrrdict[str, Any]cCs"t|dsiSttttf|jS)Nr+)hasattrrdictrrr+)r8rrrr*Us r*handlerHandlerObject | dict[str, Any]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 r<flags) isinstancer;r:r>)r<rrr extract_flags[s  r@N)defaultrrrAr&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)r<rrAr>rrrget_flagis rCmagicr cCst|}|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 )r<rDr>rrr check_flags{srF)r8rrr9)r<r=rr9)r<r=rrrAr&rr)r<r=rDr rr) __future__rcollections.abcr dataclassesrtypingrrrr magic_filterr r aiogram.dispatcher.event.handlerr r rr.r3r*r@rCrFrrrrs&   )