Commited backup

GeekTG/FTG-Modules/pmlog.py

@@ -0,0 +1,75 @@
+# -*- coding: utf-8 -*-
+
+import asyncio
+import logging
+
+from telethon import types
+
+from .. import loader, utils
+
+logger = logging.getLogger(__name__)
+
+
+@loader.tds
+class PMLogMod(loader.Module):
+    """Logs unwanted PMs to a channel"""
+
+    strings = {
+        "name": "PM Logger",
+        "start": "<b>Your conversation is now being logged</b>",
+        "not_pm": "<b>You can't log a group</b>",
+        "stopped": "<b>Your conversation is no longer being logged</b>",
+        "log_group_cfg_doc": "Group or channel ID where to send the logged PMs",
+    }
+
+    def __init__(self):
+        self.config = loader.ModuleConfig(
+            "LOG_GROUP", None, lambda m: self.strings("log_group_cfg_doc", m)
+        )
+
+    async def logpmcmd(self, message):
+        """Begins logging PMs"""
+        if (
+            not message.is_private
+            or message.to_id.user_id == (await message.client.get_me(True)).user_id
+        ):
+            await utils.answer(message, self.strings("not_pm", message))
+            return
+        self._db.set(
+            __name__,
+            "users",
+            list(set(self._db.get(__name__, "users", []) + [message.to_id.user_id])),
+        )
+        msgs = await utils.answer(message, self.strings("start", message))
+        await asyncio.sleep(1)
+        await message.client.delete_messages(message.to_id, msgs)
+
+    async def unlogpmcmd(self, message):
+        """Stops logging PMs"""
+        if (
+            not message.is_private
+            or message.to_id.user_id == (await message.client.get_me(True)).user_id
+        ):
+            await utils.answer(message, self.strings("not_pm", message))
+            return
+        self._db.set(
+            __name__,
+            "users",
+            list(
+                set(self._db.get(__name__, "users", [])).difference(
+                    [message.to_id.user_id]
+                )
+            ),
+        )
+        await utils.answer(message, self.strings("stopped", message))
+
+    async def watcher(self, message):
+        if not message.is_private or not isinstance(message, types.Message):
+            return
+        if self.config["LOG_GROUP"] and utils.get_chat_id(message) in self._db.get(
+            __name__, "users", []
+        ):
+            await message.forward_to(self.config["LOG_GROUP"])
+
+    async def client_ready(self, client, db):
+        self._db = db