async def _start_polling(self):
executor = Executor(self._dispatcher, skip_updates=False)
executor._prepare_polling()
loop: asyncio.AbstractEventLoop = executor.loop
try:
await executor._startup_polling()
await self._dispatcher.start_polling()
except (KeyboardInterrupt, SystemExit):
loop.stop()
pass
finally:
await executor._shutdown_polling()
class TelegramBotService:
def __init__(
self,
redis_host,
redis_port,
redis_db,
redis_password,
config:
TelegramBotServiceConfig = default_telegram_bot_service_config,
loop: Optional[asyncio.AbstractEventLoop] = None):
self._config = config
self.loop = loop or asyncio.get_event_loop()
self.redis_host = redis_host
self.redis_port = int(redis_port)
self.redis_db = int(redis_db)
self.redis_password = redis_password
self._storage = RedisStorage(host=self.redis_host,
port=self.redis_port,
db=self.redis_db,
password=self.redis_password)
self._bot = Bot(token=self._config.token,
proxy=self._config.proxy,
loop=self.loop)
self._dispatcher = Dispatcher(self._bot,
loop=self.loop,
storage=self._storage)
self._executor = Executor(self._dispatcher,
skip_updates=True,
loop=self.loop)
async def run_bot_task(self):
logger.info('Bot polling started')
self._dispatcher.register_message_handler(self._bot_start,
commands=['start'])
self._dispatcher.register_message_handler(self._show_menu,
commands=['menu'],
state='*')
self._dispatcher.register_message_handler(self._registration_step_1,
state=Registration.step_1)
self._dispatcher.register_callback_query_handler(self._show_links,
text='links')
self._dispatcher.register_callback_query_handler(self._book,
text='book')
# self._dispatcher.register_callback_query_handler(self._edit_order, text='show_orders')
self._dispatcher.register_callback_query_handler(self._book_step_1,
state=Book.step_1)
self._dispatcher.register_callback_query_handler(self._book_step_2_1,
text='done',
state=Book.step_2)
self._dispatcher.register_callback_query_handler(self._book_step_2_2,
text='cancel',
state=Book.step_2)
self._executor._prepare_polling()
await self._executor._startup_polling()
self.loop.create_task(
self._dispatcher.start_polling(reset_webhook=True))
async def _bot_start(self, message: Message):
telegram_id = message.chat.id
check_admin = await Admin.query.where(
Admin.telegram_id == str(telegram_id)).gino.first()
if check_admin:
await message.answer(f'Здравствуй, {check_admin.name}')
await self._bot.send_message(
telegram_id,
'Введите команду /menu посмотреть список доступных функций')
return
check_user = await User.query.where(
User.telegram_id == str(telegram_id)).gino.first()
if check_user:
await message.answer(f'Здравствуйте, {check_user.name}')
await self._bot.send_message(
telegram_id,
'Введите команду /menu посмотреть список доступных функций')
if all([(not check_user), (not check_admin)]):
await message.answer(
'Здравствуйте, мы с Вами не знакомы. \n'
'Введите ваше ФИО, номер телефона, рост и вес.')
example_registration = ' '.join([
'Пример:', 'Иванов Иван Иванович', '+79020007126', '175', '80'
])
await self._bot.send_message(chat_id=telegram_id,
text=example_registration)
await Registration.step_1.set()
async def _show_menu(self, message: Message):
telegram_id = message.chat.id
state = self._dispatcher.current_state(user=telegram_id)
await state.reset_state()
check_admin = await Admin.query.where(
Admin.telegram_id == str(telegram_id)).gino.first()
if check_admin:
inline_menu = await get_kb_menu_for_admin()
await message.answer(
'Здравствуйте, админстратор. Выберите интересующий вас пункт из меню ниже:',
reply_markup=inline_menu)
check_customer = await User.query.where(
User.telegram_id == str(telegram_id)).gino.first()
if check_customer:
inline_menu = await get_kb_menu_for_customer()
await message.answer(
'Выберите интересующий вас пункт из меню ниже:',
reply_markup=inline_menu)
if all([(not check_customer), (not check_admin)]):
await message.answer(
'Здравствуйте, мы с Вами не знакомы. \n'
'Введите ваше ФИО, номер телефона, рост и вес.')
example_registration = ' '.join([
'Пример:', 'Иванов Иван Иванович', '+79020007126', '175', '80'
])
await self._bot.send_message(chat_id=telegram_id,
text=example_registration)
await Registration.step_1.set()
# async def _edit_order(self, callback_query: CallbackQuery):
# telegram_id = callback_query.id
#
# await callback_query.answer(cache_time=60)
# await self._bot.answer_callback_query(callback_query.id)
# await callback_query.message.edit_reply_markup(reply_markup=None)
#
# inline_menu = await get_kb_orders_menu()
# await self._bot.send_message(telegram_id, 'Выберите интересующую вас заявку:',
# reply_markup=inline_menu)
async def _registration_step_1(self, message: Message, state: FSMContext):
telegram_id = message.chat.id
message_user = message.text.split(' ')
if len(message_user) != 6:
await message.answer('Введите как показано в примере')
return
name_message_user = message_user[0] + ' ' + message_user[
1] + ' ' + message_user[2]
phone_number_message_user = message_user[3]
height_message_user = message_user[4]
weight_message_user = message_user[5]
new_user = await User.create(name=name_message_user,
phone_number=phone_number_message_user,
height=height_message_user,
weight=weight_message_user,
telegram_id=str(telegram_id))
result = f"""
Данные успешно сохранены!
Ваше ФИО: {new_user.name}
Ваш номер телефона: {new_user.phone_number}
Ваш вес: {new_user.weight} кг
Ваш рост: {new_user.height} см"""
await self._bot.send_message(telegram_id, result)
await self._bot.send_message(
telegram_id,
'Введите команду /menu посмотреть список доступных функций')
await state.finish()
async def _show_links(self, callback_query: CallbackQuery):
telegram_id = callback_query.from_user.id
await callback_query.answer(cache_time=60)
await self._bot.answer_callback_query(callback_query.id)
await callback_query.message.edit_reply_markup(reply_markup=None)
inline_kb = await get_kb_out_links()
await self._bot.send_message(telegram_id,
'Мы в соцсетях!',
reply_markup=inline_kb)
async def _book(self, callback_query: CallbackQuery):
telegram_id = callback_query.from_user.id
await callback_query.answer(cache_time=60)
await self._bot.answer_callback_query(callback_query.id)
await callback_query.message.edit_reply_markup(reply_markup=None)
inline_kb = await get_kb_items_to_book()
await self._bot.send_message(telegram_id,
'Выберите интересующий вас инвентарь:',
reply_markup=inline_kb)
await Book.step_1.set()
async def _book_step_1(self, callback_query: CallbackQuery):
call = callback_query.data
telegram_id = callback_query.from_user.id
await callback_query.answer(cache_time=60)
await self._bot.answer_callback_query(callback_query.id)
await callback_query.message.edit_reply_markup(reply_markup=None)
inline_kb = await get_kb_order()
item_data = await Item.query.where(Item.data == str(call)).gino.first()
item_name = item_data.name
item_price = item_data.price
await Order.create(telegram_id=str(telegram_id),
ordered_item=item_name,
status='in treatment')
await self._bot.send_message(
telegram_id, f'Вы выбрали: {item_name}.\n'
f'Стоимость бронирования этого инвентаря: {item_price}.\n'
f'Хотите подать заявку?',
reply_markup=inline_kb)
await Book.step_2.set()
async def _book_step_2_1(self, callback_query: CallbackQuery,
state: FSMContext):
telegram_id = callback_query.from_user.id
await callback_query.answer(cache_time=60)
await self._bot.answer_callback_query(callback_query.id)
await callback_query.message.edit_reply_markup(reply_markup=None)
right_order = await Order.query.where(
Order.telegram_id == str(telegram_id)).gino.first()
user_data = await User.query.where(User.telegram_id == str(telegram_id)
).gino.first()
await self._bot.send_message(telegram_id,
'Заявка подана. Ожидайте звонка!')
order_text = f"""
Поступила заявка. Информация о заказчике:
Имя: {user_data.name}.
Номер телефона: {user_data.phone_number}.
Заявка на следующий инвентарь: {right_order.ordered_item}.
Заказчик ждет вашего звонка!"""
await self._bot.send_message(227448700, order_text)
await state.finish()
async def _book_step_2_2(self, callback_query: CallbackQuery,
state: FSMContext):
telegram_id = callback_query.from_user.id
await callback_query.answer(cache_time=60)
await self._bot.answer_callback_query(callback_query.id)
await callback_query.message.edit_reply_markup(reply_markup=None)
await self._bot.send_message(telegram_id, 'Заявка сброшена.')
await self._bot.send_message(
telegram_id,
'Введите команду /menu посмотреть список доступных функций')
wrong_order = await Order.query.where(
Order.telegram_id == str(telegram_id)).gino.first()
wrong_order.delete()
await state.finish()