From c8a43135275f862781908493e4b4f1bad6b5fd9a Mon Sep 17 00:00:00 2001 From: HexRabbit Date: Sat, 19 Aug 2017 22:56:00 +0800 Subject: [PATCH] Add character filter --- telebot.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/telebot.py b/telebot.py index 6519232..eac53bd 100644 --- a/telebot.py +++ b/telebot.py @@ -1,12 +1,16 @@ -from telegram.ext import Updater, CommandHandler, MessageHandler, Filters +from telegram.ext import Updater, CommandHandler, MessageHandler, Filters, BaseFilter from telegram import ReplyKeyboardMarkup import requests +class charFilter(BaseFilter): + def filter(self, message): + return len(message.text) == 1 and message.text in ['1','2','3','4'] entry = {} keyboard = [['0', '1'], ['2', '3']] -url = '' -updater = Updater(token='') +url = '' +updater = Updater(token='') dispatcher = updater.dispatcher +charfilter = charFilter() def generate_problem(username): @@ -50,7 +54,7 @@ def reply_and_new_prob(bot, update): bot.send_message(chat_id=update.message.chat_id, text=generate_problem(username)) -dispatcher.add_handler(MessageHandler(Filters.text, reply_and_new_prob)) +dispatcher.add_handler(MessageHandler(charfilter, reply_and_new_prob)) dispatcher.add_handler(CommandHandler('start', start)) updater.start_polling()