FilesExpand file tree

 master

/

timerbot.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

110 lines (77 loc) · 3.21 KB

 master

/

timerbot.py

Copy path

Top

File metadata and controls

• Code
• Blame

110 lines (77 loc) · 3.21 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

100

101

102

103

104

105

106

107

108

109

110

#!/usr/bin/env python

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

"""Simple Bot to send timed Telegram messages.

# This program is dedicated to the public domain under the CC0 license.

This Bot uses the Updater class to handle the bot and the JobQueue to send

timed messages.

First, a few handler functions are defined. Then, those functions are passed to

the Dispatcher and registered at their respective places.

Then, the bot is started and runs until we press Ctrl-C on the command line.

Usage:

Basic Alarm Bot example, sends a message after a set time.

Press Ctrl-C on the command line or send a signal to the process to stop the

bot.

"""

from telegram.ext import Updater, CommandHandler

import logging

# Enable logging

logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',

level=logging.INFO)

logger = logging.getLogger(__name__)

# Define a few command handlers. These usually take the two arguments bot and

# update. Error handlers also receive the raised TelegramError object in error.

def start(bot, update):

update.message.reply_text('Hi! Use /set <seconds> to set a timer')

def alarm(bot, job):

"""Send the alarm message."""

bot.send_message(job.context, text='Beep!')

def set_timer(bot, update, args, job_queue, chat_data):

"""Add a job to the queue."""

chat_id = update.message.chat_id

try:

# args[0] should contain the time for the timer in seconds

due = int(args[0])

if due < 0:

update.message.reply_text('Sorry we can not go back to future!')

return

# Add job to queue

job = job_queue.run_once(alarm, due, context=chat_id)

chat_data['job'] = job

update.message.reply_text('Timer successfully set!')

except (IndexError, ValueError):

update.message.reply_text('Usage: /set <seconds>')

def unset(bot, update, chat_data):

"""Remove the job if the user changed their mind."""

if 'job' not in chat_data:

update.message.reply_text('You have no active timer')

return

job = chat_data['job']

job.schedule_removal()

del chat_data['job']

update.message.reply_text('Timer successfully unset!')

def error(bot, update, error):

"""Log Errors caused by Updates."""

logger.warning('Update "%s" caused error "%s"', update, error)

def main():

"""Run bot."""

updater = Updater("TOKEN")

# Get the dispatcher to register handlers

dp = updater.dispatcher

# on different commands - answer in Telegram

dp.add_handler(CommandHandler("start", start))

dp.add_handler(CommandHandler("help", start))

dp.add_handler(CommandHandler("set", set_timer,

pass_args=True,

pass_job_queue=True,

pass_chat_data=True))

dp.add_handler(CommandHandler("unset", unset, pass_chat_data=True))

# log all errors

dp.add_error_handler(error)

# Start the Bot

updater.start_polling()

# Block 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()