jy|ddlmZddlZddlmZddlmZddlmZm Z ddl m Z m Z ddl mZmZmZmZddlmZdd lmZdd lmZdd lmZmZdd lmZdd lmZddlm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*m+Z+ddl,m-Z-m.Z.erddl/m0Z0GddZ1GddZ2Gdde Z3GddZ4edGd d!Z5eGd"d#Z6d5d$Z7Gd%d&Z8Gd'd(Z9Gd)d*Z:Gd+d,Z;Gd-d.ZGd3d4Z?e?Z@y)6) annotationsN) defaultdict)Mapping) dataclassreplace)Enumauto) TYPE_CHECKINGAnyClassVaroverload)Self)loggers) Dispatcher)CallableObject CallbackType)extract_flags_from_object)Router)SceneException) StateFilter) FSMContext)State)MemoryStorageRecord)TelegramObjectUpdate)ClassAttrsResolverget_sorted_mro_attrs_resolver)NextMiddlewareTypecVeZdZd 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__!s2  (MM 73 c~K|jjd{}|jdg}|j||dt ||j kDr||j d}t jjd|||jj|d{y77w)Nhistoryr)dataz Push state=%s data=%s to historyr/) r'get_data setdefaultappendlenr%rscenedebug update_data)r(r)r1 history_datar/s r+pushzHistoryManager.push)s!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)Nr/r)r1r2z!Pop state=%s data=%s from historyr0) r'r3r4popset_datar9rr7r8r)r(r:r/recordr)r1s r+r=zHistoryManager.pop3s!0099;; )))R8wf~%%..r2 2 2%%11'1B B B ?M"T::< 3 Bs4CC AC4C5$CC3CCCcK|jjd{}|jdg}|sytdi|dS7'w)Nr/r'r3r4r)r(r:r/s r+getzHistoryManager.getBsL!0099;; )))R8"1WR[11 r(s r+clearzHistoryManager.clearNs2 O,!!**2...s>AAAcK|jjd{}|jjd{}|j||d{y7A7!7 wN)r& get_stater3r;r(r)r1s r+snapshotzHistoryManager.snapshotRsQkk++--[[))++iit$$$.+$s3A(A"!A(A$A(A&A($A(&A(cK|jj|d{|jj|d{y7(7wrL)r& set_stater>rNs r+ _set_statezHistoryManager._set_stateWs?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)r=rRrr7r8r)r1)r(previous_states r+rollbackzHistoryManager.rollback[s#xxz)//$+ + +  *     oon22N4G4GHHH###* + Is2B-B'B-B)A#B-B+B-)B-+B-N)scenes_history )r)rr"strr*int)r) str | Noner1dict[str, Any]returnNone)r\zMemoryStorageRecord | None)r\zlist[MemoryStorageRecord]r\r])r\rZ) __name__ __module__ __qualname__r,r;r=rDrGrJrOrRrUrBr-r+r r s0 ? ;2A /% ) $r-r cdeZdZ d d dZd dZd dZddZddZddZddZ dd Z y)ObserverDecoratorNc<||_||_||_||_yrL)namefiltersactionafter)r(rerfrgrhs r+r,zObserverDecorator.__init__ks     r-ct|dd}|s g}||_|jt|j||j |j y)N__aiogram_handler__)rehandlerrfrh)getattrrjr5HandlerContainerrerfrh)r(targethandlerss r+ _wrap_filterzObserverDecorator._wrap_filterwsM6#8$?H)1F & YY jj   r-c|jJdt|dd}|tt}||_t |||j|j <y)NzScene action is not specified__aiogram_action__)rgrlrdictrrrre)r(rnrgs r+ _wrap_actionzObserverDecorator._wrap_actionsZ{{&G(GG&!5t< > &F(.F %)7)?t{{DII&r-ctj|r2|j|j||S|j ||Sd}t |)Nz"Only function or method is allowed)inspect isfunctionrgrprt TypeError)r(rnmsgs r+__call__zObserverDecorator.__call__sW   f %{{"!!&)  !!&) 7CC. r-c`t|j|jtjSrL)ActionContainerrerf SceneActionleaverIs r+r~zObserverDecorator.leavestyy$,, 8I8IJJr-cbt|j|jtj|SrL)r|rerfr}enter)r(rns r+rzObserverDecorator.enters!tyy$,, 8I8I6RRr-c`t|j|jtjSrL)r|rerfr}exitrIs r+rzObserverDecorator.exittyy$,, 8H8HIIr-c`t|j|jtjSrL)r|rerfr}backrIs r+rzObserverDecorator.backrr-)NN) rerXrftuple[CallbackType, ...]rgzSceneAction | Nonerh After | Noner\r])rnztype[Scene] | CallbackTyper\r])rnrr\r])rnrr\r)r\r|)rn type[Scene]r\r|) r_r`rar,rprtrzr~rrrrBr-r+rcrcjsg &*"  * #       @ KSJJr-rccDeZdZeZeZeZeZy)r}N)r_r`rar rr~rrrBr-r+r}r}s FE FE 6D 6Dr-r}c2eZdZ d ddZddZy)r|Nc<||_||_||_||_yrL)rerfrgrn)r(rerfrgrns r+r,zActionContainer.__init__s     r-cK|jtjk(r0|j$|j |jd{y|jtj k(r|j d{y|jtj k(r|j d{y|jtjk(r|jd{yy77t7@7 wrL)rgr}rrngotor~rrr(wizards r+executezActionContainer.executes ;;+++ + 0G++dkk* * * [[K-- -,,. [[K,, ,++-   [[K,, ,++-  - +  sHAC; C3 5C;C55C;6C775C;,C9-C;5C;7C;9C;rL) rerXrfrrgr}rn type[Scene] | State | str | Noner\r]r SceneWizardr\r])r_r`rar,rrBr-r+r|r|sA 48  *   1    r-r|T)slotsc:eZdZUded<ded<ded<dZded <y) rmrXrerrkrrfNrrh)r_r`ra__annotations__rhrBr-r+rmrms I  %%E<r-rmcpeZdZUded< ded< ded< dZded < dZded < dZded < eZd ed <y) SceneConfigrZr)zlist[HandlerContainer]roz,dict[SceneAction, dict[str, CallableObject]]actionsNz bool | Nonereset_data_on_enterreset_history_on_entercallback_query_without_staterattrs_resolver) r_r`rarrrrrrrBr-r+rrsW $$ 99'++#*.K.&04 +4,)FN&Fr-rc KywrLrB)argskwargss r+_empty_handlerrs cJeZdZ d ddZ ddZd dZd dZy) SceneHandlerWrapperNc@||_t||_||_yrL)r7rrkrh)r(r7rkrhs r+r,zSceneHandlerWrapper.__init__s  %g.  r-c XK |d}|d}|d}i|j||_|j t |jj |||j||} |jj| |fi|d{} |jrYtd d |jj|jj} | j| jd{| S#t$r&}|jd}d|d}t|dd}~wwxYw77:w) Nr)scenesrzScene context key z@ is not available. Ensure FSM is enabled and pipeline is intact. event_update scene_configmanagerr) update_typeeventr1rrhrB)KeyErrorrrr1r7r__scene_config__ event_typerkcallrhr|rgrr) r(rrr)rerror missing_keyryrr7resultaction_containers r+rzzSceneHandlerWrapper.__call__sA 0 &wE$*8$4F &n5 ///  !ZZ88(33   )t||((@@@ ::. !!      #**5<<8 8 8 = 0**Q-K$[O4@@ !%4 /  0(A 9sGD* C4A:D*D& A$D*-D(.D*4 D#=!DD##D*(D*c|SrLrBrIs r+ __await__zSceneHandlerWrapper.__await__s r-cd|jd|jj}|jr|d|jz }|dz }|S)NzSceneHandlerWrapper(z, z, after=))r7rkcallbackrh)r(rs r+__str__zSceneHandlerWrapper.__str__!sM' |2dll6K6K5LM ::  - -F#  r-rL)r7rrkrrhrr\r])rrrr r\r )r\r)r\rX)r_r`rar,rzrrrBr-r+rrs^ #    &&& &Pr-rczeZdZUdZded< 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_yrL)rr7rs r+r,zScene.__init__:s   r-c |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)rrrrrrjrhrr)r)rorrrrrrB)r=super__init_subclass__rrs __bases__ issubclassrrlrrrrrextend isinstancercr5rmrerrfrhhasattrrritemsupdaterr)clsr state_namerrrrrorbaseparent_scene_config_namevaluescene_handlersrgaction_handlers __class__s r+rzScene.__init_subclass__AsZZ. $jj)>E!',Dd!K'-zz2PRV'W$$4d; !+F++-GRSWGXMM DDdE*")$0BD"I &"*&9&M&M#%-)<)S)S&+3/B/_/_,%!4!C!C D"  !:N*3/ >> router.message.register(MyScene.as_handler(), Command("start")) cPK|jfii|d{y7wrL)r)rrmiddleware_kwargsrhandler_kwargss r+enter_to_scene_handlerz0Scene.as_handler..enter_to_scene_handlers. &,,sN&M&M;L&MN N Ns &$&)rrr ScenesManagerrr r\r]rB)rrrs`` r+ as_handlerzScene.as_handlers9 O! O! O"% O  O&%r-rrr r\r]rrr\r]rL)rerZr\r)rr r\r) r_r`ra__doc__rr,r classmethodrrr __classcell__)rs@r+rr)sn ,+!! !8 tTT6  &&r-rceZdZdZ 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)rrr1r7)r(rrr)rrr1s r+r,zSceneWizard.__init__s7&)  &  #' r-c: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)rr7r8rr)rr>rrr/rJrQ _on_actionr}rr(rs r+rzSceneWizard.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) rr7r8rr)rr/rOrr}r~)r( _with_historyrs r+r~zSceneWizard.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_activeFrL) rr7r8rr)rr/rJrr}rrrs r+rzSceneWizard.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 %srFNrrB) rr7r8rr)r~rr/rUr)r(r new_scenes r+rzSceneWizard.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 r+retakezSceneWizard.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 NrFrB)r~rr)r(r7rs r+rzSceneWizard.goto&sLdjj"6""" dll  FeFvFFF #Fs A A&A AA A cK|js d}t|tjjd|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) r7rrr8rerr)rrDrrrr1)r(rgrry action_configrs r+rzSceneWizard._on_action5szz,C % % 8&++tGXGXG^G^_))1155fbA  MM  1 !!''  %% ] * MM  > !!''   ,mJ',,TZZ_G^$))G^W]G^___ `sEEEEcXK|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 r1N)r)r>)r(r1s r+r>zSceneWizard.set_dataQs#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)r3rIs r+r3zSceneWizard.get_dataZs! 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. NrB)r(r$s r+ get_valuezSceneWizard.get_valuebrc 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. NrBr(r$defaults r+rzSceneWizard.get_valuelrrNcVK|jj||d{S7wrL)r)rrs r+rzSceneWizard.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)r9)r(r1rs r+r9zSceneWizard.update_datazs5  MM$ ZZ+++8888s 3<:<cBK|jid{y7w)z9 Clears the data. :return: None N)r>rIs r+ clear_datazSceneWizard.clear_datas mmBs ) rrrrr)rrrXrrr1r[rT)rboolrr r\r])r7ztype[Scene] | State | strrr r\r])rgr}rr r\r)r1zMapping[str, Any]r\r])r\r[)r$rXr\ Any | None)r$rXrr r\r rL)r$rXrrr\r)r1zMapping[str, Any] | Nonerr r\r[r^)r_r`rarr,rr~rrrrrr>r3r rr9r rBr-r+rrs (!(( (  (  ((8;" ; F K; G8-+    8 *.9&99  9  r-rcbeZdZdZ 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|_yrL)registryrrr)r1r r/)r(rrrr)r1s r+r,zScenesManager.__init__s8! &   %djj1 r-c K|jj|}|t|j||j|j |j |jSw)Nrr)rrDrrr)rrr1)r( scene_types r+ _get_scenezScenesManager._get_scenesY]]&&z2 '88jj ,,jjYY   sA&A(cK|jjd{} |j|d{S77#t$rYywxYwwrL)r)rMrr)r(r)s r+_get_active_scenezScenesManager._get_active_scenesLjj**,, // /-/  s@A?AAAAAA A AAAcK|ri|j||_|r>|jd{}|$|jjdi|d{ |j |d{}|jj di|d{y7i7E7-7 #t $r*||jjdd{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 NrB) r1rrrrrrr)rQ)r(rrr active_scener7s r+rzScenesManager.enters /499//DI !%!7!7!99L'.l))..8888 ///*55E %%,,$$.v. . .:86 /  -%**&&t, , , -su-CB%CBCB!/B0B!4CBCCB!C!*C C CCCCcK|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 NrB)rrr)r(rr7s r+closezScenesManager.closesG,,.. ell)&)))/ *sAA&AAAAN) r SceneRegistryrrXrrr)rr1r[r\r])rrr\r)r\z Scene | Noner )rrrrrr r\r]r) r_r`rarr,rrrrrBr-r+rrs 222 2  2  2 2  #/4// /  /@ *r-rcpeZdZdZd 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(rrs r+r,zSceneRegistry.__init__s( .68  v&r-ct|tr&|jj|jy|j j D],}|jdvr|j|j.y)N>rr) rrrouter_middleware_update_middlewarervalues event_name _middleware)r(robservers r+r!zSceneRegistry._setup_middlewaresm fj ) MM * *4+B+B C ((//1 8H""&99  % %d&6&6 7 8r-cKt|tsJd|jd}||||d{St||j|j |||d<|||d{S7<7w)Nz Event must be an Update instancer)rrrr)r1r)rrrDrrr)r(rkrr1r)s r+r$z SceneRegistry._update_middlewares %(L*LL(! = -- -&((++  XUD))).*s!7A:A67A:1A82A:8A:cK|jd}||||d{S|d}t||j||||d<|||d{S777w)Nr)rr*r)rDrr)r(rkrr1r)rs r+r'zSceneRegistry._middlewaresu ! = -- -n-&))  XUD))).*s! AA2AAAAN)rc|s d}t||D]}|jj|jvr%d|jjd}t |||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_routerrrr)r(rrryr7s r+rzSceneRegistry.add0s$8CS/ ! >E%%++t||;)%*@*@*F*F)IY$S))9>DLL//55 6%%eoo&78%% **5??+<= >r-c<|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 r+rzSceneRegistry.registerRs &--r-cLtj|r&t|tr|jj }t |tr |j }|t |ts d}t| |j|S#t$rd|d}t|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. Nz4Scene must be a subclass of Scene, State or a stringzScene z is not registered) rvisclassrrrr)rrrXrr r)r(r7rys r+rDzSceneRegistry.get[s ??5 !j&>**00E eU #KKE  Zs%;HC % % 0<<& & 05)#56C %4 / 0s 8BB#r )rrrrr\r]r)rkz"NextMiddlewareType[TelegramObject]rrr1r[r\r )rrrz Router | Noner\r])rrr\r])r7rr\r) r_r`rarr,r!r$r'rrrDrBr-r+rrs~ ' 8*3** * *(*3** * *(AE >D.0r-rc\eZdZUded<dZded<ed dZed dZed dZy) Afterr}rgNrr7c0|tjSNrg)r}rrs r+rz After.exit+**++r-c0|tjSr5)r}rr7s r+rz After.backr8r-c2|tj|S)N)rgr7)r}r)rr7s r+rz After.gotos+++599r-)r\r3)r7rr\r3) r_r`rarr7rrrrrBr-r+r3r3zsM .2E +2,,,,::r-r3cLeZdZd dZdd d dZd dZd dZd dZd dZy) ObserverMarkerc||_yrLr)r(res r+r,zObserverMarker.__init__s  r-Nrc2t|j||S)Nr)rcre)r(rhrfs r+rzzObserverMarker.__call__s ! II   r-cNt|j|tjSr5)rcrer}r)r(rfs r+rzObserverMarker.enters GK?!.1L)*@A#$45N#$45N'(<= & !D /K#$45N /K&':;r-rF)rr rr r\r])A __future__rrv collectionsrcollections.abcr dataclassesrrenumrr typingr r r r typing_extensionsraiogramraiogram.dispatcher.dispatcherr aiogram.dispatcher.event.handlerrraiogram.dispatcher.flagsraiogram.dispatcher.routerraiogram.exceptionsraiogram.filtersraiogram.fsm.contextraiogram.fsm.stateraiogram.fsm.storage.memoryr aiogram.typesrr"aiogram.utils.class_attrs_resolverrraiogram.dispatcher.event.basesrr rcr}r|rmrrrrrrrr3r<rFonrBr-r+ris2"##*99"4I>,-'*#:0 AG$G$T:J:Jz$  0   0 ;;|N&N&bV V rT*T*nM0M0` :: :"II8)<)