Commited backup

vsecoder/hikka_modules/speechcensorship.py

@@ -0,0 +1,103 @@
+"""
+                              _
+__   _____  ___  ___ ___   __| | ___ _ __
+\ \ / / __|/ _ \/ __/ _ \ / _` |/ _ \ '__|
+ \ V /\__ \  __/ (_| (_) | (_| |  __/ |
+  \_/ |___/\___|\___\___/ \__,_|\___|_|
+
+  Copyleft 2022 t.me/vsecoder
+  This program is free software; you can redistribute it and/or modify
+
+"""
+# meta developer: @vsecoder_m
+# requires: git+https://github.com/vsecoder/py-censure
+
+__version__ = (2, 0, 0)
+
+import logging
+from .. import loader, utils  # type: ignore
+from censure import Censor
+
+logger = logging.getLogger(__name__)
+
+
+@loader.tds
+class SpeechCensorshipMod(loader.Module):
+    """Module for censoring your speech"""
+
+    strings = {
+        "name": "SpeechCensorship",
+        "_cfg_replace_symbols": "Replace symbols for censoring",
+        "_cfg_language": "Language",
+        "turn": "<emoji document_id=5235698355119596341>😆</emoji> <b>Censorship mode {}</b>",
+    }
+
+    strings_ru = {
+        "_cfg_replace_symbols": "Символы для замены мата",
+        "_cfg_language": "Язык",
+        "turn": "<emoji document_id=5235698355119596341>😆</emoji> <b>Режим цензуры {}</b>",
+    }
+
+    async def client_ready(self, client, db):
+        self.client = client
+        self.db = db
+        self.get("censorship_work", False)
+        self.me = await client.get_me()
+
+    def __init__(self):
+        self.config = loader.ModuleConfig(
+            loader.ConfigValue(
+                "replace_symbols",
+                "#",
+                lambda m: self.strings["_cfg_replace_symbols"],
+            ),
+            loader.ConfigValue(
+                "language",
+                ["ru", "en"],
+                lambda m: self.strings["_cfg_language"],
+                validator=loader.validators.MultiChoice(["ru", "en"]),
+            ),
+        )
+        self.name = self.strings["name"]
+
+    async def censorshipcmd(self, message):
+        """
+        Turn on/off censorship mode
+        """
+        censorship_work = self.get("censorship_work", False)
+
+        work = "on" if not censorship_work else "off"
+        self.set("censorship_work", not censorship_work)
+
+        await utils.answer(message, self.strings["turn"].format(work))
+
+    async def censor_task(self, text):
+        languages = self.config["language"]
+        replace_symbols = self.config["replace_symbols"]
+
+        censors = [Censor.get(lang) for lang in languages]
+
+        for censor in censors:
+            text = censor.clean_line(text, beep=replace_symbols)[0]
+
+        return text
+
+    async def watcher(self, message):
+        try:
+            if self.me.id != message.from_id:
+                return
+
+            censorship_work = self.get("censorship_work", False)
+
+            if not censorship_work:
+                return
+
+            text = message.text
+
+            if text:
+                censored = await self.censor_task(text)
+
+                if text != censored:
+                    await message.edit(censored)
+        except:
+            return