Merge pull request #1 from SouceCalve/dev

>Chat exclusion func
Added by me, merged by me)

NekoSpeak.py

@@ -1,125 +1,171 @@
 from hikkatl.types import Message
 from .. import loader, utils
 import random
+
 @loader.tds
 class NekoSpeakModule(loader.Module):
     """Модуль для стилизации речи под неко"""
-    strings = {"name": "NekoSpeak", "converted": "{text}"}
-    strings_ru = {"converted": "{text}"}
+
+    strings = {
+        "name": "NekoSpeak",
+        "converted": "{text}",
+        "banned": "Ня~ Чат {} добавлен в черный список.",
+        "unbanned": "Ня~ Чат {} удален из черного списка.",
+        "already_banned": "Ня... Этот чат уже в черном списке!",
+        "not_banned": "Ня... Этот чат не был забанен!",
+        "no_access": "Ня? У меня нет доступа к этому чату.",
+        "banned_list": "Список забаненных чатов:\n{}"
+    }
 
     def __init__(self):
         self.active = False
         self.pm_enabled = False
         self.public_enabled = False
-        self.realcatlike = False
+        # Загружаем из БД или создаём новый список
+        self.banned_chats = set()
+
+    async def client_ready(self):
+        saved = self.db.get("NekoSpeak", "banned_chats", [])
+        self.banned_chats = set(saved)
+
+    def save_banned_chats(self):
+        # Сохраняем список забаненных чатов в БД
+        self.db.set("NekoSpeak", "banned_chats", list(self.banned_chats))
 
     def neko_speak(self, text):
-        # Замена "н" на "ня" в некоторых словах
         words = text.split()
         new_words = []
         for word in words:
             if word.lower().startswith("н") and len(word) > 1:
                 word = "ня" + word[2:]
-            word=word.replace("р","р"*random.randint(2,4))
+            word = word.replace("р", "р" * random.randint(2,4))
             new_words.append(word)
         text = " ".join(new_words)
 
-        # Замена стандартных фраз
         replacements = {
-            "ничего": "нячего",
-            "что?": "ня?",
-            "ура!": "ня!",
-            "о нет": "о-ня!",
-            "привет": "мяувет",
-            "Привет": "Мяувет",
-            "хорошо":"мяу-ряско",
-            "плохо":"мяу-чалька",
-            "пиздец":"мря!",
-            "Пиздец":"Мря!",
-            "блять":"мяр!",
-            "Блять":"Мяр!",
-            "замечательные":"замуррчательные",
-            "замечательный":"замуррчательный",
-            "замечательных":"замуррчательных",
-            "Замечательные":"Замуррчательные",
-            "Замечательный":"Замуррчательный",
-            "Замечательных":"Замуррчательных",
-            "прекрасное":"замурчательное",
-            "Прекрасное":"Замурчательное",
-            "прекрасный":"замурчательный",
-            "Прекрасный":"Замурчательный",
-            "прекрасная":"замурчательная",
-            "Прекрасная":"Замурчательная",
-            "заебись":"замурчательно",
-            "Заебись":"Замурчательно",
-            "охуенно":"замурчательно",
-            "Охуенно":"Замурчательно",
-            "няхуя":"МРЯВ?! Ньё-ньё!!",
-            "Няхуя":"МРЯВ?! Ньё-ньё!!",
-            "няко":"неко",
-            "Няко":"неко",
-            "нядо":"надо",
-            "Нядо":"Надо",
-            "В рот ебал":"МРЯФФ!!",
-            "в рот ебал":"МРЯФФ!!",
-            "ебал":"мряфф!!",
-            "Ебал":"МРЯФФ!!",
-            "хуёво":"ньёрфф...",
-            "Хуёво":"Ньёрфф...",
-            "пидрилит":"мряк-мрря!",
+            "ничего": "нячего", "что?": "ня?", "ура!": "ня!", "о нет": "о-ня!",
+            "привет": "мяувет", "Привет": "Мяувет", "хорошо":"мяу-ряско",
+            "плохо":"мяу-чалька", "пиздец":"мря!", "Пиздец":"Мря!",
+            "блять":"мяр!", "Блять":"Мяр!", "замечательные":"замуррчательные",
+            "замечательный":"замуррчательный", "замечательных":"замуррчательных",
+            "Замечательные":"Замуррчательные", "Замечательный":"Замуррчательный",
+            "Замечательных":"Замуррчательных", "прекрасное":"замурчательное",
+            "Прекрасное":"Замурчательное", "прекрасный":"замурчательный",
+            "Прекрасный":"Замурчательный", "прекрасная":"замурчательная",
+            "Прекрасная":"Замурчательная", "заебись":"замурчательно",
+            "Заебись":"Замурчательно", "охуенно":"замурчательно",
+            "Охуенно":"Замурчательно", "няхуя":"МРЯВ?! Ньё-ньё!!",
+            "Няхуя":"МРЯВ?! Ньё-ньё!!", "няко":"неко", "Няко":"неко",
+            "нядо":"надо", "Нядо":"Надо", "В рот ебал":"МРЯФФ!!",
+            "в рот ебал":"МРЯФФ!!", "ебал":"мряк-мрря!", "Ебал":"МРЯФФ!!",
+            "хуёво":"ньёрфф...", "Хуёво":"Ньёрфф...", "пидрилит":"мряк-мрря!",
             "Пидрилит":"Мряк-мрря!",
         }
