chore: Added message deletions and small code refactoring

app/utils/states.py

@@ -1,9 +1,12 @@
 __all__ = ("RegistrationForm",)
 
+from aiogram.exceptions import TelegramBadRequest
 from aiogram.fsm.state import State, StatesGroup
+from aiogram.fsm.context import FSMContext
 
 
 class RegistrationForm(StatesGroup):
+    previous_message = State()
     username = State()
     age = State()
     bio = State()
@@ -12,6 +15,34 @@ class RegistrationForm(StatesGroup):
 
 
 class UserAltering(StatesGroup):
+    successfully = State()
     message_id = State()
+    input_message = State()
+    previous_message = State()
     column = State()
     value = State()
+
+
+async def delete_message_from_state(state: FSMContext) -> None:
+    data = await state.get_data()
+
+    if "previous_message" in data and data["previous_message"] is not None:
+        try:
+            await data["previous_message"].delete()
+        except TelegramBadRequest:
+            pass
+
+        await state.update_data(previous_message=None)
+
+    if (
+        "input_message" in data
+        and data["input_message"] is not None
+        and "successfully" in data
+        and data["successfully"]
+    ):
+        try:
+            await data["input_message"].delete()
+        except TelegramBadRequest:
+            pass
+
+        await state.update_data(info_message=None)