mirror of
https://github.com/python-telegram-bot/python-telegram-bot.git
synced 2024-12-23 06:50:29 +01:00
36d49ea9cd
* Render-fixes for BP * docs: fix simple typo, submition -> submission (#2260) There is a small typo in tests/test_bot.py. Should read `submission` rather than `submition`. * Type on rawapibot.py docstring * typo * Typo: Filters.document(s) * Typo fix * Doc fix for messageentity (#2311) * Add New Shortcuts to Chat (#2291) * Add shortcuts * Add a note * Add run_async Parameter to ConversationHandler (#2292) * Add run_async parameter * Update docstring * Update test to explicitly specify parameter * Fix test job queue * Add version added tag to docs * Update docstring Co-authored-by: Poolitzer <25934244+Poolitzer@users.noreply.github.com> * Doc nitpicking Co-authored-by: Poolitzer <25934244+Poolitzer@users.noreply.github.com> Co-authored-by: Hinrich Mahler <hinrich.mahler@freenet.de> * Fix rendering in messageentity Co-authored-by: Bibo-Joshi <hinrich.mahler@freenet.de> Co-authored-by: zeshuaro <joshuaystang@gmail.com> Co-authored-by: Poolitzer <25934244+Poolitzer@users.noreply.github.com> * fix: type hints for TelegramError changed :class:`telegram.TelegramError` to :class:`telegram.error.TelegramError` * fix: the error can be more then just a Telegram error * Doc fix for inlinekeyboardbutton.py added missing colon which broke rendering * fix: remove context argument and doc remark look at us already being in post 12 * use rtd badge * filters doc fixes * fix some rendering * Doc & Rendering fixes for helpers.py Co-authored-by: Tim Gates <tim.gates@iress.com> Co-authored-by: Harshil <37377066+harshil21@users.noreply.github.com> Co-authored-by: zeshuaro <joshuaystang@gmail.com> Co-authored-by: Poolitzer <25934244+Poolitzer@users.noreply.github.com> Co-authored-by: Harshil <ilovebhagwan@gmail.com>
180 lines
5.3 KiB
Python
180 lines
5.3 KiB
Python
#!/usr/bin/env python
|
|
# pylint: disable=W0613, C0116, C0103
|
|
# type: ignore[union-attr]
|
|
# This program is dedicated to the public domain under the CC0 license.
|
|
|
|
"""
|
|
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, Update
|
|
from telegram.ext import (
|
|
Updater,
|
|
CommandHandler,
|
|
MessageHandler,
|
|
Filters,
|
|
ConversationHandler,
|
|
PicklePersistence,
|
|
CallbackContext,
|
|
)
|
|
|
|
|
|
# Enable logging
|
|
logging.basicConfig(
|
|
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO
|
|
)
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
CHOOSING, TYPING_REPLY, TYPING_CHOICE = range(3)
|
|
|
|
reply_keyboard = [
|
|
['Age', 'Favourite colour'],
|
|
['Number of siblings', 'Something else...'],
|
|
['Done'],
|
|
]
|
|
markup = ReplyKeyboardMarkup(reply_keyboard, one_time_keyboard=True)
|
|
|
|
|
|
def facts_to_str(user_data):
|
|
facts = list()
|
|
|
|
for key, value in user_data.items():
|
|
facts.append(f'{key} - {value}')
|
|
|
|
return "\n".join(facts).join(['\n', '\n'])
|
|
|
|
|
|
def start(update: Update, context: CallbackContext) -> None:
|
|
reply_text = "Hi! My name is Doctor Botter."
|
|
if context.user_data:
|
|
reply_text += (
|
|
f" You already told me your {', '.join(context.user_data.keys())}. Why don't you "
|
|
f"tell me something more about yourself? Or change anything I already know."
|
|
)
|
|
else:
|
|
reply_text += (
|
|
" I will hold a more complex conversation with you. Why don't you tell me "
|
|
"something about yourself?"
|
|
)
|
|
update.message.reply_text(reply_text, reply_markup=markup)
|
|
|
|
return CHOOSING
|
|
|
|
|
|
def regular_choice(update: Update, context: CallbackContext) -> None:
|
|
text = update.message.text.lower()
|
|
context.user_data['choice'] = text
|
|
if context.user_data.get(text):
|
|
reply_text = (
|
|
f'Your {text}, I already know the following about that: {context.user_data[text]}'
|
|
)
|
|
else:
|
|
reply_text = f'Your {text}? Yes, I would love to hear about that!'
|
|
update.message.reply_text(reply_text)
|
|
|
|
return TYPING_REPLY
|
|
|
|
|
|
def custom_choice(update: Update, context: CallbackContext) -> None:
|
|
update.message.reply_text(
|
|
'Alright, please send me the category first, ' 'for example "Most impressive skill"'
|
|
)
|
|
|
|
return TYPING_CHOICE
|
|
|
|
|
|
def received_information(update: Update, context: CallbackContext) -> None:
|
|
text = update.message.text
|
|
category = context.user_data['choice']
|
|
context.user_data[category] = text.lower()
|
|
del context.user_data['choice']
|
|
|
|
update.message.reply_text(
|
|
"Neat! Just so you know, this is what you already told me:"
|
|
f"{facts_to_str(context.user_data)}"
|
|
"You can tell me more, or change your opinion on "
|
|
"something.",
|
|
reply_markup=markup,
|
|
)
|
|
|
|
return CHOOSING
|
|
|
|
|
|
def show_data(update: Update, context: CallbackContext) -> None:
|
|
update.message.reply_text(
|
|
f"This is what you already told me: {facts_to_str(context.user_data)}"
|
|
)
|
|
|
|
|
|
def done(update: Update, context: CallbackContext) -> None:
|
|
if 'choice' in context.user_data:
|
|
del context.user_data['choice']
|
|
|
|
update.message.reply_text(
|
|
"I learned these facts about you:" f"{facts_to_str(context.user_data)}" "Until next time!"
|
|
)
|
|
return ConversationHandler.END
|
|
|
|
|
|
def main():
|
|
# Create the Updater and pass it your bot's token.
|
|
pp = PicklePersistence(filename='conversationbot')
|
|
updater = Updater("TOKEN", persistence=pp)
|
|
|
|
# Get the dispatcher to register handlers
|
|
dp = updater.dispatcher
|
|
|
|
# Add conversation handler with the states CHOOSING, TYPING_CHOICE and TYPING_REPLY
|
|
conv_handler = ConversationHandler(
|
|
entry_points=[CommandHandler('start', start)],
|
|
states={
|
|
CHOOSING: [
|
|
MessageHandler(
|
|
Filters.regex('^(Age|Favourite colour|Number of siblings)$'), regular_choice
|
|
),
|
|
MessageHandler(Filters.regex('^Something else...$'), custom_choice),
|
|
],
|
|
TYPING_CHOICE: [
|
|
MessageHandler(
|
|
Filters.text & ~(Filters.command | Filters.regex('^Done$')), regular_choice
|
|
)
|
|
],
|
|
TYPING_REPLY: [
|
|
MessageHandler(
|
|
Filters.text & ~(Filters.command | Filters.regex('^Done$')),
|
|
received_information,
|
|
)
|
|
],
|
|
},
|
|
fallbacks=[MessageHandler(Filters.regex('^Done$'), done)],
|
|
name="my_conversation",
|
|
persistent=True,
|
|
)
|
|
|
|
dp.add_handler(conv_handler)
|
|
|
|
show_data_handler = CommandHandler('show_data', show_data)
|
|
dp.add_handler(show_data_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()
|