Add add_user command

1 files changed, 3 insertions, 3 deletions

diff --git a/handlers/admin/new_invoice.py b/handlers/admin/new_invoice.py
index 7e1a64d..0427d90 100644
--- a/handlers/admin/new_invoice.py
+++ b/handlers/admin/new_invoice.py
@@ -1,7 +1,7 @@
 from datetime import UTC, datetime
 
 from aiogram import Bot, F, Router
-from aiogram.enums import ButtonStyle, ContentType
+from aiogram.enums import ButtonStyle
 from aiogram.exceptions import TelegramAPIError
 from aiogram.filters import Command
 from aiogram.fsm.context import FSMContext
@@ -69,7 +69,7 @@ async def invoice_send(
     status_template = "Рассылка счёта...\nОтправлено: {}"
     status_msg = await msg.answer(status_template.format(0))
 
-    invoice = Invoice(message=data, datetime=datetime.now(UTC))
+    invoice = Invoice(message=data.rich_text, datetime=datetime.now(UTC))
     session.add(invoice)
     await session.flush()
 
@@ -95,7 +95,7 @@ async def invoice_cancel(msg: Message, state: FSMContext) -> None:
 
 @router.message(NewInvoiceStates.message)
 async def invoice_message(msg: Message, bot: Bot, state: FSMContext) -> None:
-    if msg.content_type != ContentType.TEXT or msg.text is None:
+    if msg.text is None:
         await msg.answer(
             "Неверный тип сообщения.\n"
             "Бот поддерживает отправку только текстовых счетов."