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

Add del_file_after_loading param to Saver.return_load_handlers and Saver.load_handlers methods.

Browse Source 
Update telebot/__init__.py
This commit is contained in:
Waffle 2018-05-25 09:40:29 +03:00
parent 7df6b3d4c9
commit 018e4597a2
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
telebot/__init__.py
View File

@ -48,8 +48,8 @@ class Saver:
def save_handlers(self): def save_handlers(self):
self.dump_handlers(self.handlers, self.filename) self.dump_handlers(self.handlers, self.filename)
def load_handlers(self, filename): def load_handlers(self, filename, del_file_after_loading=True):
tmp = self.return_load_handlers(filename) tmp = self.return_load_handlers(filename, del_file_after_loading=del_file_after_loading)
if tmp is not None: if tmp is not None:
self.handlers.update(tmp) self.handlers.update(tmp)
@ -79,7 +79,7 @@ class Saver:
os.rename(filename + ".tmp", filename) os.rename(filename + ".tmp", filename)
@staticmethod @staticmethod
def return_load_handlers(filename): def return_load_handlers(filename, del_file_after_loading=True):
if os.path.isfile(filename) and os.path.getsize(filename) > 0: if os.path.isfile(filename) and os.path.getsize(filename) > 0:
with open(filename, "r") as file: with open(filename, "r") as file:
handlers = json.load(file) handlers = json.load(file)
@ -105,7 +105,8 @@ class Saver:
callback = getattr(sys.modules["__main__"], "next_") callback = getattr(sys.modules["__main__"], "next_")
handler["callback"] = callback""" handler["callback"] = callback"""
if del_file_after_loading:
os.remove(filename)
return result return result