mirror of
https://github.com/python-telegram-bot/python-telegram-bot.git
synced 2024-11-22 23:27:49 +01:00
84 lines
3.2 KiB
Python
84 lines
3.2 KiB
Python
#!/usr/bin/env python
|
|
# pylint: disable=unused-argument
|
|
# 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 json
|
|
import logging
|
|
import traceback
|
|
|
|
from telegram import Update
|
|
from telegram.constants import ParseMode
|
|
from telegram.ext import Application, CallbackContext, CommandHandler
|
|
|
|
# Enable logging
|
|
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.
|
|
# You can use the /start command of this bot to see your chat id.
|
|
DEVELOPER_CHAT_ID = 123456789
|
|
|
|
|
|
async def error_handler(update: object, context: CallbackContext.DEFAULT_TYPE) -> None:
|
|
"""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_string = ''.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.
|
|
update_str = update.to_dict() if isinstance(update, Update) else str(update)
|
|
message = (
|
|
f'An exception was raised while handling an update\n'
|
|
f'<pre>update = {html.escape(json.dumps(update_str, indent=2, ensure_ascii=False))}'
|
|
'</pre>\n\n'
|
|
f'<pre>context.chat_data = {html.escape(str(context.chat_data))}</pre>\n\n'
|
|
f'<pre>context.user_data = {html.escape(str(context.user_data))}</pre>\n\n'
|
|
f'<pre>{html.escape(tb_string)}</pre>'
|
|
)
|
|
|
|
# Finally, send the message
|
|
await context.bot.send_message(
|
|
chat_id=DEVELOPER_CHAT_ID, text=message, parse_mode=ParseMode.HTML
|
|
)
|
|
|
|
|
|
async def bad_command(update: Update, context: CallbackContext.DEFAULT_TYPE) -> None:
|
|
"""Raise an error to trigger the error handler."""
|
|
await context.bot.wrong_method_name() # type: ignore[attr-defined]
|
|
|
|
|
|
async def start(update: Update, context: CallbackContext.DEFAULT_TYPE) -> None:
|
|
"""Displays info on how to trigger an error."""
|
|
await update.effective_message.reply_html(
|
|
'Use /bad_command to cause an error.\n'
|
|
f'Your chat id is <code>{update.effective_chat.id}</code>.'
|
|
)
|
|
|
|
|
|
def main() -> None:
|
|
"""Run the bot."""
|
|
# Create the Application and pass it your bot's token.
|
|
application = Application.builder().token("TOKEN").build()
|
|
|
|
# Register the commands...
|
|
application.add_handler(CommandHandler('start', start))
|
|
application.add_handler(CommandHandler('bad_command', bad_command))
|
|
|
|
# ...and the error handler
|
|
application.add_error_handler(error_handler)
|
|
|
|
# Run the bot until the user presses Ctrl-C
|
|
application.run_polling()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|