python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2025-12-12 18:19:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Refactor examples/callback_data_factory_simple.py

Browse source
This commit is contained in:
Suren Khorenyan 2019-07-20 23:37:52 +03:00
parent 846f83f117
commit cb7ec3edb8
1 changed files with 20 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

42
examples/callback_data_factory_simple.py
View file

@ -27,42 +27,40 @@ likes = {} # user_id: amount_of_likes
def get_keyboard(): def get_keyboard():
return types.InlineKeyboardMarkup().row( return types.InlineKeyboardMarkup().row(
types.InlineKeyboardButton('👍', callback_data=vote_cb.new(action='up')), types.InlineKeyboardButton('👍', callback_data=vote_cb.new(action='up')),
types.InlineKeyboardButton('👎', callback_data=vote_cb.new(action='down'))) types.InlineKeyboardButton('👎', callback_data=vote_cb.new(action='down')),
)
@dp.message_handler(commands=['start']) @dp.message_handler(commands=['start'])
async def cmd_start(message: types.Message): async def cmd_start(message: types.Message):
amount_of_likes = likes.get(message.from_user.id, 0) # get value if key exists else set to 0 amount_of_likes = likes.get(message.from_user.id, 0) # get value if key exists else set to 0
await message.reply(f'Vote! Now you have {amount_of_likes} votes.', reply_markup=get_keyboard()) await message.reply(f'Vote! You have {amount_of_likes} votes now.', reply_markup=get_keyboard())
@dp.callback_query_handler(vote_cb.filter(action='up')) @dp.callback_query_handler(vote_cb.filter(action=['up', 'down']))
async def vote_up_cb_handler(query: types.CallbackQuery, callback_data: dict): async def callback_vote_action(query: types.CallbackQuery, callback_data: dict):
logging.info(callback_data) # callback_data contains all info from callback data logging.info('Got this callback data: %r', callback_data) # callback_data contains all info from callback data
likes[query.from_user.id] = likes.get(query.from_user.id, 0) + 1 # update amount of likes in storage await query.answer() # don't forget to answer callback query as soon as possible
amount_of_likes = likes[query.from_user.id] callback_data_action = callback_data['action']
likes_count = likes.get(query.from_user.id, 0)
await bot.edit_message_text(f'You voted up! Now you have {amount_of_likes} votes.', if callback_data_action == 'up':
query.from_user.id, likes_count += 1
query.message.message_id, else:
reply_markup=get_keyboard()) likes_count -= 1
likes[query.from_user.id] = likes_count # update amount of likes in storage
@dp.callback_query_handler(vote_cb.filter(action='down')) await bot.edit_message_text(
async def vote_down_cb_handler(query: types.CallbackQuery, callback_data: dict): f'You voted {callback_data_action}! Now you have {likes_count} vote[s].',
logging.info(callback_data) # callback_data contains all info from callback data query.from_user.id,
likes[query.from_user.id] = likes.get(query.from_user.id, 0) - 1 # update amount of likes in storage query.message.message_id,
amount_of_likes = likes[query.from_user.id] reply_markup=get_keyboard(),
)
await bot.edit_message_text(f'You voted down! Now you have {amount_of_likes} votes.',
query.from_user.id,
query.message.message_id,
reply_markup=get_keyboard())
@dp.errors_handler(exception=MessageNotModified) # handle the cases when this exception raises @dp.errors_handler(exception=MessageNotModified) # handle the cases when this exception raises
async def message_not_modified_handler(update, error): async def message_not_modified_handler(update, error):
# pass
return True return True