Add new_announcement command

1 files changed, 25 insertions, 0 deletions

diff --git a/libs/msg.py b/libs/msg.py
new file mode 100644
index 0000000..9bcc52a
--- /dev/null
+++ b/libs/msg.py
@@ -0,0 +1,25 @@
+import asyncio
+from typing import AsyncGenerator, Iterable
+
+from aiogram import Bot
+from aiogram.exceptions import TelegramAPIError, TelegramRetryAfter
+
+from models import RichText
+
+
+async def publish(
+    bot: Bot,
+    users: Iterable[int],
+    rich_text: RichText,
+) -> AsyncGenerator[int]:
+    for n, user_id in enumerate(users, start=1):
+        for _ in range(5):
+            try:
+                await rich_text.send(bot, user_id)
+                break
+            except TelegramRetryAfter as e:
+                await asyncio.sleep(e.retry_after + 1)
+            except TelegramAPIError:
+                await asyncio.sleep(5)
+
+        yield n