o @Mh!@sddlZddlZddlZddlmZz ddlmZdZWndZYddlm Z m Z Gddde Z Gd d d e Z Gd d d e ZGd dde ZGdddZGdddZGdddZGdddZdS)N) apihelper)RedisTF)State StatesGroupc@s2eZdZdZd ddZddZddZd d ZdS) HandlerBackendzJ Class for saving (next step|reply) handlers. :meta private: NcCs|duri}||_dSN)handlers)selfrr K/var/www/html/venv/lib/python3.10/site-packages/telebot/handler_backends.py__init__s zHandlerBackend.__init__cCtrNotImplementedErrorr handler_group_idhandlerr r r register_handlerzHandlerBackend.register_handlercCr rrr rr r r clear_handlersrzHandlerBackend.clear_handlerscCr rrrr r r get_handlers rzHandlerBackend.get_handlersr)__name__ __module__ __qualname____doc__r rrrr r r r rs   rc@s0eZdZdZddZddZddZdd Zd S) MemoryHandlerBackend :meta private: cCs.||jvr|j||dS|g|j|<dSr)rappendrr r r r(s z%MemoryHandlerBackend.register_handlercCs|j|ddSrrpoprr r r r.sz#MemoryHandlerBackend.clear_handlerscCs|j|dSrrrr r r r1sz!MemoryHandlerBackend.get_handlerscCr rr)r filenamedel_file_after_loadingr r r load_handlers4rz"MemoryHandlerBackend.load_handlersN)rrrrrrrr#r r r r r$s  rcspeZdZdZdfdd ZddZd d Zd d Zd dZddZ dddZ e dddZ e dddZ ZS)FileHandlerBackendrN./.handler-saves/handlers.savexcs0tt||||_||_t||j|_dSr) superr$r r!delay threadingTimer save_handlerstimer)r rr!r( __class__r r r <szFileHandlerBackend.__init__cCs4||jvr|j||n|g|j|<|dSr)rrstart_save_timerrr r r rBs   z#FileHandlerBackend.register_handlercCs|j|d|dSrrr r/rr r r rIs z!FileHandlerBackend.clear_handlerscCs|j|d}||Srr0)r rrr r r rMszFileHandlerBackend.get_handlerscCsD|js |jdkr|dSt|j|j|_|jdSdS)Nr)r,is_aliver(r+r)r*startr r r r r/Rs   z#FileHandlerBackend.start_save_timercCs||j|jdSr) dump_handlersrr!r3r r r r+Zz FileHandlerBackend.save_handlersTcCs4|s|j}|j||d}|dur|j|dSdS)N)r")r!return_load_handlersrupdate)r r!r"tmpr r r r#]s z FileHandlerBackend.load_handlerswbcCs|jdddd}tj|ddt|d|}tjdur$t||ntj||Wdn1s5wYtj |rEt |t |d|dS)N/)maxsplitrT)exist_okz.tmp) rsplitosmakedirsopenrCUSTOM_SERIALIZERpickledumppathisfileremoverename)rr! file_modedirsfiler r r r4ds   z FileHandlerBackend.dump_handlerscCstj|r=tj|dkr?t|d}tjdurt|}ntj|}Wdn1s/wY|r;t ||SdSdS)Nrrb) r?rErFgetsizerArrBrCloadrG)r!r"rKrr r r r6ts     z'FileHandlerBackend.return_load_handlers)Nr%r&)NT)r9)T)rrrrr rrrr/r+r# staticmethodr4r6 __classcell__r r r-r r$8s  r$csBeZdZdZdfdd Zd d Zd d Zd dZddZZ S)RedisHandlerBackendrN localhostrtelebotcs6tt||tstd||_t|||||_dS)Nz:Redis is not installed. Install it via 'pip install redis')r'rQr redis_installed Exceptionprefixrredis)r rhostportdbrWpasswordr-r r r s zRedisHandlerBackend.__init__cCsd|jt|fS)N:)joinrWstr)r handle_group_idr r r _keyr5zRedisHandlerBackend._keycCsLg}|j||}|rt|}|||j||t|dSr)rXgetrarCloadsrsetdumps)r rrrvaluer r r rs   z$RedisHandlerBackend.register_handlercCs|j||dSr)rXdeleterarr r r rsz"RedisHandlerBackend.clear_handlerscCs2d}|j||}|rt|}|||Sr)rXrbrarCrcr)r rrrfr r r rs   z RedisHandlerBackend.get_handlers)NrRrSrrTN) rrrrr rarrrrPr r r-r rQsrQc@s6eZdZUdZdZeed<ddZddZdd Z d S) BaseMiddlewarea- Base class for middleware. Your middlewares should be inherited from this class. Set update_sensitive=True if you want to get different updates on different functions. For example, if you want to handle pre_process for message update, then you will have to create pre_process_message function, and so on. Same applies to post_process. .. note:: If you want to use middleware, you have to set use_class_middlewares=True in your TeleBot instance. .. code-block:: python3 :caption: Example of class-based middlewares. class MyMiddleware(BaseMiddleware): def __init__(self): self.update_sensitive = True self.update_types = ['message', 'edited_message'] def pre_process_message(self, message, data): # only message update here pass def post_process_message(self, message, data, exception): pass # only message update here for post_process def pre_process_edited_message(self, message, data): # only edited_message update here pass def post_process_edited_message(self, message, data, exception): pass # only edited_message update here for post_process Fupdate_sensitivecCdSrr r3r r r r zBaseMiddleware.__init__cCtrr)r messagedatar r r pre_processrkzBaseMiddleware.pre_processcCrlrr)r rmrn exceptionr r r post_processrkzBaseMiddleware.post_processN) rrrrribool__annotations__r rorqr r r r rhs  $ rhc@eZdZdZdddZdS) SkipHandlerz Class for skipping handlers. Just return instance of this class in middleware to skip handler. Update will go to post_process, but will skip execution of handler. returnNcCrjrr r3r r r r rkzSkipHandler.__init__rvNrrrrr r r r r ruruc@rt) CancelUpdatez Class for canceling updates. Just return instance of this class in middleware to skip update. Update will skip handler and execution of post_process in middlewares. rvNcCrjrr r3r r r r rkzCancelUpdate.__init__rwrxr r r r rzryrzc@rt)ContinueHandlinga  Class for continue updates in handlers. Just return instance of this class in handlers to continue process. .. code-block:: python3 :caption: Example of using ContinueHandling @bot.message_handler(commands=['start']) def start(message): bot.send_message(message.chat.id, 'Hello World!') return ContinueHandling() @bot.message_handler(commands=['start']) def start2(message): bot.send_message(message.chat.id, 'Hello World2!') rvNcCrjrr r3r r r r rkzContinueHandling.__init__rwrxr r r r r{sr{)r?rCr)rTrrXrrUtelebot.statesrrobjectrrr$rQrhrurzr{r r r r s$  K"1