+
         for key, value in replacements.items():
             text = text.replace(key, value)
 
-        # Добавление "мяу" и концовок
-        if random.random() < 0.3:  # 30% шанс добавить "мяу"
+        if random.random() < 0.3:
             text += " мяу~"
-        elif random.random() < 0.2:  # 20% шанс добавить "-ня"
+        elif random.random() < 0.2:
             text += "-ня"
 
         return text
 
     @loader.command(ru_doc="Стилизовать текст под неко")
     async def neko(self, message: Message):
-        """Стилизовать текст под неко"""
         text = utils.get_args_raw(message)
         if not text:
             return await utils.answer(message, "Ня? Введи текст!")
-
         converted_text = self.neko_speak(text)
         await utils.answer(message, self.strings("converted").format(text=converted_text))
 
     @loader.command(ru_doc="Включить автоматическое преобразование речи в неко-стиль")
     async def nekospeak(self, message: Message):
-        """Включить автоматическое преобразование речи в неко-стиль"""
         self.active = True
         await utils.answer(message, "Ня~ Теперь все твои сообщения будут преобразовываться!")
 
     @loader.command(ru_doc="Отключить автоматическое преобразование речи в неко-стиль")
     async def nekospeakstop(self, message: Message):
-        """Отключить автоматическое преобразование речи в неко-стиль"""
         self.active = False
         await utils.answer(message, "Ня... Теперь сообщения не будут изменяться.")
 
     @loader.command(ru_doc="Переключить преобразование сообщений в ЛС")
     async def nekospeakpm(self, message: Message):
-        """Переключить преобразование сообщений в ЛС"""
         self.pm_enabled = not self.pm_enabled
         status = "включено" if self.pm_enabled else "выключено"
         await utils.answer(message, f"Ня! Преобразование в ЛС теперь {status}.")
 
     @loader.command(ru_doc="Переключить преобразование сообщений в публичных чатах")
     async def nekospeakpublic(self, message: Message):
-        """Переключить преобразование сообщений в публичных чатах"""
         self.public_enabled = not self.public_enabled
         status = "включено" if self.public_enabled else "выключено"
         await utils.answer(message, f"Ня! Преобразование в публичных чатах теперь {status}.")
 
+    @loader.command(ru_doc="Заблокировать чат для работы модуля")
+    async def nekospeakex(self, message: Message):
+        args = utils.get_args_raw(message)
+        try:
+            chat_id = int(args) if args.isdigit() else None
+            if not chat_id:
+                reply = await message.get_reply_message()
+                if reply:
+                    chat_id = reply.sender_id or reply.chat_id
+                else:
+                    chat_id = message.chat_id
+        except Exception:
+            chat_id = message.chat_id
+
+        if chat_id in self.banned_chats:
+            await utils.answer(message, self.strings("already_banned"))
+            return
+
+        self.banned_chats.add(chat_id)
+        self.save_banned_chats()
+        await utils.answer(message, self.strings("banned").format(str(chat_id)))
+
+    @loader.command(ru_doc="Разблокировать чат")
+    async def nekospeakin(self, message: Message):
+        args = utils.get_args_raw(message)
+        try:
+            chat_id = int(args) if args.isdigit() else None
+            if not chat_id:
+                reply = await message.get_reply_message()
+                if reply:
+                    chat_id = reply.sender_id or reply.chat_id
+                else:
+                    chat_id = message.chat_id
+        except Exception:
+            chat_id = message.chat_id
+
+        if chat_id not in self.banned_chats:
+            await utils.answer(message, self.strings("not_banned"))
+            return
+
+        self.banned_chats.remove(chat_id)
+        self.save_banned_chats()
+        await utils.answer(message, self.strings("unbanned").format(str(chat_id)))
+
+    @loader.command(ru_doc="Показать список заблокированных чатов (только в ЛС)")
+    async def nekospeaklist(self, message: Message):
+        if not message.is_private:
+            await utils.answer(message, self.strings("no_access"))
+            return
+
+        if not self.banned_chats:
+            await utils.answer(message, "Ня... Список забаненных чатов пуст.")
+            return
+
+        banned_list = "\n".join([f"- {chat}" for chat in self.banned_chats])
+        await utils.answer(message, self.strings("banned_list").format(banned_list))
+
     async def watcher(self, message: Message):
         if not self.active:
             return
         if message.out:
+            chat_id = message.chat_id
+            if chat_id in self.banned_chats:
+                return
             if message.is_private and not self.pm_enabled:
                 return
             if not message.is_private and not self.public_enabled: