o 4jy@sddlmZddlZddlmZddlmZddlmZm 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/m0Z0GdddZ1GdddZ2Gddde Z3GdddZ4eddGd d!d!Z5eGd"d#d#Z6d;d)d*Z7Gd+d,d,Z8Gd-d.d.Z9Gd/d0d0Z:Gd1d2d2Z;Gd3d4d4ZGd9d:d:Z?e?Z@dS)<) annotationsN) defaultdict)Mapping) dataclassreplace)Enumauto) TYPE_CHECKINGAnyClassVaroverload)Self)loggers) Dispatcher)CallableObject CallbackType)extract_flags_from_object)Router)SceneException) StateFilter) FSMContext)StateMemoryStorageRecord)TelegramObjectUpdate)ClassAttrsResolverget_sorted_mro_attrs_resolver)NextMiddlewareTypec@sheZdZd#d$d d Zd%ddZd&ddZd&ddZd'ddZd(ddZd(ddZ d%ddZ d)d d!Z d"S)*HistoryManagerscenes_history staterdestinystrsizeintcCs*||_||_t|jt|j|dd|_dS)N)r#)storagekey)_size_staterr'rr(_history_state)selfr"r#r%r-N/var/www/kodo/Anonymous/send/lib/python3.10/site-packages/aiogram/fsm/scene.py__init__!s   zHistoryManager.__init__ str | Nonedatadict[str, Any]returnNonecst|jIdH}|dg}|||dt||jkr&||j d}tjd|||jj |dIdHdS)Nhistoryr"r1z Push state=%s data=%s to historyr5) r+get_data setdefaultappendlenr)rscenedebug update_data)r,r"r1 history_datar5r-r-r.push)s zHistoryManager.pushMemoryStorageRecord | Nonecs|jIdH}|dg}|sdS|}|d}|d}|s+|jiIdHn |jj|dIdHtjd||t ||dS)Nr5r"r1r7z!Pop state=%s data=%s from historyr6) r+r8r9popset_datar>rr<r=r)r,r?r5recordr"r1r-r-r.rB3s  zHistoryManager.popcs8|jIdH}|dg}|sdStdi|dS)Nr5r-)r+r8r9rr,r?r5r-r-r.getBs  zHistoryManager.getlist[MemoryStorageRecord]cs,|jIdH}|dg}dd|DS)Nr5cSsg|] }tdi|qS)r-r).0itemr-r-r. Lsz&HistoryManager.all..)r+r8r9rFr-r-r.allIs zHistoryManager.allcs$tjd|jiIdHdS)Nz Clear history)rr<r=r+rCr,r-r-r.clearNs zHistoryManager.clearcs8|jIdH}|jIdH}|||IdHdSN)r* get_stater8r@r,r"r1r-r-r.snapshotRszHistoryManager.snapshotcs*|j|IdH|j|IdHdSrO)r* set_staterCrQr-r-r. _set_stateWszHistoryManager._set_statecsZ|IdH}|s|diIdHdStjd|j|j||j|jIdH|jS)NzRollback to state=%s data=%s)rBrTrr<r=r"r1)r,previous_stater-r-r.rollback[szHistoryManager.rollbackN)r r!)r"rr#r$r%r&)r"r0r1r2r3r4)r3rA)r3rHr3r4)r3r0) __name__ __module__ __qualname__r/r@rBrGrLrNrRrTrVr-r-r-r.r s       rc@sbeZdZ  d!d"d d Zd#ddZd$ddZd%ddZd&ddZd'ddZd&ddZ d&dd Z dS)(ObserverDecoratorNnamer$filterstuple[CallbackType, ...]actionSceneAction | Noneafter After | Noner3r4cC||_||_||_||_dSrO)r\r]r_ra)r,r\r]r_rar-r-r.r/k zObserverDecorator.__init__targettype[Scene] | CallbackTypecCs:t|dd}|s g}||_|t|j||j|jddS)N__aiogram_handler__)r\handlerr]ra)getattrrgr:HandlerContainerr\r]ra)r,rehandlersr-r-r. _wrap_filterws zObserverDecorator._wrap_filterrcCsL|jdus Jdt|dd}|durtt}||_t|||j|j<dS)NzScene action is not specified__aiogram_action__)r_rirdictrmrr\)r,rer_r-r-r. _wrap_actions  zObserverDecorator._wrap_actioncCs<t|r|jdur|||S|||Sd}t|)Nz"Only function or method is allowed)inspect isfunctionr_rlro TypeError)r,remsgr-r-r.__call__s    zObserverDecorator.__call__ActionContainercCt|j|jtjSrO)rur\r] SceneActionleaverMr-r-r.rxzObserverDecorator.leave type[Scene]cCst|j|jtj|SrO)rur\r]rwenter)r,rer-r-r.r{szObserverDecorator.entercCrvrO)rur\r]rwexitrMr-r-r.r|ryzObserverDecorator.exitcCrvrO)rur\r]rwbackrMr-r-r.r}ryzObserverDecorator.back)NN) r\r$r]r^r_r`rarbr3r4)rerfr3r4)rerr3r4)rerr3r)r3ru)rerzr3ru) rXrYrZr/rlrortrxr{r|r}r-r-r-r.r[js   r[c@s$eZdZeZeZeZeZdS)rwN)rXrYrZrr{rxr|r}r-r-r-r.rws  rwc@s$eZdZ ddd d ZdddZdS)ruNr\r$r]r^r_rwre type[Scene] | State | str | Noner3r4cCrcrO)r\r]r_re)r,r\r]r_rer-r-r.r/rdzActionContainer.__init__wizard SceneWizardcs|jtjkr|jdur||jIdHdS|jtjkr&|IdHdS|jtjkr5|IdHdS|jtjkrD|IdHdSdSrO)r_rwr{regotorxr|r}r,rr-r-r.executes   zActionContainer.executerO) r\r$r]r^r_rwrer~r3r4rrr3r4)rXrYrZr/rr-r-r-r.rus  ruT)slotsc@s2eZdZUded<ded<ded<dZded <dS) rjr$r\rrhr^r]Nrbra)rXrYrZ__annotations__rar-r-r-r.rjs rjc@sbeZdZUded< ded< ded< dZded < dZded < dZded < eZd ed <dS) SceneConfigr0r"zlist[HandlerContainer]rkz,dict[SceneAction, dict[str, CallableObject]]actionsNz bool | Nonereset_data_on_enterreset_history_on_entercallback_query_without_staterattrs_resolver) rXrYrZrrrrrrr-r-r-r.rs     rargsr kwargsr3r4csdSrOr-)rrr-r-r._empty_handlersrc@s8eZdZ ddd d ZdddZdddZdddZdS)SceneHandlerWrapperNr<rzrhrrarbr3r4cCs||_t||_||_dSrO)r<rrhra)r,r<rhrar-r-r.r/s  zSceneHandlerWrapper.__init__eventrrr c sz |d}|d}Wnty'}z|jd}d|d}t|dd}~ww|d}i|j||_|jt|jj|||j||dd} |jj | |fi|IdH} |j rjt d d |j j |j j} | | jIdH| S) Nr"scenesrzScene context key z@ is not available. Ensure FSM is enabled and pipeline is intact. event_update scene_configmanagerr" update_typerr1rrar-)KeyErrorrrr1r<r__scene_config__ event_typerhcallrarur_rr) r,rrr"rerror missing_keyrsrr<resultaction_containerr-r-r.rtsB     zSceneHandlerWrapper.__call__r cCs|SrOr-rMr-r-r. __await__szSceneHandlerWrapper.__await__r$cCs8d|jd|jj}|jr|d|j7}|d7}|S)NzSceneHandlerWrapper(z, z, after=))r<rhcallbackra)r,rr-r-r.__str__!s zSceneHandlerWrapper.__str__rO)r<rzrhrrarbr3r4)rrrr r3r )r3r )r3r$)rXrYrZr/rtrrr-r-r-r.rs  (rcsdeZdZUdZded< ddd Zdfd d ZedddZeddddZ ed ddZ Z 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]rrrr3r4cCs||_||j_dSrO)rr<rr-r-r.r/:s zScene.__init__rr c sf|dd}|dd}|dd}|dd}|dd}tjd i|g}tt}|jD]-} t| ts8q0t| dd} | sAq0|durH| j }|durO| j }|durV| j }|dur]| j }q0|durdt }||D]:\} } t| dd} ry|| t| tr|t| jt| j| jdt| d r| jD] \}}|||qqht||t|||||d |_dS) Nr"rrrrrrgrarm)r"rkrrrrrr-)rBsuper__init_subclass__rrn __bases__ issubclassrrirrrrrextend isinstancer[r:rjr\rr]rahasattrrmitemsupdaterr)clsr state_namerrrrrkrbaseparent_scene_config_namevaluescene_handlersr_action_handlers __class__r-r.rAsd            zScene.__init_subclass__routerrcCs|j}t}|jD]#}|j|jjt||j|jdg|j Rdt |ji| |jq |D]}|j r9|dkr9q/|j| t|jq/dS)z^ Adds the scene to the given router. :param router: :return: rflagscallback_queryN)rsetrk observersr\registerrrhrar]raddrfilterrr")rrrused_observersrh observer_namer-r-r. add_to_router{s(   zScene.add_to_routerNr\r0cCs>|durd|jd|jd|jj}t|d}|||S)zM Returns the scene as a router. :return: new router NzScene '.z ' for state r\)rYrZrr"rr)rr\rr-r-r. as_routers  zScene.as_routerhandler_kwargsrc sd fd d }|S) z Create an entry point handler for the scene, can be used to simplify the handler that starts the scene. >>> router.message.register(MyScene.as_handler(), Command("start")) rrr ScenesManagermiddleware_kwargsr r3r4cs&|jfii|IdHdSrO)r{)rrrrrr-r.enter_to_scene_handlers$z0Scene.as_handler..enter_to_scene_handlerN)rrrrrr r3r4r-)rrrr-rr. as_handlers zScene.as_handlerrrr r3r4rrr3r4rO)r\r0r3r)rr r3r) rXrYrZ__doc__rr/r classmethodrrr __classcell__r-r-rr.r)s   : rc@seZdZdZd;ddZdddZd)r,r1rr-r-r.r>zs zSceneWizard.update_datacs|iIdHdS)z9 Clears the data. :return: None N)rCrMr-r-r. clear_dataszSceneWizard.clear_data) rrrrr"rrr$rrr1r2rT)rrrr r3r4)r<rrr r3r4)r_rwrr r3r)r1rr3r4)r3r2)r(r$r3r)r(r$rr r3r rO)r(r$rrr3r)r1rrr r3r2rW)rXrYrZrr/r{rxr|r}rrrrCr8r rr>rr-r-r-r.rs(          rc@sFeZdZdZd"ddZd#ddZd$ddZ d%d&ddZd'dd Zd!S)(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. registry SceneRegistryrr$rrr"rr1r2r3r4cCs.||_||_||_||_||_t|j|_dSrO)rrrr"r1rr5)r,rrrr"r1r-r-r.r/s zScenesManager.__init__ scene_typer~rc s2|j|}|t|j||j|j|j|jddS)Nrr)rrGrrr"rrr1)r,rr-r-r. _get_scenes zScenesManager._get_scene Scene | Nonecs:|jIdH}z ||IdHWStyYdSwrO)r"rPrr)r,r"r-r-r._get_active_scenes zScenesManager._get_active_sceneTrrrr cs|r i|j||_|r$|IdH}|dur$|jjdi|IdHz ||IdH}WntyF|dur:|jdIdHYdSw|jjdi|IdHdS)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 Nr-) r1rrr|rrr"rSr{)r,rrr active_scener<r-r-r.r{s zScenesManager.entercs4|IdH}|s dS|jjdi|IdHdS)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 Nr-)rrr|)r,rr<r-r-r.closes zScenesManager.closeN) rrrr$rrr"rr1r2r3r4)rr~r3r)r3rr)rr~rrrr r3r4r) rXrYrZrr/rrr{rr-r-r-r.rs     rc@s^eZdZdZd%d&d d Zd'd d Zd(ddZd(ddZddd)ddZd*dd Z d+d#d$Z dS),rzJ A class that represents a registry for scenes in a Telegram bot. Trrregister_on_addrr3r4cCs ||_||_i|_||dS)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)rr_scenes_setup_middleware)r,rrr-r-r.r/szSceneRegistry.__init__cCsHt|tr|j|jdS|jD]}|jdvrq||jqdS)N>rr) rrrouter_middleware_update_middlewarervalues event_name _middleware)r,robserverr-r-r.rs  zSceneRegistry._setup_middlewarerh"NextMiddlewareType[TelegramObject]rrr1r2r cs`t|ts Jd|d}|dur|||IdHSt||j|j||d|d<|||IdHS)Nz Event must be an Update instancer"rrrr"r1r)rrrGrrr)r,rhrr1r"r-r-r.rs  z SceneRegistry._update_middlewarecsT|d}|dur|||IdHS|d}t||j|||d|d<|||IdHS)Nr"rrr)rGrr)r,rhrr1r"rr-r-r.rs  zSceneRegistry._middlewareN)rrrz Router | NonecGs||sd}t||D]1}|jj|jvrd|jjd}t|||j|jj<|r0||q |jr;|j|q dS)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"rrinclude_routerrrr)r,rrrsr<r-r-r.r0szSceneRegistry.addcGs|j|d|jidS)z Registers one or more scenes to the SceneRegistry. :param scenes: One or more scene classes to register. :return: None rN)rr)r,rr-r-r.rRszSceneRegistry.registerr<r~cCs|t|rt|tr|jj}t|tr|j}|dur%t|ts%d}t |z|j |WSt y=d|d}t |dw)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) rpisclassrrrr"rrr$rrr)r,r<rsr-r-r.rG[s     zSceneRegistry.getr)rrrrr3r4r)rhrrrr1r2r3r )rrzrrr3r4)rrzr3r4)r<r~r3rz) rXrYrZrr/rrrrrrGr-r-r-r.rs   " rc@sLeZdZUded<dZded<ed ddZed d d Zedd d ZdS)Afterrwr_Nr~r<r3cC |tjdSNr_)rwr|rr-r-r.r| z After.exitcCrr)rwr}rr-r-r.r}rz After.backcCs|tj|dS)N)r_r<)rwr{)rr<r-r-r.rsz After.goto)r3r)r<r~r3r) rXrYrZrr<rr|r}rr-r-r-r.rzs    rc@sNeZdZdddZdddddZdddZdddZdddZdddZdS)ObserverMarkerr\r$r3r4cCs ||_dSrOr)r,r\r-r-r.r/s zObserverMarker.__init__Nrr]rrarbr[cGst|j||dS)Nr)r[r\)r,rar]r-r-r.rts zObserverMarker.__call__cGst|j|tjdSr)r[r\rwr{)r,r]r-r-r.r{ryzObserverMarker.entercCt|jdtjdSNr-r)r[r\rwrxrMr-r-r.rxryzObserverMarker.leavecCrr)r[r\rwr|rMr-r-r.r|ryzObserverMarker.exitcCrr)r[r\rwr}rMr-r-r.r}ryzObserverMarker.back)r\r$r3r4)r]rrarbr3r[)r]rr3r[)r3r[) rXrYrZr/rtr{rxr|r}r-r-r-r.rs   rc@seZdZdZedZedZedZedZedZ edZ edZ ed Z ed Z ed Zed Zed ZedZedZdS)OnMarkera} The `OnMarker` class is used as a marker class to define different types of events in the Scenes. Attributes: - :code:`message`: Event marker for handling `Message` events. - :code:`edited_message`: Event marker for handling edited `Message` events. - :code:`channel_post`: Event marker for handling channel `Post` events. - :code:`edited_channel_post`: Event marker for handling edited channel `Post` events. - :code:`inline_query`: Event marker for handling `InlineQuery` events. - :code:`chosen_inline_result`: Event marker for handling chosen `InlineResult` events. - :code:`callback_query`: Event marker for handling `CallbackQuery` events. - :code:`shipping_query`: Event marker for handling `ShippingQuery` events. - :code:`pre_checkout_query`: Event marker for handling `PreCheckoutQuery` events. - :code:`poll`: Event marker for handling `Poll` events. - :code:`poll_answer`: Event marker for handling `PollAnswer` events. - :code:`my_chat_member`: Event marker for handling my chat `Member` events. - :code:`chat_member`: Event marker for handling chat `Member` events. - :code:`chat_join_request`: Event marker for handling chat `JoinRequest` events. - :code:`error`: Event marker for handling `Error` events. .. note:: This is a marker class and does not contain any methods or implementation logic. messageedited_message channel_postedited_channel_post inline_querychosen_inline_resultrshipping_querypre_checkout_querypoll poll_answermy_chat_member chat_memberchat_join_requestN)rXrYrZrrrr r r r r rrrrrrrrr-r-r-r.rs  r)rr rr r3r4)A __future__rrp collectionsrcollections.abcr dataclassesrrenumrrtypingr r r r typing_extensionsr aiogramraiogram.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.basesrrr[rwrurjrrrrrrrrrronr-r-r-r.sX              J= >ZW ,