Add cancel option to all two-step actions

Author: coder8080

src/entities/constants.py

@@ -15,20 +15,21 @@
     "used-link": "Ссылка уже использована",
     "ok-link": "Количество сертификатов у вас: {%COUNT%}",
     "create-link": "➕ Создать новую ссылку",
-    "input-count": "Отправьте одно число: количество сертификатов",
+    "input-count": "Отправьте одно число: количество сертификатов или \
+/cancel, чтобы отменить",
     "link-created": "Вот ваша ссылка (нажмите, чтобы скопировать): \
 <code>{%LINK%}</code>",
     "unknown-message": "🥲 Я вас не понял\n/start - перезапустить бота",
     "create-cert": "📄 Выпустить новый сертификат",
     "input-name": "Напишите имя, на которое нужно выпустить сертификат. \
-Вы можете написать в столбик несколько имен",
+Вы можете написать в столбик несколько имен. /cancel, чтобы отменить",
     "not-enough-count": "Вам не выдано достаточно сертификатов, чтобы \
 выписать их на каждого из этих людей",
     "certs-ready": "Все сертификаты сгенерированы успешно",
     "error": "🥲 Произошла непредвиденная ошибка. Мы уже знаем о проблеме \
 и решаем ее",
     "check-cert": "❓ Проверить сертификат",
-    "input-cert": "Отправьте номер сертификата",
+    "input-cert": "Отправьте номер сертификата. /cancel, чтобы отменить",
     "no-cert": "Сертификата с таким номером не существует.",
     "used-cert": "Этот сертификат уже был использован.",
     "ok-cert": "✅ Сертификат был создан на имя {%NAME%} и еще не был \
@@ -41,5 +42,9 @@
     "check-count": "🧮 Проверить количество сертификатов",
     "admin-count": "Вы являетесь администратором и можете создавать \
 сколько угодно сертификатов",
-    "user-count": "У вас осталось сертификатов: {%COUNT%}"
+    "user-count": "У вас осталось сертификатов: {%COUNT%}",
+    "cancel": "❌ Отмена",
+    "create-link-cancelled": '👌 Создание ссылки отменено',
+    "create-cert-cancelled": "👌 Создание сертификата отменено",
+    "check-cert-cancelled": "👌 Проверка сертификата отменена"
 }

src/models/link.py

@@ -1,6 +1,6 @@
 from peewee import TextField, IntegerField, BooleanField
 
-from entities import BOT_NAME
+from entities.constants import BOT_NAME
 from .db import BaseModel, db
 from .utils import str_uuid
 

src/routers/check_cert.py

@@ -19,6 +19,13 @@ async def check_start(message: Message, state: FSMContext):
                          reply_markup=ReplyKeyboardRemove())
 
 
+@router.message(StateFilter(States.TypingCert), Command("cancel"))
+async def cancel(message: Message, state: FSMContext, user: User):
+    await state.set_state(None)
+    await message.answer(TEXT['check-cert-cancelled'],
+                         reply_markup=get_keyboard(user))
+
+
 @router.message(StateFilter(States.TypingCert), F.chat.id.in_(ADMINS))
 async def check_cert(message: Message, state: FSMContext, user: User):
     await state.set_state(None)

src/routers/create_cert.py

@@ -13,10 +13,19 @@
 @router.message(StateFilter(None), F.text == TEXT['create-cert'])
 async def open_create_cert(message: Message, state: FSMContext):
     await state.set_state(States.TypingName)
+
     await message.answer(TEXT['input-name'],
                          reply_markup=ReplyKeyboardRemove())
 
 
+@router.message(StateFilter(States.TypingName), Command("cancel"))
+async def cancel(message: Message, state: FSMContext, user: User):
+    await state.set_state(None)
+
+    await message.answer(TEXT['create-cert-cancelled'],
+                         reply_markup=get_keyboard(user))
+
+
 @router.message(StateFilter(States.TypingName))
 async def create_cert(message: Message, user: User, state: FSMContext):
     names = message.text.split("\n")

src/routers/link.py

@@ -1,8 +1,7 @@
 from aiogram import Router, F
 from aiogram.filters import Command, StateFilter
-from aiogram.types import Message
+from aiogram.types import Message, ReplyKeyboardRemove
 from aiogram.fsm.context import FSMContext
-from aiogram.types import ReplyKeyboardRemove
 
 from entities import ADMINS, TEXT, States, get_keyboard
 from models import User, Link
@@ -21,6 +20,13 @@ async def create_link(message: Message, state: FSMContext):
                          reply_markup=ReplyKeyboardRemove())
 
 
+@router.message(StateFilter(States.TypingNumber), Command("cancel"))
+async def cancel(message: Message, state: FSMContext, user: User):
+    await state.set_state(None)
+    await message.answer(TEXT['create-link-cancelled'],
+                         reply_markup=get_keyboard(user))
+
+
 @router.message(StateFilter(States.TypingNumber), F.chat.id.in_(ADMINS))
 async def get_number(message: Message, user: User, state: FSMContext):
     if not message.text.isnumeric():