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

Rewrite.

Browse Source 
Add class 'Saver' that provides methods for saving (next step|reply) handlers.
Add methods
enable_save_next_step_handlers,
enable_save_reply_handlers,
disable_save_next_step_handlers,
disable_save_reply_handlers,
load_next_step_handlers,
load_reply_handlers
to Telebot and AsyncTelebot.

update telebot/__init__.py
This commit is contained in:
Waffle 2018-05-25 08:56:31 +03:00
parent 00c8dcc19b
commit b1d5cb2129
1 changed files with 100 additions and 96 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

196
telebot/__init__.py
View File

@ -30,6 +30,84 @@ Module : telebot
"""
class Saver:
def __init__(self, handlers, filename, delay):
self.handlers = handlers
self.filename = filename
self.delay = delay
self.timer = threading.Timer(delay, self.save_handlers)
def start_save_timer(self):
if not self.timer.is_alive():
self.timer = threading.Timer(self.delay, self.save_handlers)
self.timer.start()
def save_handlers(self):
self.dump_handlers(self.handlers, self.filename)
def load_handlers(self, filename):
tmp = self.return_load_handlers(filename)
if tmp is not None:
self.handlers.update(tmp)
@staticmethod
def dump_handlers(handlers, filename, file_mode="w"):
dirs = filename.rsplit('/', maxsplit=1)[0]
os.makedirs(dirs, exist_ok=True)
to_dump = {}
with open(filename + ".tmp", file_mode) as file:
for id_, handlers_ in handlers.items():
for handler in handlers_:
name = handler['callback'].__name__
module = handler['callback'].__module__
tmp = {"callback": {"module": module, "name": name}, "args": handler["args"],
"kwargs": handler["kwargs"]}
if id_ in to_dump.keys():
to_dump[id_].append(tmp)
else:
to_dump[id_] = [tmp]
json.dump(to_dump, file)
if os.path.isfile(filename):
os.remove(filename)
os.rename(filename + ".tmp", filename)
@staticmethod
def return_load_handlers(filename):
if os.path.isfile(filename) and os.path.getsize(filename) > 0:
with open(filename, "r") as file:
handlers = json.load(file)
result = {}
for id_, handlers_ in handlers.items():
for handler in handlers_:
name = handler['callback']["name"]
module = handler['callback']["module"]
callback = getattr(sys.modules[module], name)
tmp = {"callback": callback, "args": handler["args"], "kwargs": handler["kwargs"]}
if int(id_) in result.keys():
result[int(id_)].append(tmp)
else:
result[int(id_)] = [tmp]
a = """ for key, handlers_ in handlers.items():
for handler in handlers_:
name = handler["callback"]["name"]
module = handler["callback"]["module"]
callback = getattr(sys.modules["__main__"], "next_")
handler["callback"] = callback"""
return result
class TeleBot:
""" This is TeleBot Class
Methods:
@ -88,15 +166,8 @@ class TeleBot:
# key: chat_id, value: handler list
self.next_step_handlers = {}
self.save_reply_handlers = False
self.reply_handlers_save_file = None
self.reply_save_delay = None
self.save_reply_timer = None
self.save_step_handlers = False
self.step_handlers_save_file = None
self.step_save_delay = None
self.save_step_timer = None
self.next_step_saver = None
self.reply_saver = None
self.message_handlers = []
self.edited_message_handlers = []
@ -112,72 +183,23 @@ class TeleBot:
if self.threaded:
self.worker_pool = util.ThreadPool(num_threads=num_threads)
@staticmethod
def dump_handlers(handlers, filename, file_mode="w"):
dirs = filename.rsplit('/', maxsplit=1)[0]
os.makedirs(dirs, exist_ok=True)
with open(filename + ".tmp", file_mode) as file:
for mid, handler in handlers.items():
name = handler['callback'].__name__
module = handler['callback'].__module__
tmp = {str(mid): {"callback": {"module": module, "name": name}, "args": handler["args"],
"kwargs": handler["kwargs"]}}
json.dump(tmp, file)
if os.path.isfile(filename):
os.remove(filename)
os.rename(filename + ".tmp", filename)
def save_next_step_handlers(self):
self.dump_handlers(self.next_step_handlers, self.step_handlers_save_file)
def save_reply_handlers_method(self):
self.dump_handlers(self.next_step_handlers, self.step_handlers_save_file)
def start_save_next_step_timer(self):
if self.save_step_timer.is_alive(): return
self.save_step_timer.start()
def start_save_reply_timer(self):
if self.save_reply_timer.is_alive(): return
self.save_reply_timer.start()
def enable_save_next_step_handlers(self, delay=120, filename="./handler-saves/step.save"):
self.save_step_handlers = True
self.step_handlers_save_file = filename
self.step_save_delay = delay
self.save_step_timer = threading.Timer(self.step_save_delay, self.save_next_step_handlers)
self.next_step_saver = Saver(self.next_step_handlers, filename, delay)
def enable_save_reply_handlers(self, delay=120, filename="./handler-saves/reply.save"):
self.save_reply_handlers = True
self.reply_handlers_save_file = filename
self.reply_save_delay = delay
self.save_reply_timer = threading.Timer(self.reply_save_delay, self.save_reply_handlers_method)
self.reply_saver = Saver(self.reply_handlers, filename, delay)
@staticmethod
def load_handlers(filename):
if os.path.isfile(filename) and os.path.getsize(filename) > 0:
with open(filename, "r") as file:
handlers = json.load(file)
def disable_save_next_step_handlers(self):
self.next_step_saver = None
for handler in handlers:
name = handler["callback"]["name"]
module = handler["callback"]["module"]
handler = getattr(sys.modules[module], name)
return handlers
def disable_save_reply_handlers(self):
self.reply_saver = None
def load_next_step_handlers(self, filename="./handler-saves/step.save"):
self.next_step_handlers.update(self.load_handlers(filename))
self.next_step_saver.load_handlers(filename)
def load_reply_handlers(self, filename="./handler-saves/reply.save"):
self.reply_handlers.update(self.load_handlers(filename))
self.reply_saver.load_handlers(filename)
def set_webhook(self, url=None, certificate=None, max_connections=None, allowed_updates=None):
return apihelper.set_webhook(self.token, url, certificate, max_connections, allowed_updates)
@ -1158,7 +1180,7 @@ class TeleBot:
else:
self.reply_handlers[message_id] = [{"callback": callback, "args": args, "kwargs": kwargs}]
self.start_save_reply_timer()
self.reply_saver.start_save_timer()
def _notify_reply_handlers(self, new_messages):
for message in new_messages:
@ -1169,7 +1191,7 @@ class TeleBot:
for handler in handlers:
self._exec_task(handler["callback"], message, *handler["args"], **handler["kwargs"])
self.reply_handlers.pop(reply_msg_id)
self.start_save_reply_timer()
self.reply_saver.start_save_timer()
def register_next_step_handler(self, message, callback, *args, **kwargs):
"""
@ -1197,7 +1219,7 @@ class TeleBot:
else:
self.next_step_handlers[chat_id] = [{"callback": callback, "args": args, "kwargs": kwargs}]
self.start_save_next_step_timer()
self.next_step_saver.start_save_timer()
def clear_step_handler(self, message):
"""
@ -1216,7 +1238,7 @@ class TeleBot:
"""
self.next_step_handlers[chat_id] = []
self.start_save_next_step_timer()
self.next_step_saver.start_save_timer()
def clear_reply_handlers(self, message):
"""
@ -1235,7 +1257,7 @@ class TeleBot:
"""
self.reply_handlers[message_id] = []
self.start_save_next_step_timer()
self.reply_saver.start_save_timer()
def _notify_next_handlers(self, new_messages):
i = 0
@ -1250,7 +1272,7 @@ class TeleBot:
new_messages.pop(i) # removing message that detects with next_step_handler
i += 1
self.start_save_next_step_timer()
self.next_step_saver.start_save_timer()
@staticmethod
def _build_handler_dict(handler, **filters):
@ -1441,27 +1463,6 @@ class AsyncTeleBot(TeleBot):
def __init__(self, *args, **kwargs):
TeleBot.__init__(self, *args, **kwargs)
@staticmethod
@util.async()
def dump_handlers(handlers, filename, file_mode="w"):
return TeleBot.dump_handlers(handlers, filename, file_mode)
@util.async()
def save_next_step_handlers(self):
return TeleBot.save_next_step_handlers(self)
@util.async()
def save_reply_handlers_method(self):
return TeleBot.save_reply_handlers_method(self)
@util.async()
def start_save_next_step_timer(self):
return TeleBot.start_save_next_step_timer(self)
@util.async()
def start_save_reply_timer(self):
return TeleBot.start_save_reply_timer(self)
@util.async()
def enable_save_next_step_handlers(self, delay=120, filename="./handler-saves/step.save"):
return TeleBot.enable_save_next_step_handlers(self, delay, filename)
@ -1470,10 +1471,13 @@ class AsyncTeleBot(TeleBot):
def enable_save_reply_handlers(self, delay=120, filename="./handler-saves/reply.save"):
return TeleBot.enable_save_reply_handlers(self, delay, filename)
@staticmethod
@util.async()
def load_handlers(filename):
return TeleBot.load_handlers(filename)
def disable_save_next_step_handlers(self):
return TeleBot.disable_save_next_step_handlers(self)
@util.async()
def disable_save_reply_handlers(self):
return TeleBot.enable_save_reply_handlers(self)
@util.async()
def load_next_step_handlers(self, filename="./handler-saves/step.save"):