Add new_invoice command

1 files changed, 5 insertions, 4 deletions

diff --git a/handlers/admin/new_announcement.py b/handlers/admin/new_announcement.py
index 79cf8d4..0920c47 100644
--- a/handlers/admin/new_announcement.py
+++ b/handlers/admin/new_announcement.py
@@ -1,6 +1,7 @@
 from datetime import UTC, datetime
 
 from aiogram import Bot, F, Router
+from aiogram.enums import ButtonStyle
 from aiogram.exceptions import TelegramAPIError
 from aiogram.filters import Command
 from aiogram.fsm.context import FSMContext
@@ -17,7 +18,7 @@ from sqlalchemy import select
 from sqlalchemy.ext.asyncio import AsyncSession
 
 from libs.fsm import get_data, set_data
-from libs.msg import publish
+from libs.msg import publish_announcement
 from models import Announcement, RichText, User
 
 router = Router(name="new_announcement")
@@ -42,8 +43,8 @@ async def new_announcement_command(msg: Message, state: FSMContext) -> None:
         reply_markup=ReplyKeyboardMarkup(
             keyboard=[
                 [
-                    KeyboardButton(text=SEND_BUTTON),
-                    KeyboardButton(text=CANCEL_BUTTON),
+                    KeyboardButton(text=SEND_BUTTON, style=ButtonStyle.SUCCESS),
+                    KeyboardButton(text=CANCEL_BUTTON, style=ButtonStyle.DANGER),
                 ]
             ],
             resize_keyboard=True,
@@ -69,7 +70,7 @@ async def announcement_send(
     status_template = "Публикация анонса...\nОпубликовано: {}"
     status_msg = await msg.answer(status_template.format(0))
 
-    async for n in publish(bot, users, data.rich_text):
+    async for n in publish_announcement(bot, users, data.rich_text):
         try:
             await status_msg.edit_text(status_template.format(n))
         except TelegramAPIError: