1 files changed, 121 insertions, 0 deletions
diff --git a/ATRI/plugins/send.py b/ATRI/plugins/send.py
new file mode 100644
index 0000000..6cc57e1
--- /dev/null
+++ b/ATRI/plugins/send.py
@@ -0,0 +1,121 @@
+# -*- coding:utf-8 -*-
+import time
+import nonebot
+from nonebot import on_command, CommandSession
+
+
+bot = nonebot.get_bot()
+master = bot.config.SUPERUSERS
+ban_group = bot.config.BANGROUP # type: ignore
+
+
+@on_command('send_all_group', aliases=['公告', '群发', '推送'], only_to_me=False)
+async def send_all_group(session: CommandSession):
+    if session.event.user_id in master:
+        msg = session.current_arg.strip()
+
+        if not msg:
+            msg = session.get('message', prompt='请告诉吾辈需要群发的内容~！')
+
+        group_list = await session.bot.get_group_list() # type: ignore
+
+        for group in group_list:
+
+            if group['group_id'] not in ban_group:
+
+                try:
+                    await bot.send_group_msg(group_id = group['group_id'], message = msg) # type: ignore
+
+                except:
+                    pass
+
+        await session.send('推送完成！')
+
+
+@on_command('send_to_group', aliases=['对群'], only_to_me=False)
+async def send_to_group(session: CommandSession):
+    if session.event.user_id in master:
+        msg = session.current_arg.strip()
+    
+        if not msg:
+            msg = session.get('message', prompt='请告诉吾辈完整内容呢...\n例：对群 12345647(群号) message 1')
+        
+        lg = msg.split(' ')
+
+        group = lg[0]
+        msg = lg[1]
+        rei = int(lg[2]) + 1
+        
+        if rei:
+            for i in range(1, rei):
+                try:
+                    await bot.send_group_msg(group_id = group, message = msg) # type: ignore
+                except:
+                    await session.send('发送失败，请重试')
+        
+        else:
+            try:
+                await bot.send_group_msg(group_id = group, message = msg) # type: ignore
+            except:
+                await session.send('发送失败，请重试')
+        
+        await session.send('推送完成！')
+
+
+@on_command('send_to_qq', aliases=['对QQ'], only_to_me=False)
+async def send_to_qq(session: CommandSession):
+    if session.event.user_id in master:
+        msg = session.current_arg.strip()
+    
+        if not msg:
+            msg = session.get('message', prompt='请告诉吾辈完整内容呢...\n例：对QQ 12345647(QQ号) message')
+        
+        lg = msg.split(' ')
+
+        qq = lg[0]
+        msg = lg[1]
+
+        try:
+            await bot.send_private_msg(user_id = qq, message = msg) # type: ignore
+        except:
+            await session.send('发送失败，请重试')
+        
+        await session.send('推送完成！')
+
+
+@on_command('send_to_group_pr', aliases=['对群私聊'], only_to_me=False)
+async def _(session: CommandSession):
+    if session.event.user_id in master:
+        msg = session.current_arg.strip()
+    
+        if not msg:
+            msg = session.get('message', prompt='请告诉吾辈完整内容呢...\n例：对群 12345647(群号) message 1')
+        
+        lg = msg.split(' ')
+
+        group = lg[0]
+        msg = lg[1]
+        rei = int(lg[2]) + 1
+
+        group_user_list = await session.bot.get_group_member_list(group_id = group) # type: ignore
+
+        for i in group_user_list:
+            time.sleep(0.5)
+            userid = i['user_id']
+            print(userid)
+        
+            if rei:
+                for a in range(1, rei):
+                    time.sleep(0.5)
+                    try:
+                        await bot.send_private_msg(user_id = userid, message = msg) # type: ignore
+                    except:
+                        await session.send('发送失败，请重试')
+            
+            else:
+                try:
+                    await bot.send_private_msg(user_id = userid, message = msg) # type: ignore
+                except:
+                    await session.send('发送失败，请重试')
+        
+        await session.send('推送完成！')
+\ No newline at end of file