From 039abb570390804465d64b11aac817bd5753ebbe Mon Sep 17 00:00:00 2001
From: user <unexpected@gmail.com>
Date: Fri, 5 Apr 2024 08:24:03 +0300
Subject: [PATCH] dev@locazia: fixes

---
 app/bot/middleware.py                    |  2 +-
 app/core/models/_telegram/wrapped_bot.py | 21 +++++++++++++++------
 app/core/models/messages.py              |  6 ++++--
 3 files changed, 20 insertions(+), 9 deletions(-)

diff --git a/app/bot/middleware.py b/app/bot/middleware.py
index a9add20..5ecfa5e 100644
--- a/app/bot/middleware.py
+++ b/app/bot/middleware.py
@@ -67,7 +67,7 @@ class UserDataMiddleware(BaseMiddleware):
             data['chat_wrap'] = Wrapped_CBotChat(data['bot'], chat_id=user_id, db_session=session, user=user)
             data['memory'] = data['dispatcher']._s_memory
 
-            if isinstance(update_body, types.Message):
+            if getattr(update_body, 'text', None):
                 message_type = 'common'
                 if update_body.text.startswith('/start'):
                     message_type = 'start_command'
diff --git a/app/core/models/_telegram/wrapped_bot.py b/app/core/models/_telegram/wrapped_bot.py
index 9434ea3..9fabfaf 100644
--- a/app/core/models/_telegram/wrapped_bot.py
+++ b/app/core/models/_telegram/wrapped_bot.py
@@ -112,17 +112,26 @@ class Wrapped_CBotChat(T, PlayerTemplates):
 
     async def delete_message(self, message_id):
         assert self._chat_id, "No chat_id"
+        message_id = int(message_id)
         try:
             make_log(self, f"Delete message {self._chat_id}/{message_id}", level='debug')
-            return await self._bot.delete_message(
+            if (await self._bot.delete_message(
                 self._chat_id,
                 message_id
-            )
+            )):
+                if self.db_session:
+                    known_message = self.db_session.query(KnownTelegramMessage).filter(
+                        KnownTelegramMessage.chat_id == self._chat_id,
+                        KnownTelegramMessage.message_id == message_id
+                    ).first()
+                    if known_message:
+                        known_message.deleted = True
+                        self.db_session.commit()
         except Exception as e:
             make_log(self, f"Error deleting message {self._chat_id}/{message_id}. Error: {e}", level='warning')
             return None
 
-    async def send_photo(self, file_id, message_type='common', message_meta={}, **kwargs):
+    async def send_photo(self, file_id, message_type='common', message_meta={}, content_id=None, **kwargs):
         assert self._chat_id, "No chat_id"
         try:
             make_log(self, f"Send photo to {self._chat_id}. File: {file_id}", level='debug')
@@ -136,7 +145,7 @@ class Wrapped_CBotChat(T, PlayerTemplates):
             make_log(self, f"Error sending photo to {self._chat_id}. Error: {e}", level='warning')
             return None
 
-    async def send_document(self, file_id, message_type='common', message_meta={}, **kwargs):
+    async def send_document(self, file_id, message_type='common', message_meta={}, content_id=None, **kwargs):
         assert self._chat_id, "No chat_id"
         try:
             make_log(self, f"Send document to {self._chat_id}. File: {file_id}", level='debug')
@@ -150,7 +159,7 @@ class Wrapped_CBotChat(T, PlayerTemplates):
             make_log(self, f"Error sending document to {self._chat_id}. Error: {e}", level='warning')
             return None
 
-    async def send_video(self, file_id, message_type='common', message_meta={}, **kwargs):
+    async def send_video(self, file_id, message_type='common', message_meta={}, content_id=None, **kwargs):
         assert self._chat_id, "No chat_id"
         try:
             make_log(self, f"Send video to {self._chat_id}. File: {file_id}", level='debug')
@@ -164,7 +173,7 @@ class Wrapped_CBotChat(T, PlayerTemplates):
             make_log(self, f"Error sending video to {self._chat_id}. Error: {e}", level='warning')
             return None
 
-    async def send_audio(self, file_id, message_type='common', message_meta={}, **kwargs):
+    async def send_audio(self, file_id, message_type='common', message_meta={}, content_id=None, **kwargs):
         assert self._chat_id, "No chat_id"
         try:
             make_log(self, f"Send audio to {self._chat_id}. File: {file_id}", level='debug')
diff --git a/app/core/models/messages.py b/app/core/models/messages.py
index 3cba9c9..edfd5dd 100644
--- a/app/core/models/messages.py
+++ b/app/core/models/messages.py
@@ -1,6 +1,6 @@
 from base58 import b58decode
-from sqlalchemy import Column, Integer, String, DateTime, JSON, BigInteger, Boolean
-
+from sqlalchemy import Column, Integer, String, DateTime, JSON, BigInteger, Boolean, ForeignKey
+from sqlalchemy.orm import relationship
 from .base import AlchemyBase
 
 
@@ -20,3 +20,5 @@ class KnownTelegramMessage(AlchemyBase):
     created = Column(DateTime, nullable=False, default=0)
     deleted = Column(Boolean, nullable=True, default=False)
     meta = Column(JSON, nullable=False, default={})
+    content_id = Column(Integer, ForeignKey('node_storage.id'), nullable=True)
+