summaryrefslogtreecommitdiff
path: root/ATRIbot/plugins/welcome.py
blob: da0a4c3e6db6b2da1eb2e00d34727dc9b4a19bfb (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
import nonebot
from nonebot import on_command, on_notice
from nonebot import CommandSession
from nonebot.notice_request import NoticeSession, RequestSession
from nonebot.plugin import on_request
from aiocqhttp.exceptions import ActionFailed

bot = nonebot.get_bot()
master = bot.config.SUPERUSERS


@on_notice('group_increase')
async def _(session: NoticeSession):
    q = session.event.user_id
    me = session.event.self_id

    if q == me :
        await session.send('在下アトリ,今后请多指教呐❤')
    else:
        await session.send(f'ねえ❤...是新人欸!\nここでは遠慮はいらないのだからね❤')

@on_notice('group_decrease')
async def _(session: NoticeSession):
    q = session.event.user_id
    qm = session.event.operator_id

    if qm == q:
        inf = await bot.get_stranger_info(user_id = q)

        name = inf['nickname']

        await session.send(f'ねえ...{name}({q}) 跑了...')

@on_notice('firend_add')
async def _(session: NoticeSession):
    u = session.event.user_id
    await bot.send_private_msg(self_id = session.self_id, user_id = u, message=f'有人来加我啦!\n{u}')


@on_request('group')
async def bot_request(session: RequestSession):
    g = session.event.group_id
    u = session.event.user_id
    mt = master[0]
    print(mt)

    if session.event.sub_type == 'invite':
        
        if welcome_switch:

            await session.bot.send_private_msg(self_id = session.self_id, user_id = u, message='嗯哼?想邀请吾辈入群嘛,既然这样,那吾辈来了呐。\n作者联系方式:\nhttps://lolihub.icu/#/about')

            await session.bot.send_private_msg(self_id = session.self_id, user_id = mt, message=f'吾辈收到一个新邀请:\n裙: {g}\n邀请人: {u}')

            try:
                await session.approve()
            except ActionFailed as e:
                print(e.retcode)

        else:
            await session.bot.send_private_msg(self_id = session.self_id, user_id = u, message='主人告诉吾辈不能同意任何人的请求呢...\n作者联系方式:\nhttps://lolihub.icu/#/about')
            
            await session.bot.send_private_msg(self_id = session.self_id, user_id = mt, message=f'吾辈收到一个新的邀请请求,由于主人并未告知吾辈可以邀请,故做出拒绝的回应。\n裙: {g}\n邀请人: {u}')


welcome_switch = True
@on_command('welcome_switch', aliases=['开启', '关闭'], only_to_me=False)
async def _(session: CommandSession):
    if session.event.user_id in master:
        command = session.event.raw_message.split(' ', 1)
        switch = command[0]
        com = command[1]
        global welcome_switch
        if switch == '开启':
            if com == '邀请':
                welcome_switch = True
            else:
                pass

        elif switch == '关闭':
            if com == '邀请':
                welcome_switch = False
            else:
                pass
        
        await session.send('完成')
    
    else:
        await session.send('恁哪位?')