Астат ты не вознесешься

database/repository.py

@@ -2,7 +2,7 @@
 Repository для работы с банвордами через SQLAlchemy ORM.
 """
 from typing import Set, List, Optional
-from datetime import datetime, timedelta
+from datetime import datetime, timedelta, timezone
 
 from sqlalchemy import select, delete, func, and_
 
@@ -15,7 +15,8 @@ from .models import (
     Admin,
     Setting,
     SpamStat,
-    BanWordType
+    BanWordType,    
+    AutoComment
 )
 
 __all__ = ("BanWordsRepository",)
@@ -796,3 +797,252 @@ class BanWordsRepository:
                 log_type="DATABASE"
             )
             return {}
+        
+# === AUTO COMMENTS ===
+
+    async def get_auto_comment(self, channel_id: int) -> Optional['AutoComment']:
+        """
+        Получает настройки автокомментариев для канала.
+
+        Args:
+            channel_id: ID канала
+
+        Returns:
+            AutoComment или None
+        """
+        try:
+            async with self.db.get_session() as session:
+                result = await session.execute(
+                    select(AutoComment).where(AutoComment.channel_id == channel_id)
+                )
+                return result.scalar_one_or_none()
+        except Exception as e:
+            logger.error(
+                f"Ошибка получения автокомментария: {e}",
+                log_type="DATABASE"
+            )
+            return None
+
+    async def set_auto_comment(
+            self,
+            channel_id: int,
+            text: str,
+            button_text: str,
+            button_url: str,
+            photo_url: str,
+            updated_by: Optional[int] = None,
+            is_enabled: bool = True
+    ) -> bool:
+        """
+        Сохраняет или обновляет настройки автокомментариев.
+
+        Args:
+            channel_id: ID канала
+            text: Текст комментария
+            button_text: Текст кнопки
+            button_url: URL кнопки
+            photo_url: URL фото
+            updated_by: ID админа
+            is_enabled: Включены ли комментарии
+
+        Returns:
+            bool: True если успешно
+        """
+        try:
+            async with self.db.get_session() as session:
+                # Проверяем существование
+                result = await session.execute(
+                    select(AutoComment).where(AutoComment.channel_id == channel_id)
+                )
+                auto_comment = result.scalar_one_or_none()
+
+                if auto_comment:
+                    # Обновляем существующую
+                    auto_comment.text = text
+                    auto_comment.button_text = button_text
+                    auto_comment.button_url = button_url
+                    auto_comment.photo_url = photo_url
+                    auto_comment.is_enabled = is_enabled
+                    auto_comment.updated_by = updated_by
+                    auto_comment.updated_at = datetime.now(timezone.utc)
+                else:
+                    # Создаём новую
+                    auto_comment = AutoComment(
+                        channel_id=channel_id,
+                        text=text,
+                        button_text=button_text,
+                        button_url=button_url,
+                        photo_url=photo_url,
+                        is_enabled=is_enabled,
+                        updated_by=updated_by
+                    )
+                    session.add(auto_comment)
+
+                await session.commit()
+
+                logger.info(
+                    f"Автокомментарий для канала {channel_id} обновлён",
+                    log_type="DATABASE"
+                )
+                return True
+
+        except Exception as e:
+            logger.error(
+                f"Ошибка сохранения автокомментария: {e}",
+                log_type="DATABASE"
+            )
+            return False
+
+    async def update_auto_comment_field(
+            self,
+            channel_id: int,
+            field: str,
+            value: str,
+            updated_by: Optional[int] = None
+    ) -> bool:
+        """
+        Обновляет одно поле автокомментария.
+
+        Args:
+            channel_id: ID канала
+            field: Имя поля (text, button_text, button_url, photo_url)
+            value: Новое значение
+            updated_by: ID админа
+
+        Returns:
+            bool: True если успешно
+        """
+        try:
+
+            async with self.db.get_session() as session:
+                result = await session.execute(
+                    select(AutoComment).where(AutoComment.channel_id == channel_id)
+                )
+                auto_comment = result.scalar_one_or_none()
+
+                if not auto_comment:
+                    return False
+
+                # Обновляем поле
+                if hasattr(auto_comment, field):
+                    setattr(auto_comment, field, value)
+                    auto_comment.updated_by = updated_by
+                    auto_comment.updated_at = datetime.now(timezone.utc)
+                    await session.commit()
+
+                    logger.info(
+                        f"Поле '{field}' автокомментария для канала {channel_id} обновлено",
+                        log_type="DATABASE"
+                    )
+                    return True
+                else:
+                    logger.error(
+                        f"Поле '{field}' не существует в AutoComment",
+                        log_type="DATABASE"
+                    )
+                    return False
+
+        except Exception as e:
+            logger.error(
+                f"Ошибка обновления поля автокомментария: {e}",
+                log_type="DATABASE"
+            )
+            return False
+
+    async def toggle_auto_comment(
+            self,
+            channel_id: int,
+            is_enabled: bool,
+            updated_by: Optional[int] = None
+    ) -> bool:
+        """
+        Включает/выключает автокомментарии для канала.
+
+        Args:
+            channel_id: ID канала
+            is_enabled: True - включить, False - выключить
+            updated_by: ID админа
+
+        Returns:
+            bool: True если успешно
+        """
+        try:
+
+            async with self.db.get_session() as session:
+                result = await session.execute(
+                    select(AutoComment).where(AutoComment.channel_id == channel_id)
+                )
+                auto_comment = result.scalar_one_or_none()
+
+                if not auto_comment:
+                    return False
+
+                auto_comment.is_enabled = is_enabled
+                auto_comment.updated_by = updated_by
+                auto_comment.updated_at = datetime.now(timezone.utc)
+                await session.commit()
+
+                logger.info(
+                    f"Автокомментарии для канала {channel_id} {'включены' if is_enabled else 'выключены'}",
+                    log_type="DATABASE"
+                )
+                return True
+
+        except Exception as e:
+            logger.error(
+                f"Ошибка переключения автокомментария: {e}",
+                log_type="DATABASE"
+            )
+            return False
+
+    async def get_all_auto_comments(self) -> list['AutoComment']:
+        """
+        Получает все настройки автокомментариев.
+
+        Returns:
+            List[AutoComment]: Список всех автокомментариев
+        """
+        try:
+
+            async with self.db.get_session() as session:
+                result = await session.execute(select(AutoComment))
+                return list(result.scalars().all())
+        except Exception as e:
+            logger.error(
+                f"Ошибка получения всех автокомментариев: {e}",
+                log_type="DATABASE"
+            )
+            return []
+
+    async def delete_auto_comment(self, channel_id: int) -> bool:
+        """
+        Удаляет настройки автокомментариев для канала.
+
+        Args:
+            channel_id: ID канала
+
+        Returns:
+            bool: True если удалено
+        """
+        try:
+
+            async with self.db.get_session() as session:
+                result = await session.execute(
+                    delete(AutoComment).where(AutoComment.channel_id == channel_id)
+                )
+                await session.commit()
+                deleted = result.rowcount > 0
+
+                if deleted:
+                    logger.info(
+                        f"Автокомментарий для канала {channel_id} удалён",
+                        log_type="DATABASE"
+                    )
+                return deleted
+
+        except Exception as e:
+            logger.error(
+                f"Ошибка удаления автокомментария: {e}",
+                log_type="DATABASE"
+            )
+            return False