summaryrefslogtreecommitdiff
path: root/ATRI/utils/cqcode.py
blob: 3102a92d7cd98ae4a8de86d8bb60628f05d19802 (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
import re
from typing import Optional

from ATRI.service import Service as sv


tencent_gchat_url = "gchat.qpic.cn"
noob_code = ["record", "video", "music", "xml", "json"]


async def coolq_code_check(cq_code: str,
                     user: Optional[int] = None,
                     group: Optional[int] = None):
    _type = re.findall(r"CQ:(.*?),", cq_code)
    for i in _type:
        if i == "image":
            result = re.findall(r"url=(.*?)]", cq_code)
            url = "" if not result else result[0]
            if tencent_gchat_url not in url:
                msg = "你注你🐎呢"
                await sv.NetworkPost.send_msg(user_id=user,
                                              group_id=group,
                                              message=msg)
            else:
                return True
        elif i in noob_code:
            return False
        else:
            return True