mirror of
https://github.com/python-telegram-bot/python-telegram-bot.git
synced 2024-11-26 00:56:22 +01:00
107 lines
3.1 KiB
Python
107 lines
3.1 KiB
Python
#!/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, Job
|
|
import logging
|
|
|
|
# Enable logging
|
|
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
|
level=logging.DEBUG)
|
|
|
|
logger = logging.getLogger(__name__)
|
|
job_queue = None
|
|
timers = dict()
|
|
|
|
|
|
# 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):
|
|
bot.sendMessage(update.message.chat_id, text='Hi! Use /set <seconds> to ' 'set a timer')
|
|
|
|
|
|
def set(bot, update, args):
|
|
"""Adds 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:
|
|
bot.sendMessage(chat_id, text='Sorry we can not go back to future!')
|
|
|
|
def alarm(bot, job):
|
|
"""Inner function to send the alarm message"""
|
|
bot.sendMessage(chat_id, text='Beep!')
|
|
|
|
# Add job to queue
|
|
job = Job(alarm, due, repeat=False)
|
|
timers[chat_id] = job
|
|
job_queue.put(job)
|
|
|
|
bot.sendMessage(chat_id, text='Timer successfully set!')
|
|
|
|
except (IndexError, ValueError):
|
|
bot.sendMessage(chat_id, text='Usage: /set <seconds>')
|
|
|
|
|
|
def unset(bot, update):
|
|
"""Removes the job if the user changed their mind"""
|
|
chat_id = update.message.chat_id
|
|
|
|
if chat_id not in timers:
|
|
bot.sendMessage(chat_id, text='You have no active timer')
|
|
return
|
|
|
|
job = timers[chat_id]
|
|
job.schedule_removal()
|
|
bot.sendMessage(chat_id, text='Timer successfully unset!')
|
|
|
|
|
|
def error(bot, update, error):
|
|
logger.warn('Update "%s" caused error "%s"' % (update, error))
|
|
|
|
|
|
def main():
|
|
global job_queue
|
|
|
|
updater = Updater("TOKEN")
|
|
job_queue = updater.job_queue
|
|
|
|
# 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, pass_args=True))
|
|
dp.add_handler(CommandHandler("unset", unset))
|
|
|
|
# log all errors
|
|
dp.add_error_handler(error)
|
|
|
|
# Start the Bot
|
|
updater.start_polling()
|
|
|
|
# Block until the you presses 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()
|