a /3e"@sddlZddlZddlZddlmZzddlmZdZWndZYn0GdddeZ Gdd d e Z Gd d d e Z Gd d d e Z GdddZ GdddZGdddZGdddZGdddZGdddZdS)N) apihelper)RedisTFc@s2eZdZdZd ddZddZddZd d ZdS) HandlerBackendzJ Class for saving (next step|reply) handlers. :meta private: NcCs|dur i}||_dSN)handlers)selfrrK/home/pi/bot/my_env/lib/python3.9/site-packages/telebot/handler_backends.py__init__szHandlerBackend.__init__cCs tdSrNotImplementedErrorrhandler_group_idhandlerrrr register_handlerszHandlerBackend.register_handlercCs tdSrr rrrrr clear_handlersszHandlerBackend.clear_handlerscCs tdSrr rrrr get_handlersszHandlerBackend.get_handlers)N)__name__ __module__ __qualname____doc__r rrrrrrr r s  rc@s0eZdZdZddZddZddZdd Zd S) MemoryHandlerBackend :meta private: cCs,||jvr|j||n |g|j|<dSr)rappendr rrr r&s z%MemoryHandlerBackend.register_handlercCs|j|ddSrrpoprrrr r,sz#MemoryHandlerBackend.clear_handlerscCs|j|dSrrrrrr r/sz!MemoryHandlerBackend.get_handlerscCs tdSrr )rfilenamedel_file_after_loadingrrr load_handlers2sz"MemoryHandlerBackend.load_handlersN)rrrrrrrrrrrr 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 rdelay threadingTimer save_handlerstimer)rrrr$ __class__rr r :szFileHandlerBackend.__init__cCs4||jvr|j||n |g|j|<|dSr)rrstart_save_timerr rrr r@s  z#FileHandlerBackend.register_handlercCs|j|d|dSrrrr+rrrr rGsz!FileHandlerBackend.clear_handlerscCs|j|d}||Srr,)rrrrrr rKszFileHandlerBackend.get_handlerscCs>|js:|jdkr|nt|j|j|_|jdS)Nr)r(is_aliver$r'r%r&startrrrr r+Ps    z#FileHandlerBackend.start_save_timercCs||j|jdSr) dump_handlersrrr/rrr r'Xsz FileHandlerBackend.save_handlersTcCs0|s |j}|j||d}|dur,|j|dS)N)r)rreturn_load_handlersrupdate)rrrtmprrr r[s z FileHandlerBackend.load_handlerswbcCs|jdddd}tj|ddt|d|6}tjdurHt||ntj||Wdn1sj0Ytj |rt |t |d|dS)N/)maxsplitrT)exist_okz.tmp) rsplitosmakedirsopenrCUSTOM_SERIALIZERpickledumppathisfileremoverename)rrZ file_modedirsfilerrr r0bs ,  z FileHandlerBackend.dump_handlerscCs~tj|rztj|dkrzt|d2}tjdur>t|}n tj|}Wdn1s^0Y|rvt ||SdS)Nrrb) r:r@rAgetsizer<rr=r>loadrB)rrrErrrr r1rs   * z'FileHandlerBackend.return_load_handlers)Nr!r")NT)r4)T)rrrrr rrrr+r'r staticmethodr0r1 __classcell__rrr)r r 6s  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#rKr redis_installed Exceptionprefixrredis)rrhostportdbrQpasswordr)rr r s zRedisHandlerBackend.__init__cCsd|jt|fS)N:)joinrQstr)rZhandle_group_idrrr _keyszRedisHandlerBackend._keycCsLg}|j||}|r$t|}|||j||t|dSr)rRgetrZr>loadsrsetdumps)rrrrvaluerrr rs   z$RedisHandlerBackend.register_handlercCs|j||dSr)rRdeleterZrrrr rsz"RedisHandlerBackend.clear_handlerscCs2d}|j||}|r.t|}|||Sr)rRr[rZr>r\r)rrrr_rrr rs   z RedisHandlerBackend.get_handlers)NrLrMrrNN) rrrrr rZrrrrJrrr)r rKs rKc@s,eZdZdZddddZedddZdS)Statez Class representing a state. .. code-block:: python3 class MyStates(StatesGroup): my_state = State() # returns my_state:State string. NreturncCs d|_dSrnamer/rrr r szState.__init__cCs|jSrrdr/rrr __str__sz State.__str__)rrrrr rYrfrrrr rasrac@s*eZdZdZddddZeddZdS) StatesGroupz Class representing common states. .. code-block:: python3 class MyStates(StatesGroup): my_state = State() # returns my_state:State string. NrbcCs`g}|jD]F\}}|dst|st|trd|j|f|_||_ | |q||_ dS)N__rW) __dict__items startswithcallable isinstancerarXrregroupr _state_list)cls state_listrer_rrr __init_subclass__s zStatesGroup.__init_subclass__cCs|jSr)ror/rrr rqszStatesGroup.state_list)rrrrrrpropertyrqrrrr rgs rgc@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_sensitivecCsdSrrr/rrr r szBaseMiddleware.__init__cCstdSrr )rmessagedatarrr pre_processszBaseMiddleware.pre_processcCstdSrr )rrvrw exceptionrrr post_processszBaseMiddleware.post_processN) rrrrrubool__annotations__r rxrzrrrr rts $ rtc@seZdZdZddddZdS) 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. NrbcCsdSrrr/rrr r szSkipHandler.__init__rrrrr rrrr r}sr}c@seZdZdZddddZdS) 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. NrbcCsdSrrr/rrr r szCancelUpdate.__init__r~rrrr rsrc@seZdZdZddddZdS)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!') NrbcCsdSrrr/rrr r &szContinueHandling.__init__r~rrrr rsr)r:r>r%rNrrRrrOobjectrrr rKrargrtr}rrrrrr s$   K"1