summaryrefslogtreecommitdiff
path: root/ATRI/plugins/send.py
blob: 69fc1b0d15df0630960621a29da0252e4cff0ae8 (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
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('推送完成!')