Delete acknowledge text on upload

app_bot/bot/core/callbacks.py

@@ -39,16 +39,23 @@ class TelegramCallback:
                 self._log.debug('No urls to download, skipping message')
                 return
 
-        urls = self._url_parser.parse_urls(urls=urls, message=message, user=user)
-        await self._url_service.process_urls(urls)
-        await self._send_acknowledge_message(message=message, url_count=len(urls))
+        acknowledge_message = await self._send_acknowledge_message(
+            message=message, url_count=len(urls)
+        )
+        context = {
+            'message': message,
+            'user': user,
+            'acknowledge_message': acknowledge_message,
+        }
+        url_objects = self._url_parser.parse_urls(urls=urls, context=context)
+        await self._url_service.process_urls(url_objects)
 
     async def _send_acknowledge_message(
         self,
         message: Message,
         url_count: int,
-    ) -> None:
-        await message.reply(
+    ) -> Message:
+        return await message.reply(
             text=self._format_acknowledge_text(url_count),
             parse_mode=ParseMode.HTML,
             reply_to_message_id=message.id,

app_bot/bot/core/handlers/success.py

@@ -35,11 +35,17 @@ class SuccessDownloadHandler(AbstractDownloadHandler):
             self._cleanup()
 
     async def _handle(self) -> None:
-        coro_tasks = []
+        coro_tasks = [self._delete_acknowledge_message()]
         for media_object in self._body.media.get_media_objects():
             coro_tasks.append(self._handle_media_object(media_object))
         await asyncio.gather(*coro_tasks)
 
+    async def _delete_acknowledge_message(self) -> None:
+        await self._bot.delete_messages(
+            chat_id=self._body.from_chat_id,
+            message_ids=[self._body.context.acknowledge_message_id],
+        )
+
     async def _publish_error_message(self, err: Exception) -> None:
         err_payload = ErrorDownloadGeneralPayload(
             task_id=self._body.task_id,

app_bot/bot/core/service.py

@@ -24,6 +24,7 @@ class UrlService:
         payload = InbMediaPayload(
             url=url.url,
             message_id=url.message_id,
+            acknowledge_message_id=url.acknowledge_message_id,
             from_user_id=url.from_user_id,
             from_chat_id=url.from_chat_id,
             from_chat_type=url.from_chat_type,
@@ -42,7 +43,12 @@ class UrlParser:
         self._log = logging.getLogger(self.__class__.__name__)
 
     @staticmethod
-    def parse_urls(urls: list[str], message: Message, user: UserSchema) -> list[URL]:
+    def parse_urls(
+        urls: list[str], context: dict[str, Message | UserSchema]
+    ) -> list[URL]:
+        message: Message = context['message']
+        user: UserSchema = context['user']
+        acknowledge_message: Message = context['acknowledge_message']
         return [
             URL(
                 url=url,
@@ -50,6 +56,7 @@ class UrlParser:
                 from_chat_type=TelegramChatType(message.chat.type.value),
                 from_user_id=message.from_user.id,
                 message_id=message.id,
+                acknowledge_message_id=acknowledge_message.id,
                 save_to_storage=user.save_to_storage,
                 download_media_type=user.download_media_type,
             )

yt_shared/yt_shared/schemas/media.py

@@ -24,6 +24,7 @@ class InbMediaPayload(RealBaseModel):
     from_chat_type: TelegramChatType | None
     from_user_id: StrictInt | None
     message_id: StrictInt | None
+    acknowledge_message_id: StrictInt | None
     url: StrictStr
     source: TaskSource
     save_to_storage: StrictBool

yt_shared/yt_shared/schemas/url.py

@@ -10,5 +10,6 @@ class URL(RealBaseModel):
     from_chat_type: TelegramChatType
     from_user_id: StrictInt
     message_id: StrictInt
+    acknowledge_message_id: StrictInt
     save_to_storage: StrictBool
     download_media_type: DownMediaType