summaryrefslogtreecommitdiff
path: root/ATRI/plugins/send.py
blob: f1b83af4184218a9239c08c385e79131b79c96eb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
# -*- coding:utf-8 -*-
import time
import nonebot
from nonebot import on_command, CommandSession

import config # type: ignore


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()

        start = time.perf_counter()

        if not msg:
            msg = session.get('message', prompt='请告诉吾辈需要群发的内容~!')

        group_list = await session.bot.get_group_list() # type: ignore
        g_list = len(group_list)

        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
        
        end = time.perf_counter()

        await session.send(f'已推送到[{g_list}]个群\n耗时:{round(end - start, 3)}')

@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 = 1 
        try:
            rei = int(lg[2]) + 1
        except:
            pass

        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_all_friend', aliases = ['全体用户'], only_to_me = False)
async def send_all_friend(session: CommandSession):
    if session.event.user_id in master:
        msg = session.current_arg.strip()

        start = time.perf_counter()

        if not msg:
            msg = session.get('message', prompt='请告诉吾辈需要群发的内容~!')
        
        friend_list = await session.bot.get_friend_list() # type: ignore
        print(friend_list)
        f_list = len(friend_list)

        for friend in friend_list:

            try:
                await bot.send_private_msg(user_id = friend['user_id'], message = msg) # type: ignore
            
            except:
                await bot.send_private_msg(user_id = master, message = f"列表用户[{friend['user_id']}]推送失败") # type: ignore
        
        end = time.perf_counter()

        await session.send(f'已推送到[{f_list}]位用户\n耗时:{round(end - start, 3)}')

@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('推送完成!')