mirror of
https://github.com/python-telegram-bot/python-telegram-bot.git
synced 2025-01-07 11:05:19 +01:00
83 lines
2.9 KiB
Python
83 lines
2.9 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
# This program is dedicated to the public domain under the CC0 license.
|
|
|
|
"""
|
|
This is a very simple example on how one could implement a custom error handler
|
|
"""
|
|
import html
|
|
import logging
|
|
import traceback
|
|
|
|
from telegram import Update, ParseMode
|
|
from telegram.ext import Updater, CallbackContext, CommandHandler
|
|
|
|
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
|
level=logging.INFO)
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# this can be your own ID, or one for a developer group/channel
|
|
DEVELOPER_CHAT_ID = 208589966
|
|
|
|
|
|
def error_handler(update: Update, context: CallbackContext):
|
|
"""Log the error and send a telegram message to notify the developer."""
|
|
# Log the error before we do anything else, so we can see it even if something breaks.
|
|
logger.error(msg="Exception while handling an update:", exc_info=context.error)
|
|
|
|
# traceback.format_exception returns the usual python message about an exception, but as a
|
|
# list of strings rather than a single string, so we have to join them together.
|
|
tb_list = traceback.format_exception(None, context.error, context.error.__traceback__)
|
|
tb = ''.join(tb_list)
|
|
|
|
# Build the message with some markup and additional information about what happened.
|
|
# You might need to add some logic to deal with messages longer than the 4096 character limit.
|
|
message = (
|
|
'An exception was raised while handling an update\n'
|
|
'<pre>update = {}</pre>\n\n'
|
|
'<pre>context.chat_data = {}</pre>\n\n'
|
|
'<pre>context.user_data = {}</pre>\n\n'
|
|
'<pre>{}</pre>'
|
|
).format(
|
|
html.escape(update.to_json()),
|
|
html.escape(str(context.chat_data)),
|
|
html.escape(str(context.user_data)),
|
|
html.escape(tb)
|
|
)
|
|
|
|
# Finally, send the message
|
|
context.bot.send_message(chat_id=DEVELOPER_CHAT_ID, text=message, parse_mode=ParseMode.HTML)
|
|
|
|
|
|
def bad_command(udpate: Update, context: CallbackContext):
|
|
"""Raise an error to trigger the error handler."""
|
|
context.bot.wrong_method_name()
|
|
|
|
|
|
def main():
|
|
# Create the Updater and pass it your bot's token.
|
|
# Make sure to set use_context=True to use the new context based callbacks
|
|
# Post version 12 this will no longer be necessary
|
|
updater = Updater("TOKEN", use_context=True)
|
|
|
|
# Get the dispatcher to register handlers
|
|
dp = updater.dispatcher
|
|
|
|
# Register the command...
|
|
dp.add_handler(CommandHandler('bad_command', bad_command))
|
|
|
|
# ...and the error handler
|
|
dp.add_error_handler(error_handler)
|
|
|
|
# Start the Bot
|
|
updater.start_polling()
|
|
|
|
# Run the bot until you press Ctrl-C or the process receives SIGINT,
|
|
# SIGTERM or SIGABRT. This should be used most of the time, since
|
|
# start_polling() is non-blocking and will stop the bot gracefully.
|
|
updater.idle()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|