mirror of https://github.com/nocturn9x/BotBase.git
45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
from pyrogram.errors import RPCError, FloodWait
|
|
import time
|
|
import logging
|
|
|
|
|
|
def edit_message_text(update, *args, **kwargs):
|
|
"""Edits a message in a way that never triggers exceptions and logs errors"""
|
|
|
|
try:
|
|
return update.edit_message_text(*args, **kwargs)
|
|
except FloodWait as fw:
|
|
logging.warning(f"FloodWait! Sleeping {fw.x} seconds")
|
|
time.sleep(fw.x)
|
|
except RPCError as generic_error:
|
|
logging.error(f"An exception occurred: {generic_error}")
|
|
return False
|
|
|
|
|
|
def edit_message_caption(update, *args, **kwargs):
|
|
"""Edits a message caption in a way that never triggers exceptions and logs errors"""
|
|
|
|
try:
|
|
return update.edit_message_caption(*args, **kwargs)
|
|
except FloodWait as fw:
|
|
logging.warning(f"FloodWait! Sleeping {fw.x} seconds")
|
|
time.sleep(fw.x)
|
|
except RPCError as generic_error:
|
|
logging.error(f"An exception occurred: {generic_error}")
|
|
return False
|
|
|
|
|
|
def edit_message_media(update, *args, **kwargs):
|
|
"""Edits a message media in a way that never triggers exceptions and logs errors"""
|
|
|
|
try:
|
|
return update.edit_message_media(*args, **kwargs)
|
|
except FloodWait as fw:
|
|
logging.warning(f"FloodWait! Sleeping {fw.x} seconds")
|
|
time.sleep(fw.x)
|
|
except RPCError as generic_error:
|
|
logging.error(f"An exception occurred: {generic_error}")
|
|
return False
|
|
|
|
|