From 56bb7e3cacdff61e6cc94c3d2fecf2d7249780c7 Mon Sep 17 00:00:00 2001
From: b miao <blucartfhg@outlook.com>
Date: Sat, 28 Nov 2020 18:04:53 +0800
Subject: [PATCH] portball v1.40 - ignore msg if sender is not admin

---
 list.json   |  2 +-
 portball.py | 73 +++++++++++++++++++++++++++++------------------------
 2 files changed, 41 insertions(+), 34 deletions(-)

diff --git a/list.json b/list.json
index 8bf927d..c6fe57f 100644
--- a/list.json
+++ b/list.json
@@ -152,7 +152,7 @@
         },
         {
             "name": "portball",
-            "version": "1.31",
+            "version": "1.40",
             "section": "chat",
             "maintainer": "Pentacene",
             "size": "3.6 kb",
diff --git a/portball.py b/portball.py
index 1edc625..9cf9f1f 100644
--- a/portball.py
+++ b/portball.py
@@ -6,44 +6,50 @@ from datetime import timedelta
 from telethon.tl.types import ChannelParticipantsAdmins
 
 
-@listener(is_plugin=True, outgoing=True, command="portball",
+@listener(is_plugin=True, outgoing=True, incoming=True, command="portball",
           description="回复你要临时禁言的人的消息来实现XX秒的禁言",
           parameters="<理由>(空格)<时间/秒>")
 async def portball(context):
     reply = await context.get_reply_message()
     if context.is_group:
-        if reply:
-            action = context.arguments.split()
-            if reply.sender.last_name == None:
-                last_name = ''
-            else:
-                last_name = reply.sender.last_name
-            if len(action) < 2:
-                notification = await bot.send_message(context.chat_id, '格式是\n-portball 理由 秒数\n真蠢', reply_to = context.id)
-                await sleep(10)
-                await notification.delete()
-                try:
-                    await context.delete()
-                except:
-                    pass
-                return False
-            if int(action[1])<60:
-                notification = await bot.send_message(context.chat_id, '诶呀不要小于60秒啦', reply_to = context.id)
-                await sleep(10)
-                await notification.delete()
-                try:
-                    await context.delete()
-                except:
-                    pass
-                return False
-            admins = await context.client.get_participants(context.chat, filter=ChannelParticipantsAdmins)
-            if (context.sender in admins) or (context.sender_id == 1087968824):
+        admins = await context.client.get_participants(context.chat, filter=ChannelParticipantsAdmins)
+        # if message sender is one of admins or group anonymous bot
+        if (context.sender in admins) or (context.sender_id == 1087968824):
+            # if this msg reply to a message
+            if reply:
+                action = context.arguments.split()
+                # get sender's last_name
+                if reply.sender.last_name == None:
+                    last_name = ''
+                else:
+                    last_name = reply.sender.last_name
+                # if something wrong with the command
+                if len(action) < 2:
+                    notification = await bot.send_message(context.chat_id, '格式是\n-portball 理由 秒数\n真蠢', reply_to = context.id)
+                    await sleep(10)
+                    await notification.delete()
+                    try:
+                        await context.delete()
+                    except:
+                        pass
+                    return False
+                # if time of portball less than 60 seconds
+                if int(action[1])<60:
+                    notification = await bot.send_message(context.chat_id, '诶呀不要小于60秒啦', reply_to = context.id)
+                    await sleep(10)
+                    await notification.delete()
+                    try:
+                        await context.delete()
+                    except:
+                        pass
+                    return False
+                # portball
                 try:
                     await bot.edit_permissions(context.chat_id, reply.sender.id, timedelta(seconds=int(action[1].replace(' ',''))), send_messages=False, send_media=False, send_stickers=False, send_gifs=False, send_games=False, send_inline=False, send_polls=False, invite_users=False, change_info=False, pin_messages=False)
                     portball_message = await bot.send_message(
                         context.chat_id,
-                        f'[{reply.sender.first_name}{last_name}](tg://user?id={reply.sender.id}) 由于 {action[0]} \n'
-                        f'被暂时禁言{action[1]}秒',
+                        f'[{reply.sender.first_name}{last_name}](tg://user?id={reply.sender.id}) 由于 {action[0]} 被塞了{action[1]}秒口球.\n
+                        f'到期自动拔出,无后遗症.',
                         reply_to = reply.id
                     )
                     await context.delete()
@@ -57,14 +63,15 @@ async def portball(context):
                     notification = await bot.send_message(context.chat_id, '错误：我没有管理员权限或我的权限比被封禁的人要小', reply_to = context.id)
                     await sleep(10)
                     await notification.delete()
+
+            # if not reply to any message
             else:
-                notification = await bot.send_message(context.chat_id, '诶呀你不是管理员，不要给人家塞口球啊', reply_to = context.id)
+                notification = await bot.send_message(context.chat_id, '你好蠢诶，都没有回复人，我哪知道你要搞谁的事情……', reply_to = context.id)
                 await sleep(10)
                 await notification.delete()
+        # if sender is not admin
         else:
-            notification = await bot.send_message(context.chat_id, '你好蠢诶，都没有回复人，我哪知道你要搞谁的事情……', reply_to = context.id)
-            await sleep(10)
-            await notification.delete()
+            return
     else:
         notification = await bot.send_message(context.chat_id, '你好蠢诶，又不是群组，怎么禁言啦！', reply_to = context.id)
         await sleep(10)