FilesExpand file tree

 tighten-pre-commit

/

errorhandlerbot.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

99 lines (75 loc) · 3.44 KB

 tighten-pre-commit

/

errorhandlerbot.py

Copy path

Top

File metadata and controls

• Code
• Blame

99 lines (75 loc) · 3.44 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

#!/usr/bin/env python

# -*- coding: utf-8 -*-

# pylint: disable=W0613, C0116

# type: ignore[union-attr]

# 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, 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__)

# The token you got from @botfather when you created the bot

BOT_TOKEN = "TOKEN"

# 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

def error_handler(update: Update, context: CallbackContext) -> 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.

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(json.dumps(update.to_dict(), indent=2, ensure_ascii=False)),

html.escape(str(context.chat_data)),

html.escape(str(context.user_data)),

html.escape(tb_string),

)

# Finally, send the message

context.bot.send_message(chat_id=DEVELOPER_CHAT_ID, text=message, parse_mode=ParseMode.HTML)

def bad_command(update: Update, context: CallbackContext) -> None:

"""Raise an error to trigger the error handler."""

context.bot.wrong_method_name()

def start(update: Update, context: CallbackContext) -> None:

update.effective_message.reply_html(

'Use /bad_command to cause an error.\n'

'Your chat id is <code>{}</code>.'.format(update.effective_chat.id)

)

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(BOT_TOKEN, use_context=True)

# Get the dispatcher to register handlers

dispatcher = updater.dispatcher

# Register the commands...

dispatcher.add_handler(CommandHandler('start', start))

dispatcher.add_handler(CommandHandler('bad_command', bad_command))

# ...and the error handler

dispatcher.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()