mirror of
https://github.com/python-telegram-bot/python-telegram-bot.git
synced 2024-10-24 01:46:22 +02:00
247577b2e2
See https://github.com/python-telegram-bot/python-telegram-bot/wiki/Transition-guide-to-Version-11.0 under Context based callbacks and Filters in handlers for a good guide on the changes in this commit. * Change handlers so context is supported * Attempt to make parameter "guessing" work on py < 3.5 * Document use_context in all handlers * Add Context to docs * Minor fixes to context handling * Add tests for context stuff * Allow the signature check to work on py<3.5 with methods * Fix order of operations * Address most issues raised in CR * Make CommandHandler no longer support filter lists * Fix indent (pycharm can be an arse sometimes) * Improve readability in conversationhandler * Make context have Match instead of groups & groupdict * Remove filter list support from messagehandler too * Small fix to StringCommandHandler * More small fixes to handlers * Amend CHANGES * Fix tests and fix bugs raised by tests * Don't allow users to ignore errors without messing with the warning filters themselves * Ignore our own deprecation warnings when testing * Skipping deprecationwarning test on py2 * Forgot some changes * Handler: Improved documentation and text of deprecation warnings * HandlerContext: Keep only dispatcher and use properties; improved doc * Complete fixing the documentation. - Fixes per Eldinnie's comments. - Fixes per warnings when running sphinx. * Some small doc fixes (interlinks and optionals) * Change add_error_handler to use HandlerContext too * More context based changes Context Based Handlers -> Context Based Callbacks No longer use_context args on every single Handler Instead set dispatcher/updater .use_context=True to use Works with - Handler callbacks - Error handler callbacks - Job callbacks Change examples to context based callbacks so new users are not confused Rename and move the context object from Handlers.HandlerContext to CallbackContext, since it doesn't only apply to handlers anymore. Fix tests by adding a new fixture `cpd` which is a dispatcher with use_context=True * Forgot about conversationhandler * Forgot jobqueue * Add tests for callbackcontext & for context based callback job * Fix as per review :)
162 lines
5.2 KiB
Python
162 lines
5.2 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
#
|
|
# Simple Bot to reply to Telegram messages
|
|
# This program is dedicated to the public domain under the CC0 license.
|
|
"""
|
|
This Bot uses the Updater class to handle the bot.
|
|
|
|
First, a few callback 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:
|
|
Example of a bot-user conversation using ConversationHandler.
|
|
Send /start to initiate the conversation.
|
|
Press Ctrl-C on the command line or send a signal to the process to stop the
|
|
bot.
|
|
"""
|
|
|
|
import logging
|
|
|
|
from telegram import (ReplyKeyboardMarkup, ReplyKeyboardRemove)
|
|
from telegram.ext import (Updater, CommandHandler, MessageHandler, Filters, RegexHandler,
|
|
ConversationHandler)
|
|
|
|
# Enable logging
|
|
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
|
level=logging.INFO)
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
GENDER, PHOTO, LOCATION, BIO = range(4)
|
|
|
|
|
|
def start(update, context):
|
|
reply_keyboard = [['Boy', 'Girl', 'Other']]
|
|
|
|
update.message.reply_text(
|
|
'Hi! My name is Professor Bot. I will hold a conversation with you. '
|
|
'Send /cancel to stop talking to me.\n\n'
|
|
'Are you a boy or a girl?',
|
|
reply_markup=ReplyKeyboardMarkup(reply_keyboard, one_time_keyboard=True))
|
|
|
|
return GENDER
|
|
|
|
|
|
def gender(update, context):
|
|
user = update.message.from_user
|
|
logger.info("Gender of %s: %s", user.first_name, update.message.text)
|
|
update.message.reply_text('I see! Please send me a photo of yourself, '
|
|
'so I know what you look like, or send /skip if you don\'t want to.',
|
|
reply_markup=ReplyKeyboardRemove())
|
|
|
|
return PHOTO
|
|
|
|
|
|
def photo(update, context):
|
|
user = update.message.from_user
|
|
photo_file = update.message.photo[-1].get_file()
|
|
photo_file.download('user_photo.jpg')
|
|
logger.info("Photo of %s: %s", user.first_name, 'user_photo.jpg')
|
|
update.message.reply_text('Gorgeous! Now, send me your location please, '
|
|
'or send /skip if you don\'t want to.')
|
|
|
|
return LOCATION
|
|
|
|
|
|
def skip_photo(update, context):
|
|
user = update.message.from_user
|
|
logger.info("User %s did not send a photo.", user.first_name)
|
|
update.message.reply_text('I bet you look great! Now, send me your location please, '
|
|
'or send /skip.')
|
|
|
|
return LOCATION
|
|
|
|
|
|
def location(update, context):
|
|
user = update.message.from_user
|
|
user_location = update.message.location
|
|
logger.info("Location of %s: %f / %f", user.first_name, user_location.latitude,
|
|
user_location.longitude)
|
|
update.message.reply_text('Maybe I can visit you sometime! '
|
|
'At last, tell me something about yourself.')
|
|
|
|
return BIO
|
|
|
|
|
|
def skip_location(update, context):
|
|
user = update.message.from_user
|
|
logger.info("User %s did not send a location.", user.first_name)
|
|
update.message.reply_text('You seem a bit paranoid! '
|
|
'At last, tell me something about yourself.')
|
|
|
|
return BIO
|
|
|
|
|
|
def bio(update, context):
|
|
user = update.message.from_user
|
|
logger.info("Bio of %s: %s", user.first_name, update.message.text)
|
|
update.message.reply_text('Thank you! I hope we can talk again some day.')
|
|
|
|
return ConversationHandler.END
|
|
|
|
|
|
def cancel(update, context):
|
|
user = update.message.from_user
|
|
logger.info("User %s canceled the conversation.", user.first_name)
|
|
update.message.reply_text('Bye! I hope we can talk again some day.',
|
|
reply_markup=ReplyKeyboardRemove())
|
|
|
|
return ConversationHandler.END
|
|
|
|
|
|
def error(update, context):
|
|
"""Log Errors caused by Updates."""
|
|
logger.warning('Update "%s" caused error "%s"', update, context.error)
|
|
|
|
|
|
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
|
|
|
|
# Add conversation handler with the states GENDER, PHOTO, LOCATION and BIO
|
|
conv_handler = ConversationHandler(
|
|
entry_points=[CommandHandler('start', start)],
|
|
|
|
states={
|
|
GENDER: [RegexHandler('^(Boy|Girl|Other)$', gender)],
|
|
|
|
PHOTO: [MessageHandler(Filters.photo, photo),
|
|
CommandHandler('skip', skip_photo)],
|
|
|
|
LOCATION: [MessageHandler(Filters.location, location),
|
|
CommandHandler('skip', skip_location)],
|
|
|
|
BIO: [MessageHandler(Filters.text, bio)]
|
|
},
|
|
|
|
fallbacks=[CommandHandler('cancel', cancel)]
|
|
)
|
|
|
|
dp.add_handler(conv_handler)
|
|
|
|
# log all errors
|
|
dp.add_error_handler(error)
|
|
|
|
# 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()
|