o Fhb@sddlmZddlZddlmZddlmZmZddlm 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/m0Z0m1Z1ddl2m3Z3m4Z4GdddZ5GdddZ6Gddde Z7GdddZ8GdddZ9eGdd d Z:d8d&d'Z;Gd(d)d)ZGd.d/d/Z?Gd0d1d1Z@eGd2d3d3ZAGd4d5d5ZBGd6d7d7ZCeCZDdS)9) annotationsN) defaultdict) dataclassreplace)Enumauto) AnyClassVarDictListMappingOptionalTupleTypeUnionoverload)Self)loggers) Dispatcher)NextMiddlewareType)CallableObject CallbackType)extract_flags_from_object)Router)SceneException) StateFilter) FSMContext)StateMemoryStorageRecord)TelegramObjectUpdate)ClassAttrsResolverget_sorted_mro_attrs_resolverc@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*r2D/var/www/html/venv/lib/python3.10/site-packages/aiogram/fsm/scene.py__init__)s  zHistoryManager.__init__ Optional[str]dataDict[str, Any]returnNonecst|jIdH}|dg}|||dt||jkr&||j d}tjd|||jj |dIdHdS)Nhistoryr'r6z Push state=%s data=%s to historyr:) r0get_data setdefaultappendlenr.rscenedebug update_data)r1r'r6 history_datar:r2r2r3push0s zHistoryManager.pushOptional[MemoryStorageRecord]cs|jIdH}|dg}|sdS|}|d}|d}|s+|jiIdHn |jj|dIdHtjd||t ||dS)Nr:r'r6r<z!Pop state=%s data=%s from historyr;) r0r=r>popset_datarCrrArBr)r1rDr:recordr'r6r2r2r3rG:s  zHistoryManager.popcs8|jIdH}|dg}|sdStdi|dS)Nr:r2)r0r=r>rr1rDr:r2r2r3getIs  zHistoryManager.getList[MemoryStorageRecord]cs,|jIdH}|dg}dd|DS)Nr:cSsg|] }tdi|qS)r2r).0itemr2r2r3 Ssz&HistoryManager.all..)r0r=r>rKr2r2r3allPs zHistoryManager.allcs$tjd|jiIdHdS)Nz Clear history)rrArBr0rHr1r2r2r3clearUs zHistoryManager.clearcs8|jIdH}|jIdH}|||IdHdSN)r/ get_stater=rEr1r'r6r2r2r3snapshotYszHistoryManager.snapshotcs*|j|IdH|j|IdHdSrT)r/ set_staterHrVr2r2r3 _set_state^szHistoryManager._set_statecsZ|IdH}|s|diIdHdStjd|j|j||j|jIdH|jS)NzRollback to state=%s data=%s)rGrYrrArBr'r6)r1previous_stater2r2r3rollbackbszHistoryManager.rollbackN)r%r&)r'rr(r)r*r+)r'r5r6r7r8r9)r8rF)r8rMr8r9)r8r5) __name__ __module__ __qualname__r4rErGrLrQrSrWrYr[r2r2r2r3r$(s       r$c@sbeZdZ  d!d"d d Zd#ddZd$ddZd%ddZd&ddZd'ddZd&ddZ d&dd Z dS)(ObserverDecoratorNnamer)filterstuple[CallbackType, ...]actionSceneAction | NoneafterOptional[After]r8r9cC||_||_||_||_dSrT)rarbrdrf)r1rarbrdrfr2r2r3r4r zObserverDecorator.__init__targetType[Scene] | CallbackTypecCs@t|dd}|sg}t|d||t|j||j|jddS)N__aiogram_handler__rahandlerrbrf)getattrsetattrr?HandlerContainerrarbrf)r1rjhandlersr2r2r3 _wrap_filter~s  zObserverDecorator._wrap_filterrcCsR|jdus Jdt|dd}|durtt}t|d|t|||j|j<dS)NzScene action is not specified__aiogram_action__)rdrordictrprra)r1rjrdr2r2r3 _wrap_actions   zObserverDecorator._wrap_actioncCs8t|r|jdur|||S|||Std)Nz"Only function or method is allowed)inspect isfunctionrdrsrv TypeErrorr1rjr2r2r3__call__s    zObserverDecorator.__call__ActionContainercCt|j|jtjSrT)r|rarb SceneActionleaverRr2r2r3rzObserverDecorator.leave Type[Scene]cCst|j|jtj|SrT)r|rarbr~enterrzr2r2r3rszObserverDecorator.entercCr}rT)r|rarbr~exitrRr2r2r3rrzObserverDecorator.exitcCr}rT)r|rarbr~backrRr2r2r3rrzObserverDecorator.back)NN) rar)rbrcrdrerfrgr8r9)rjrkr8r9)rjrr8r9)rjrr8r)r8r|)rjrr8r|) r]r^r_r4rsrvr{rrrrr2r2r2r3r`qs   r`c@s$eZdZeZeZeZeZdS)r~N)r]r^r_rrrrrr2r2r2r3r~s  r~c@s$eZdZ ddd d ZdddZdS)r|Nrar)rbTuple[CallbackType, ...]rdr~rj(Optional[Union[Type[Scene], State, str]]r8r9cCrhrT)rarbrdrj)r1rarbrdrjr2r2r3r4rizActionContainer.__init__wizard SceneWizardcs|jtjkr|jdur||jIdHdS|jtjkr&|IdHdS|jtjkr5|IdHdS|jtjkrD|IdHdSdSrT)rdr~rrjgotorrrr1rr2r2r3executes   zActionContainer.executerT) rar)rbrrdr~rjrr8r9rrr8r9)r]r^r_r4rr2r2r2r3r|s  r|c@seZdZ ddd d ZdS)rqNrar)rnrrbrrfrgr8r9cCrhrTrm)r1rarnrbrfr2r2r3r4rizHandlerContainer.__init__rT) rar)rnrrbrrfrgr8r9)r]r^r_r4r2r2r2r3rqsrqc@sbeZdZUded< ded< ded< dZded < dZded < dZded < eZd ed <dS) SceneConfigr5r'zList[HandlerContainer]rrz,Dict[SceneAction, Dict[str, CallableObject]]actionsNzOptional[bool]reset_data_on_enterreset_history_on_entercallback_query_without_stater"attrs_resolver) r]r^r___annotations__rrrr#rr2r2r2r3rs     rargsrkwargsr8r9csdSrTr2)rrr2r2r3_empty_handlersrc@s8eZdZ ddd d ZdddZdddZdddZdS)SceneHandlerWrapperNrArrnrrfrgr8r9cCs||_t||_||_dSrT)rArrnrf)r1rArnrfr2r2r3r4s  zSceneHandlerWrapper.__init__eventr rrc s|d}|d}|d}|jt|jj|||j||dd}|jj||fi|IdH}|jrCtdd|jj|jj}| |j IdH|S)Nr'scenes event_update scene_configmanagerr' update_typerr6rrfr2) rAr__scene_config__ event_typerncallrfr|rdrr) r1rrr'rrrAresultaction_containerr2r2r3r{s0 zSceneHandlerWrapper.__call__rcCs|SrTr2rRr2r2r3 __await__!szSceneHandlerWrapper.__await__r)cCs8d|jd|jj}|jr|d|j7}|d7}|S)NzSceneHandlerWrapper(z, z, after=))rArncallbackrf)r1rr2r2r3__str__$s zSceneHandlerWrapper.__str__rT)rArrnrrfrgr8r9)rr rrr8r)r8r)r8r))r]r^r_r4r{rrr2r2r2r3rs  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]rrrr8r9cCs||_||j_dSrT)rrArr2r2r3r4=s zScene.__init__rrc 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'rrrrrrlrfrt)r'rrrrrrrr2)rGsuper__init_subclass__rru __bases__ issubclassrrorrrrr#extend isinstancer`r?rqrarrbrfhasattrrtitemsupdaterr)clsr state_namerrrrrrrbaseparent_scene_configravaluescene_handlersrdaction_handlers __class__r2r3rDsd            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)rsetrr observersraregisterrrnrfrbraddrfilterrr')rrrused_observersrn observer_namer2r2r3 add_to_router~s(   zScene.add_to_routerNrar5cCs>|durd|jd|jd|jj}t|d}|||S)zM Returns the scene as a router. :return: new router NzScene '.z ' for state ra)r^r_rr'rr)rrarr2r2r3 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")) rr r ScenesManagermiddleware_kwargsrr8r9cs&|jfii|IdHdSrT)r)rrrrrr2r3enter_to_scene_handlers$z0Scene.as_handler..enter_to_scene_handlerN)rr rrrrr8r9r2)rrrr2rr3 as_handlers zScene.as_handlerrrrr8r9rrr8r9rT)rar5r8r)rrr8r) r]r^r___doc__rr4r classmethodrrr __classcell__r2r2rr3r,s   : rc@seZdZdZd;ddZdddZderrorr) rrrouter_middleware_update_middlewarervalues event_name _middleware)r1robserverr2r2r3rs  zSceneRegistry._setup_middlewarern"NextMiddlewareType[TelegramObject]rr r6r7rcsBt|ts Jdt||j|j|d|d|d<|||IdHS)Nz Event must be an Update instancer'rrrr'r6r)rr!rrr)r1rnrr6r2r2r3rs z SceneRegistry._update_middlewarecs6|d}t||j||d|d|d<|||IdHS)Nrr'rr)rr)r1rnrr6rr2r2r3rs zSceneRegistry._middlewareN)rrrOptional[Router]cGst|std|D]/}|jj|jvrtd|jjd||j|jj<|r,||q|jr7|j|qdS)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)r1rrrAr2r2r3r$szSceneRegistry.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)r1rr2r2r3rFszSceneRegistry.registerrArcCsrt|rt|tr|jj}t|tr|j}|dur#t|ts#t dz|j |WSt y8t d|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) rwisclassrrrr'rrr)rrKeyError)r1rAr2r2r3rLOs   zSceneRegistry.getr)rrrrr8r9r)rnrrr r6r7r8r)rrrrr8r9)rrr8r9)rArr8r) r]r^r_rr4rrrrrrLr2r2r2r3rs   " rc@sLeZdZUded<dZded<ed ddZed d d Zedd d ZdS)Afterr~rdNrrAr8cC |tjdSNrd)r~rrr2r2r3rq z After.exitcCrr)r~rrr2r2r3rurz After.backcCs|tj|dS)N)rdrA)r~r)rrAr2r2r3rysz After.goto)r8r)rArr8r) r]r^r_rrArrrrr2r2r2r3rls    rc@sNeZdZdddZdddddZdddZdddZdddZdddZdS)ObserverMarkerrar)r8r9cCs ||_dSrTr)r1rar2r2r3r4s zObserverMarker.__init__Nrrbrrfrgr`cGst|j||dS)Nr)r`ra)r1rfrbr2r2r3r{s zObserverMarker.__call__cGst|j|tjdSr)r`rar~r)r1rbr2r2r3rrzObserverMarker.entercCt|jdtjdSNr2r)r`rar~rrRr2r2r3rrzObserverMarker.leavecCr r )r`rar~rrRr2r2r3rrzObserverMarker.exitcCr r )r`rar~rrRr2r2r3rrzObserverMarker.back)rar)r8r9)rbrrfrgr8r`)rbrr8r`)r8r`) r]r^r_r4r{rrrrr2r2r2r3r~s   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)r]r^r_rrr r rrrrrrrrrrrrr2r2r2r3r s  r )rrrrr8r9)E __future__rrw collectionsr dataclassesrrenumrrtypingrr r r r r rrrrtyping_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$r`r~r|rqrrrrrrrrrr onr2r2r2r3sR  0          I< 5WT ,