Commited backup

AlpacaGang/ftg-modules/Tag.py

@@ -0,0 +1,125 @@
+# -*- coding: future_fstrings -*-
+
+#    Friendly Telegram (telegram userbot)
+#    By Magical Unicorn (based on official Anti PM & AFK Friendly Telegram modules)
+#    Copyright (C) 2020 Magical Unicorn
+
+#    This program is free software: you can redistribute it and/or modify
+#    it under the terms of the GNU Affero General Public License as published by
+#    the Free Software Foundation, either version 3 of the License, or
+#    (at your option) any later version.
+
+#    This program is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#    GNU Affero General Public License for more details.
+
+#    You should have received a copy of the GNU Affero General Public License
+#    along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+from .. import loader, utils
+
+import logging
+
+from telethon import functions, types
+from telethon.tl.types import PeerUser, PeerChat, PeerChannel, ChannelParticipantsAdmins
+logger = logging.getLogger(__name__)
+
+
+def register(cb):
+    cb(TagMod())
+
+
+@loader.tds
+class TagMod(loader.Module):
+    """
+    Tag :
+    -> Tag all admins (fast way to report).
+    -> Tag all bots (why not ?).
+    -> Tag all members (why not ?).\n
+    Commands :
+     
+    """
+    strings = {"name": "Tag",
+               "error_chat": "<b>This command can be used in channels and group chats only.</b>",
+               "unknow": ("An unknow problem as occured."
+                          "\n\nPlease report problem with logs on "
+                          "<a href='https://github.com/LegendaryUnicorn/FTG-Unofficial-Modules'>Github</a>."),
+               "user_link": "\n• <a href='tg://user?id={}'>{}</a>"}
+
+    def config_complete(self):
+        self.name = self.strings["name"]
+
+    async def admincmd(self, message):
+        """
+        .admin : Tag all admins (excepted bots).
+        .admin [message] : Tag all admins (excepted bots) with message before tags.
+         
+        """
+        if isinstance(message.to_id, PeerUser):
+            await utils.answer(message, self.strings["error_chat"])
+            return
+        if utils.get_args_raw(message):
+            rep = utils.get_args_raw(message)
+        else:
+            rep = ""
+        user = await utils.get_target(message)
+        if isinstance(message.to_id, PeerChat) or isinstance(message.to_id, PeerChannel):
+            async for user in message.client.iter_participants(message.to_id, filter=ChannelParticipantsAdmins):
+                if not user.bot:
+                    user_name = user.first_name
+                    if user.last_name is not None:
+                        user_name += " " + user.last_name
+                    rep += self.strings["user_link"].format(user.id, user_name)
+            await utils.answer(message, rep)
+        else:
+            await utils.answer(message, self.strings["unknow"])
+
+    async def allcmd(self, message):
+        """
+        .all : Tag all members.
+        .all [message] : Tag all members with message before tags.
+         
+        """
+        if isinstance(message.to_id, PeerUser):
+            await utils.answer(message, self.strings["error_chat"])
+            return
+        if utils.get_args_raw(message):
+            rep = utils.get_args_raw(message)
+        else:
+            rep = ""
+        user = await utils.get_target(message)
+        if isinstance(message.to_id, PeerChat) or isinstance(message.to_id, PeerChannel):
+            async for user in message.client.iter_participants(message.to_id):
+                user_name = user.first_name
+                if user.last_name is not None:
+                    user_name += " " + user.last_name
+                rep += self.strings["user_link"].format(user.id, user_name)
+            await utils.answer(message, rep)
+        else:
+            await utils.answer(message, self.strings["unknow"])
+
+    async def botcmd(self, message):
+        """
+        .bot : Tag all bots.
+        .bot [message] : Tag all bots with message before tags.
+         
+        """
+        if isinstance(message.to_id, PeerUser):
+            await utils.answer(message, self.strings["error_chat"])
+            return
+        if utils.get_args_raw(message):
+            rep = utils.get_args_raw(message)
+        else:
+            rep = ""
+        user = await utils.get_target(message)
+        if isinstance(message.to_id, PeerChat) or isinstance(message.to_id, PeerChannel):
+            async for user in message.client.iter_participants(message.to_id):
+                if user.bot:
+                    user_name = user.first_name
+                    if user.last_name is not None:
+                        user_name += " " + user.last_name
+                    rep += self.strings["user_link"].format(user.id, user_name)
+            await utils.answer(message, rep)
+        else:
+            await utils.answer(message, self.strings["unknow"])