From 7b9422f627b493b13789683971b52c5fb09c4e53 Mon Sep 17 00:00:00 2001
From: birdi <egor.birdi7@gmail.com>
Date: Wed, 10 Jul 2019 21:08:29 +0300
Subject: [PATCH] Add simple example of usage of callback data factory

---
 examples/callback_data_factory_simple.py | 70 ++++++++++++++++++++++++
 1 file changed, 70 insertions(+)
 create mode 100644 examples/callback_data_factory_simple.py

diff --git a/examples/callback_data_factory_simple.py b/examples/callback_data_factory_simple.py
new file mode 100644
index 00000000..2c6a8358
--- /dev/null
+++ b/examples/callback_data_factory_simple.py
@@ -0,0 +1,70 @@
+"""
+This is a simple example of usage of CallbackData factory
+For more comprehensive example see callback_data_factory.py
+"""
+
+import logging
+
+from aiogram import Bot, Dispatcher, executor, types
+from aiogram.contrib.middlewares.logging import LoggingMiddleware
+from aiogram.utils.callback_data import CallbackData
+from aiogram.utils.exceptions import MessageNotModified
+
+logging.basicConfig(level=logging.INFO)
+
+API_TOKEN = 'BOT_TOKEN_HERE'
+
+
+bot = Bot(token=API_TOKEN)
+
+dp = Dispatcher(bot)
+dp.middleware.setup(LoggingMiddleware())
+
+vote_cb = CallbackData('vote', 'action')  # vote:<action>
+likes = {}  # user_id: amount_of_likes
+
+
+def get_keyboard():
+    return types.InlineKeyboardMarkup().row(
+        types.InlineKeyboardButton('👍', callback_data=vote_cb.new(action='up')),
+        types.InlineKeyboardButton('👎', callback_data=vote_cb.new(action='down')))
+
+
+@dp.message_handler(commands=['start'])
+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
+    await message.reply(f'Vote! Now you have {amount_of_likes} votes.', reply_markup=get_keyboard())
+
+
+@dp.callback_query_handler(vote_cb.filter(action='up'))
+async def vote_up_cb_handler(query: types.CallbackQuery, callback_data: dict):
+    logging.info(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
+    amount_of_likes = likes[query.from_user.id]
+
+    await bot.edit_message_text(f'You voted up! Now you have {amount_of_likes} votes.',
+                                query.from_user.id,
+                                query.message.message_id,
+                                reply_markup=get_keyboard())
+
+
+@dp.callback_query_handler(vote_cb.filter(action='down'))
+async def vote_down_cb_handler(query: types.CallbackQuery, callback_data: dict):
+    logging.info(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
+    amount_of_likes = likes[query.from_user.id]
+
+    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
+async def message_not_modified_handler(update, error):
+    # pass
+    return True
+
+
+if __name__ == '__main__':
+    executor.start_polling(dp, skip_updates=True)