o LDiق@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-]/var/www/www-root/data/www/77.83.87.30/venv/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}|sg}t|d||t|j||j|jddS)N__aiogram_handler__)r\handlerr]ra)getattrsetattrr:HandlerContainerr\r]ra)r,rehandlersr-r-r. _wrap_filterws  zObserverDecorator._wrap_filterrcCsR|jdus Jdt|dd}|durtt}t|d|t|||j|j<dS)NzScene action is not specified__aiogram_action__)r_rirdictrjrr\)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_rmrp TypeError)r,remsgr-r-r.__call__s    zObserverDecorator.__call__ActionContainercCt|j|jtjSrO)rvr\r] SceneActionleaverMr-r-r.ryzObserverDecorator.leave type[Scene]cCst|j|jtj|SrO)rvr\r]rxenter)r,rer-r-r.r|szObserverDecorator.entercCrwrO)rvr\r]rxexitrMr-r-r.r}rzzObserverDecorator.exitcCrwrO)rvr\r]rxbackrMr-r-r.r~rzzObserverDecorator.back)NN) r\r$r]r^r_r`rarbr3r4)rerfr3r4)rerr3r4)rerr3r)r3rv)rer{r3rv) rXrYrZr/rmrpruryr|r}r~r-r-r-r.r[js   r[c@s$eZdZeZeZeZeZdS)rxN)rXrYrZrr|ryr}r~r-r-r-r.rxs  rxc@s$eZdZ ddd d ZdddZdS)rvNr\r$r]r^r_rxre 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_rxr|regotoryr}r~r,rr-r-r.executes   zActionContainer.executerO) r\r$r]r^r_rxrerr3r4rrr3r4)rXrYrZr/rr-r-r-r.rvs  rvT)slotsc@s2eZdZUded<ded<ded<dZded <dS) rkr$r\rrhr^r]Nrbra)rXrYrZ__annotations__rar-r-r-r.rks rkc@sbeZdZUded< ded< ded< dZded < dZded < dZded < eZd ed <dS) SceneConfigr0r"zlist[HandlerContainer]rlz,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<r{rhrrarbr3r4cCs||_t||_||_dSrO)r<rrhra)r,r<rhrar-r-r.r/s  zSceneHandlerWrapper.__init__eventrrr c s|d}|d}|d}|jt|jj|||j||dd}|jj||fi|IdH}|jrCtdd|jj|jj}| |j IdH|S)Nr"scenes event_update scene_configmanagerr" update_typerr1rrar-) r<r__scene_config__ event_typerhcallrarvr_rr) r,rrr"rrr<resultaction_containerr-r-r.rus0 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<r{rhrrarbr3r4)rrrr r3r )r3r )r3r$)rXrYrZr/rurrr-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/1s 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"rrrrrrgrarn)r"rlrrrrrr-)rBsuper__init_subclass__rro __bases__ issubclassrrirrrrrextend isinstancer[r:rkr\rr]rahasattrrnitemsupdaterr)clsr state_namerrrrrlrbaseparent_scene_config_namevaluescene_handlersr_action_handlers __class__r-r.r8sd            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)rsetrl observersr\registerrrhrar]raddrfilterrr")rrrused_observersrh observer_namer-r-r. add_to_routerrs(   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>qs 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_rxrr r3r)r1rr3r4)r3r2)r(r$r3r)r(r$rr r3r rO)r(r$rrr3r)r1rrr r3r2rW)rXrYrZrr/r|ryr}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_typerrc 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|IdH}|dur|jjdi|IdHz ||IdH}Wnty<|dur0|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-)rrr}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)rrr3r)r3rr)rrrrrr 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>errorr) rrrouter_middleware_update_middlewarervalues event_name _middleware)r,robserverr-r-r.rs  zSceneRegistry._setup_middlewarerh"NextMiddlewareType[TelegramObject]rrr1r2r csBt|ts Jdt||j|j|d|d|d<|||IdHS)Nz Event must be an Update instancer"rrrr"r1r)rrrrr)r,rhrr1r-r-r.rs z SceneRegistry._update_middlewarecs6|d}t||j||d|d|d<|||IdHS)Nrr"rr)rr)r,rhrr1rr-r-r.r s zSceneRegistry._middlewareN)rrr{ 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,rrrtr<r-r-r.rszSceneRegistry.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.r?szSceneRegistry.registerr<rcCszt|rt|tr|jj}t|tr|j}|dur%t|ts%d}t |z|j |WSt y<d|d}t |w)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) rqisclassrrrr"rrr$rrKeyError)r,r<rtr-r-r.rGHs    zSceneRegistry.getr)rrrrr3r4r)rhrrrr1r2r3r )rr{rrr3r4)rr{r3r4)r<rr3r{) rXrYrZrr/rrrrrrGr-r-r-r.rs   " rc@sLeZdZUded<dZded<ed ddZed d d Zedd d ZdS)Afterrxr_Nrr<r3cC |tjdSNr_)rxr}rr-r-r.r}l z After.exitcCrr)rxr~rr-r-r.r~prz After.backcCs|tj|dS)N)r_r<)rxr|)rr<r-r-r.rtsz After.goto)r3r)r<rr3r) rXrYrZrr<rr}r~rr-r-r-r.rgs    rc@sNeZdZdddZdddddZdddZdddZdddZdddZdS)ObserverMarkerr\r$r3r4cCs ||_dSrOr)r,r\r-r-r.r/zs zObserverMarker.__init__Nrr]rrarbr[cGst|j||dS)Nr)r[r\)r,rar]r-r-r.ru}s zObserverMarker.__call__cGst|j|tjdSr)r[r\rxr|)r,r]r-r-r.r|rzzObserverMarker.entercCt|jdtjdSNr-r)r[r\rxryrMr-r-r.ryrzzObserverMarker.leavecCrr)r[r\rxr}rMr-r-r.r}rzzObserverMarker.exitcCrr)r[r\rxr~rMr-r-r.r~rzzObserverMarker.back)r\r$r3r4)r]rrarbr3r[)r]rr3r[)r3r[) rXrYrZr/rur|ryr}r~r-r-r-r.rys   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__rrq 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[rxrvrkrrrrrrrrrronr-r-r-r.sX              J= 5ZT  ,