,hbvddlmZddlZddlmZddlmZmZddlm Z m Z ddl m Z m Z mZmZmZmZmZmZmZmZddlmZddlmZdd lmZdd lmZdd lmZm Z dd l!m"Z"dd l#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,ddl-m.Z.ddl/m0Z0m1Z1ddl2m3Z3m4Z4GddZ5GddZ6Gdde Z7GddZ8GddZ9eGdd Z:d2d!Z;Gd"d#Z<Gd$d%Z=Gd&d'Z>Gd(d)Z?Gd*d+Z@eGd,d-ZAGd.d/ZBGd0d1ZCeCZDy)3) annotationsN) defaultdict) dataclassreplace)Enumauto) AnyClassVarDictListMappingOptionalTupleTypeUnionoverload)Self)loggers) Dispatcher)NextMiddlewareType)CallableObject CallbackType)extract_flags_from_object)Router)SceneException) StateFilter) FSMContext)State)MemoryStorageRecord)TelegramObjectUpdate)ClassAttrsResolverget_sorted_mro_attrs_resolvercVeZdZd d dZd dZddZddZddZddZddZ d dZ dd Z y )HistoryManagerc||_||_t|jt |j ||_y)N)destiny)storagekey)_size_staterr(rr)_history_state)selfstater'sizes D/var/www/html/venv/lib/python3.12/site-packages/aiogram/fsm/scene.py__init__zHistoryManager.__init__)s1  (MMwuyy''J c~K|jjd{}|jdg}|j||dt ||j kDr||j d}t jjd|||jj|d{y77w)Nhistoryr.dataz Push state=%s data=%s to historyr4) r,get_data setdefaultappendlenr*rscenedebug update_data)r-r.r6 history_datar4s r0pushzHistoryManager.push0s!0099;; )))R856 w<$** $tzzkm,G >tL!!--g->>>< ?s"B=B9BB=3B;4B=;B=cK|jjd{}|jdg}|sy|j}|d}|d}|s$|jj id{n$|jj |d{t jjd||t||S77Y76w)Nr4r.r6r7z!Pop state=%s data=%s from historyr5) r,r8r9popset_datar>rr<r=r)r-r?r4recordr.r6s r0rBzHistoryManager.pop:s!0099;; )))R8wf~%%..r2 2 2%%11'1B B B ?M"T::< 3 Bs4CC AC4C5$CC3CCCcK|jjd{}|jdg}|sytdi|dS7'w)Nr4r,r8r9r)r-r?r4s r0getzHistoryManager.getIsL!0099;; )))R8"1WR[11 AAAcK|jjd{}|jjd{}|j||d{y7A7!7 wN)r+ get_stater8r@r-r.r6s r0snapshotzHistoryManager.snapshotYsQkk++--[[))++iit$$$.+$s3A(A"!A(A$A(A&A($A(&A(cK|jj|d{|jj|d{y7(7wrQ)r+ set_staterCrSs r0 _set_statezHistoryManager._set_state^s?kk##E***kk""4((( +(s!AA "AA A Ac^K|jd{}|s|jdid{ytjj d|j |j |j|j |j d{|j S77v7w)NzRollback to state=%s data=%s)rBrWrr<r=r.r6)r-previous_states r0rollbackzHistoryManager.rollbackbs#xxz)//$+ + +  *     oon22N4G4GHHH###* + Is2B-B'B-B)A#B-B+B-)B-+B-N)scenes_history )r.rr'strr/int)r. Optional[str]r6Dict[str, Any]returnNone)razOptional[MemoryStorageRecord])razList[MemoryStorageRecord]rarb)rar_) __name__ __module__ __qualname__r1r@rBrIrLrOrTrWrZrGr2r0r%r%(s0 ? ;2A /% ) $r2r%cdeZdZ d d dZd dZd dZddZddZddZddZ dd Z y)ObserverDecoratorNc<||_||_||_||_yrQ)namefiltersactionafter)r-rjrkrlrms r0r1zObserverDecorator.__init__rs     r2ct|dd}|sg}t|d||jt|j||j |j y)N__aiogram_handler__rjhandlerrkrm)getattrsetattrr:HandlerContainerrjrkrm)r-targethandlerss r0 _wrap_filterzObserverDecorator._wrap_filter~sR6#8$?H F18 < YY jj   r2c|jJdt|dd}|tt}t |d|t |||j|j <y)NzScene action is not specified__aiogram_action__)rlrrrdictrsrrj)r-rurls r0 _wrap_actionzObserverDecorator._wrap_actions_{{&G(GG&!5t< > &F F0& 9)7)?t{{DII&r2ctj|r2|j|j||S|j ||St d)Nz"Only function or method is allowed)inspect isfunctionrlrwr{ TypeErrorr-rus r0__call__zObserverDecorator.__call__sT   f %{{"!!&)  !!&) @A Ar2c`t|j|jtjSrQ)ActionContainerrjrk SceneActionleaverNs r0rzObserverDecorator.leavestyy$,, 8I8IJJr2cbt|j|jtj|SrQ)rrjrkrenterrs r0rzObserverDecorator.enters!tyy$,, 8I8I6RRr2c`t|j|jtjSrQ)rrjrkrexitrNs r0rzObserverDecorator.exittyy$,, 8H8HIIr2c`t|j|jtjSrQ)rrjrkrbackrNs r0rzObserverDecorator.backrr2)NN) rjr]rkztuple[CallbackType, ...]rlzSceneAction | NonermOptional[After]rarb)ruzType[Scene] | CallbackTyperarb)rurrarb)rurrar)rar)ru Type[Scene]rar) rdrerfr1rwr{rrrrrrGr2r0rhrhqsg &*!%  * #       @KSJJr2rhcDeZdZeZeZeZeZy)rN)rdrerfrrrrrrGr2r0rrs FE FE 6D 6Dr2rc2eZdZ d ddZddZy)rNc<||_||_||_||_yrQ)rjrkrlru)r-rjrkrlrus r0r1zActionContainer.__init__s     r2cK|jtjk(r0|j$|j |jd{y|jtj k(r|j d{y|jtj k(r|j d{y|jtjk(r|jd{yy77t7@7 wrQ)rlrrrugotorrrr-wizards r0executezActionContainer.executes ;;+++ + 0G++dkk* * * [[K-- -,,. [[K,, ,++-   [[K,, ,++-  - +  sHAC; C3 5C;C55C;6C775C;,C9-C;5C;7C;9C;rQ) rjr]rkTuple[CallbackType, ...]rlrru(Optional[Union[Type[Scene], State, str]]rarbr SceneWizardrarb)rdrerfr1rrGr2r0rrsB <@  *   9    r2rc*eZdZ d ddZy)rtNc<||_||_||_||_yrQrp)r-rjrqrkrms r0r1zHandlerContainer.__init__s     r2rQ) rjr]rqrrkrrmrrarb)rdrerfr1rGr2r0rtrts< "&   *     r2rtcpeZdZUded< ded< ded< dZded < dZded < dZded < eZd ed <y) SceneConfigr_r.zList[HandlerContainer]rvz,Dict[SceneAction, Dict[str, CallableObject]]actionsNzOptional[bool]reset_data_on_enterreset_history_on_entercallback_query_without_stater"attrs_resolver) rdrerf__annotations__rrrr#rrGr2r0rrsW $$ 99*..#-1N1&37 .7,)FN&Fr2rc KywrQrG)argskwargss r0_empty_handlerrs cJeZdZ d ddZ ddZd dZd dZy) SceneHandlerWrapperNc@||_t||_||_yrQ)r<rrqrm)r-r<rqrms r0r1zSceneHandlerWrapper.__init__s  %g.  r2c K|d}|d}|d}|jt|jj|||j||}|jj ||fi|d{}|j rYtdd|j j|j j}|j|jd{|S7k7w)Nr.scenes event_update scene_configmanagerr. update_typeeventr6rrmrG) r<r__scene_config__ event_typerqcallrmrrlrr) r-rrr.rrr<resultaction_containers r0rzSceneHandlerWrapper.__call__s #7O &x 0%n5  !ZZ88(33   )t||((@@@ ::. !!      #**5<<8 8 8 A 9s%A0C"2C3A$C"C C" C"c|SrQrGrNs r0 __await__zSceneHandlerWrapper.__await__!s r2cd|jd|jj}|jr|d|jz }|dz }|S)NzSceneHandlerWrapper(z, z, after=))r<rqcallbackrm)r-rs r0__str__zSceneHandlerWrapper.__str__$sM' |2dll6K6K5LM ::  - -F#  r2rQ)r<rrqrrmrrarb)rr rr rar )rar)rar])rdrerfr1rrrrGr2r0rrs] "&     >r2rczeZdZUdZded< d dZd fd Zed dZed d dZ eddZ xZ S)Scenea Represents a scene in a conversation flow. A scene is a specific state in a conversation where certain actions can take place. Each scene has a set of filters that determine when it should be triggered, and a set of handlers that define the actions to be executed when the scene is active. .. note:: This class is not meant to be used directly. Instead, it should be subclassed to define custom scenes. zClassVar[SceneConfig]rc4||_||j_yrQ)rr<rs r0r1zScene.__init__=s   r2c |jdd}|jdd}|jdd}|jdd}|jdd}t|d i|g}tt}|j D]\} t | tst| dd} | s$| | j}| | j}| | j}|Q| j}^|t}||D]\} } t| ddx} r|j| t| t r@|j#t%| j&t(| j*| j,t/| d s| j0j3D]\}}||j5|t7||t ||||| |_y) Nr.rrrrrrormry)r.rvrrrrrrG)rBsuper__init_subclass__rrz __bases__ issubclassrrrrrrrr#extend isinstancerhr:rtrjrrkrmhasattrryitemsupdaterr)clsr state_namerrrrrvrbaseparent_scene_configrjvaluescene_handlersrlaction_handlers __class__s r0rzScene.__init_subclass__DsZZ. $jj)>E!',Dd!K'-zz2PRV'W$$4d; !+F++-GRSWGXMM DDdE*")$0BD"I &"*&9&M&M#%-)<)S)S&+3/B/_/_,%!4!C!C D"  !:N)#. >> router.message.register(MyScene.as_handler(), Command("start")) cPK|jfii|d{y7wrQ)r)rrmiddleware_kwargsrhandler_kwargss r0enter_to_scene_handlerz0Scene.as_handler..enter_to_scene_handlers.&,,sN&M&M;L&MN N Ns &$&)rr r ScenesManagerrr rarbrG)rrrs`` r0 as_handlerzScene.as_handlers5 O! O+8 OOR O  O &%r2rrr rarbrrrarbrQ)rjr_rar)rr rar) rdrerf__doc__rr1r classmethodrrr __classcell__)rs@r0rr,sn ,+!! !8 tTT6   & &r2rceZdZdZ ddZddZdddZddZddZddZ ddZ dd Z dd Z dd Z edd Zedd ZdddZ d ddZddZy) ras A class that represents a wizard for managing scenes in a Telegram bot. Instance of this class is passed to each scene as a parameter. So, you can use it to transition between scenes, get and set data, etc. .. note:: This class is not meant to be used directly. Instead, it should be used as a parameter in the scene constructor. cf||_||_||_||_||_||_d|_y)a A class that represents a wizard for managing scenes in a Telegram bot. :param scene_config: The configuration of the scene. :param manager: The scene manager. :param state: The FSMContext object for storing the state of the scene. :param update_type: The type of the update event. :param event: The TelegramObject represents the event. :param data: Additional data for the scene. N)rrr.rrr6r<)r-rrr.rrr6s r0r1zSceneWizard.__init__s7&)  &  &* r2c:Ktjjd|jj|jj r#|jj id{|jjr,|jjjd{|jj|jjd{|jtjfi|d{y77g727 w)a Enter method is used to transition into a scene in the SceneWizard class. It sets the state, clears data and history if specified, and triggers entering event of the scene. :param kwargs: Additional keyword arguments. :return: None zEntering scene %rN)rr<r=rr.rrCrrr4rOrV _on_actionrrr-rs r0rzSceneWizard.enters  /1B1B1H1HI    0 0**%%b) ) )    3 3,,&&,,. . .jj""4#4#4#:#:;;;dook//:6::: * .;:sIA)D+D,AD-D.6D$D%(D DDDDDc*Ktjjd|jj|r,|j j jd{|jtjfi|d{y7.7w)a" Leaves the current scene. This method is used to exit a scene and transition to the next scene. :param _with_history: Whether to include history in the snapshot. Defaults to True. :param kwargs: Additional keyword arguments. :return: None zLeaving scene %rN) rr<r=rr.rr4rTrrr)r- _with_historyrs r0rzSceneWizard.leavesp  .0A0A0G0GH ,,&&//1 1 1dook//:6::: 2:s$AB B!(B B BBcvKtjjd|jj|j j jd{|jtjfi|d{|j jdddi|d{y7T7-7 w)z Exit the current scene and enter the default scene/state. :param kwargs: Additional keyword arguments. :return: None zExiting scene %rN _check_activeFrQ) rr<r=rr.rr4rOrrrrrs r0rzSceneWizard.exits  .0A0A0G0GHll""((***dook..9&999 dll  EUEfEEE +9Es6AB9B3(B9B5%B9-B7.B95B97B9c^Ktjjd|jj|j dddi|d{|j jjd{}|j j|fddi|d{y7X7.7 w)z This method is used to go back to the previous scene. :param kwargs: Keyword arguments that can be passed to the method. :return: None z$Back to previous scene from scene %srFNrrG) rr<r=rr.rrr4rZr)r-r new_scenes r0rzSceneWizard.backs  BDDUDUD[D[\djj7u7777,,..7799  dll  J%J6JJJ 89Js6A B-B'+B-:B);&B-!B+"B-)B-+B-cK|jjJd|j|jjfi|d{y7w)z This method allows to re-enter the current scene. :param kwargs: Additional keyword arguments to pass to the scene. :return: None NzScene state is not specified)rr.rrs r0retakezSceneWizard.retakesK  &&2R4RR2dii))//:6:::sAA A AcK|jdi|d{|jj|fddi|d{y7,7w)a The `goto` method transitions to a new scene. It first calls the `leave` method to perform any necessary cleanup in the current scene, then calls the `enter` event to enter the specified scene. :param scene: The scene to transition to. Can be either a `Scene` instance `State` instance or a string representing the scene. :param kwargs: Additional keyword arguments to pass to the `enter` method of the scene manager. :return: None NrFrG)rrr)r-r<rs r0rzSceneWizard.goto'sLdjj"6""" dll  FeFvFFF #Fs A A&A AA A cK|js tdtjjd|j|j j |j jj|i}|s@tjjd|j|j j y|j}||vrAtjjd|j||j j y||j|j|jfii|j|d{y7w)NzScene is not initializedzCall action %r in scene %rzAction %r not found in scene %rFz,Action %r for event %r not found in scene %rT) r<rrr=rjrr.rrIrrrr6)r-rlr action_configrs r0rzSceneWizard._on_action6szz !;< < 8&++tGXGXG^G^_))1155fbA  MM  16;;@Q@Q@W@W %% ] * MM  > !!''   ,mJ',,TZZ_G^$))G^W]G^___ `sE EEEcXK|jj|d{y7w)z Sets custom data in the current state. :param data: A mapping containing the custom data to be set in the current state. :return: None r6N)r.rC)r-r6s r0rCzSceneWizard.set_dataOs#jj!!t!,,,s *(*cRK|jjd{S7w)z This method returns the data stored in the current state. :return: A dictionary containing the data stored in the scene state. N)r.r8rNs r0r8zSceneWizard.get_dataXs! ZZ((****s '%'c Kyw)z This method returns the value from key in the data of the current state. :param key: The keyname of the item you want to return the value from. :return: A dictionary containing the data stored in the scene state. NrG)r-r)s r0 get_valuezSceneWizard.get_value`s  rc Kyw)aC This method returns the value from key in the data of the current state. :param key: The keyname of the item you want to return the value from. :param default: Default value to return, if ``key`` was not found. :return: A dictionary containing the data stored in the scene state. NrGr-r)defaults r0rzSceneWizard.get_valueks  rNcVK|jj||d{S7wrQ)r.rr s r0rzSceneWizard.get_valuews#ZZ))#w7777s )')c|K|r|j||jj|d{S7w)z This method updates the data stored in the current state :param data: Optional mapping of data to update. :param kwargs: Additional key-value pairs of data to update. :return: Dictionary of updated data rN)rr.r>)r-r6rs r0r>zSceneWizard.update_datazs5  MM$ ZZ+++8888s 3<:<cBK|jid{y7w)z9 Clears the data. :return: None N)rCrNs r0 clear_datazSceneWizard.clear_datas mmBs ) rrrrr.rrr]rr r6r`rT)rboolrr rarb)r<zUnion[Type[Scene], State, str]rr rarb)rlrrr rar)r6zMapping[str, Any]rarb)rar`)r)r]ra Optional[Any])r)r]r r rar rQ)r)r]r rrar)r6zOptional[Mapping[str, Any]]rr rar`rc)rdrerfrr1rrrrrrrrCr8rrr>rrGr2r0rrs +!++ +  +  ++8;" ; F K; G2-+      837 9/ 9BE 9  9 r2rcbeZdZdZ ddZd dZd dZ d d dZd dZy)rz The ScenesManager class is responsible for managing scenes in an application. It provides methods for entering and exiting scenes, as well as retrieving the active scene. c~||_||_||_||_||_t |j|_yrQ)registryrrr.r6r%r4)r-rrrr.r6s r0r1zScenesManager.__init__s8! &   %djj1 r2c K|jj|}|t|j||j|j |j |jSw)Nrr)rrIrrr.rrr6)r- scene_types r0 _get_scenezScenesManager._get_scenesY]]&&z2 '88jj ,,jjYY   sA&A(cK|jjd{} |j|d{S77#t$rYywxYwwrQ)r.rRrr)r-r.s r0_get_active_scenezScenesManager._get_active_scenesLjj**,, // /-/  s@A?AAAAAA A AAAcK|r>|jd{}|$|jjdi|d{ |j|d{}|jjdi|d{y7i7E7-7 #t $r*||j jdd{7YywxYww)a Enters the specified scene. :param scene_type: Optional Type[Scene], State or str representing the scene type to enter. :param _check_active: Optional bool indicating whether to check if there is an active scene to exit before entering the new scene. Defaults to True. :param kwargs: Additional keyword arguments to pass to the scene's wizard.enter() method. :return: None NrG)rrrrrrr.rV)r-rrr active_scener<s r0rzScenesManager.enters !%!7!7!99L'.l))..8888 ///*55E %%,,$$.v. . .:86 /  -%**&&t, , , -ssCB%CBCB BB C<B=CCB C *B=4B75B=:C<B==CcK|jd{}|sy|jjdi|d{y7,7w)z Close method is used to exit the currently active scene in the ScenesManager. :param kwargs: Additional keyword arguments passed to the scene's exit method. :return: None NrG)rrr)r-rr<s r0closezScenesManager.closesG,,.. ell)&)))/ *sAA&AAAAN) r SceneRegistryrr]rr r.rr6r`rarb)rrrar)razOptional[Scene]r)rrrrrr rarbr) rdrerfrr1rrrrrGr2r0rrs 222 2  2  2 2  #/<// /  /: *r2rcpeZdZdZd d dZd dZ ddZ ddZddddZdd Z dd Z y)rzJ A class that represents a registry for scenes in a Telegram bot. cP||_||_i|_|j|y)z Initialize a new instance of the SceneRegistry class. :param router: The router instance used for scene registration. :param register_on_add: Whether to register the scenes to the router when they are added. N)rregister_on_add_scenes_setup_middleware)r-rr!s r0r1zSceneRegistry.__init__s( .9;  v&r2ct|tr&|jj|jy|j j D],}|jdvr|j|j.y)N>errorr) rrrouter_middleware_update_middlewarervalues event_name _middleware)r-robservers r0r#zSceneRegistry._setup_middlewaresm fj ) MM * *4+B+B C ((//1 8H""&99  % %d&6&6 7 8r2cKt|tsJdt||j|j|d||d<|||d{S7w)Nz Event must be an Update instancer.rrrr.r6r)rr!rrr)r-rqrr6s r0r'z SceneRegistry._update_middlewares^ %(L*LL(&((++w-  XUD))))sAAAAczK|d}t||j||d||d<|||d{S7w)Nrr.r-r)rr)r-rqrr6rs r0r*zSceneRegistry._middlewaresN n-&))w-  XUD))))s 2;9;N)rc|s td|D]}|jj|jvr#t d|jjd||j|jj<|r |j |j |js|jj |j y)a This method adds the specified scenes to the registry and optionally registers it to the router. If a scene with the same state already exists in the registry, a SceneException is raised. .. warning:: If the router is not specified, the scenes will not be registered to the router. You will need to include the scenes manually to the router or use the register method. :param scenes: A variable length parameter that accepts one or more types of scenes. These scenes are instances of the Scene class. :param router: An optional parameter that specifies the router to which the scenes should be added. :return: None z$At least one scene must be specifiedzScene with state z already existsN) ValueErrorrr.r"rinclude_routerrr!r)r-rrr<s r0rzSceneRegistry.add$s$CD D >E%%++t||;$'(>(>(D(D'GW:?DLL//55 6%%eoo&78%% **5??+<= >r2c<|j|d|jiy)z Registers one or more scenes to the SceneRegistry. :param scenes: One or more scene classes to register. :return: None rN)rr)r-rs r0rzSceneRegistry.registerFs &--r2cBtj|r&t|tr|jj }t |tr |j }|t |ts td |j|S#t$rtd|dwxYw)ae This method returns the registered Scene object for the specified scene. The scene parameter can be either a Scene object, State object or a string representing the name of the scene. If a Scene object is provided, the state attribute of the SceneConfig object associated with the Scene object will be used as the scene name. If a State object is provided, the state attribute of the State object will be used as the scene name. If None or an invalid type is provided, a SceneException will be raised. If the specified scene is not registered in the SceneRegistry object, a SceneException will be raised. :param scene: A Scene object, State object or a string representing the name of the scene. :return: The registered Scene object corresponding to the given scene parameter. z4Scene must be a subclass of Scene, State or a stringzScene z is not registered) r}isclassrrrr.rrr]rr"KeyError)r-r<s r0rIzSceneRegistry.getOs ??5 !j&>**00E eU #KKE  Zs%; !WX X G<<& & G 6%2D!EF F Gs 6BBr)rrr!rrarbr)rqz"NextMiddlewareType[TelegramObject]rr r6r`rar )rrrzOptional[Router]rarb)rrrarb)r<rrar) rdrerfrr1r#r'r*rrrIrGr2r0rrs ' 8*3** * *"*3** * * DH >D.Gr2rc\eZdZUded<dZded<ed dZed dZed dZy) AfterrrlNrr<c0|tjSNrl)rrrs r0rz After.exitq+**++r2c0|tjSr9)rrr;s r0rz After.backur<r2c2|tj|S)N)rlr<)rr)rr<s r0rz After.gotoys+++599r2)rar7)r<rrar7) rdrerfrr<rrrrrGr2r0r7r7lsM 6:E 3:,,,,::r2r7cLeZdZd dZdd d dZd dZd dZd dZd dZy) ObserverMarkerc||_yrQr)r-rjs r0r1zObserverMarker.__init__s  r2Nrc2t|j||S)Nr)rhrj)r-rmrks r0rzObserverMarker.__call__s ! II   r2cNt|j|tjSr9)rhrjrr)r-rks r0rzObserverMarker.enters GK?!.1L)*@A#$45N#$45N'(<= & !D /K#$45N /K&':;r2rJ)rr rr rarb)E __future__rr} collectionsr dataclassesrrenumrrtypingr r r r r rrrrrtyping_extensionsraiogramraiogram.dispatcher.dispatcherraiogram.dispatcher.event.basesr aiogram.dispatcher.event.handlerrraiogram.dispatcher.flagsraiogram.dispatcher.routerraiogram.exceptionsraiogram.filtersraiogram.fsm.contextraiogram.fsm.stateraiogram.fsm.storage.memoryr aiogram.typesr r!"aiogram.utils.class_attrs_resolverr"r#r%rhrrrtrrrrrrrr7r@rJonrGr2r0rls%"#*   #4=I>,-'*#:0 F$F$R9J9Jx$  0    0 22jL&L&^S S lQ*Q*hDGDGN :: :"II8)<)