mirror/pyTelegramBotAPI
1
0
Fork 0
mirror of https://github.com/eternnoir/pyTelegramBotAPI.git synced 2023-08-10 21:12:57 +03:00
Code Releases Activity

next_step_handler done.

Browse Source
This commit is contained in:
eternnoir
2015-07-30 11:02:08 +08:00
parent a82fece950
commit 9f6a3bfb26
1 changed files with 25 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
telebot/__init__.py
View File

@ -99,6 +99,9 @@ class TeleBot:
self.message_subscribers_messages = [] self.message_subscribers_messages = []
self.message_subscribers_callbacks = [] self.message_subscribers_callbacks = []
# key: chat_id, value: handler list
self.message_subscribers_next_step = {}
self.message_handlers = [] self.message_handlers = []
if self.__create_threads: if self.__create_threads:
self.worker_pool = ThreadPool(num_threads) self.worker_pool = ThreadPool(num_threads)
@ -124,6 +127,7 @@ class TeleBot:
self.__notify_update(new_messages) self.__notify_update(new_messages)
self._notify_command_handlers(new_messages) self._notify_command_handlers(new_messages)
self._notify_message_subscribers(new_messages) self._notify_message_subscribers(new_messages)
self._notify_message_next_handler(new_messages)
def __notify_update(self, new_messages): def __notify_update(self, new_messages):
for listener in self.update_listener: for listener in self.update_listener:
@ -324,13 +328,6 @@ class TeleBot:
self.message_subscribers_messages.pop() self.message_subscribers_messages.pop()
self.message_subscribers_callbacks.pop() self.message_subscribers_callbacks.pop()
def next_step_handler(self, message):
"""
:param message:
"""
pass
def _notify_message_subscribers(self, new_messages): def _notify_message_subscribers(self, new_messages):
for message in new_messages: for message in new_messages:
if not hasattr(message, 'reply_to_message'): if not hasattr(message, 'reply_to_message'):
@ -344,6 +341,27 @@ class TeleBot:
del self.message_subscribers_messages[index] del self.message_subscribers_messages[index]
del self.message_subscribers_callbacks[index] del self.message_subscribers_callbacks[index]
def register_next_step_handler(self, message, callback):
"""
Registers a callback function to be notified when new message arrives after `message`.
:param message: The message for which we want to handle new message after that in same chat.
:param callback: The callback function which next new message arrives.
"""
chat_id = message.chat.id
if chat_id in self.message_subscribers_next_step:
self.message_subscribers_next_step.append(callback)
else:
self.message_subscribers_next_step[chat_id] = [callback]
def _notify_message_next_handler(self, new_messages):
for message in new_messages:
chat_id = message.chat.id
if chat_id in self.message_subscribers_next_step:
handlers = self.message_subscribers_next_step[chat_id]
for handler in handlers:
self.worker_pool.put(handler, message)
self.message_subscribers_next_step.pop(chat_id, None)
def message_handler(self, commands=None, regexp=None, func=None, content_types=['text']): def message_handler(self, commands=None, regexp=None, func=None, content_types=['text']):
""" """