summaryrefslogtreecommitdiff
path: root/ATRI/plugins/setu/data_source.py
blob: ba714b27ac360adc3d098abdadcd0f7889a41bb7 (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
from random import choice
from nonebot.adapters.cqhttp import MessageSegment

from ATRI.service import Service
from ATRI.rule import is_in_service
from ATRI.utils import request


LOLICON_URL = "https://api.lolicon.app/setu/v2"
SETU_TEMP_FORMAT = "https://pixiv.cat/{p_id}.{ext}"  # 为何要这样组,因为 i.pixiv.cat 不稳定!
SCHEDULER_FORMAT = """
{tag}哦~❤
{setu}
"""


class Setu(Service):
    def __init__(self):
        Service.__init__(self, "涩图", "hso!", rule=is_in_service("涩图"))

    @staticmethod
    async def random_setu() -> tuple:
        """
        随机涩图.
        """
        res = await request.get(LOLICON_URL)
        data: dict = await res.json()
        temp_data: dict = data.get("data", list())[0]

        title = temp_data.get("title", "木陰のねこ")
        p_id = temp_data.get("pid", 88124144)
        ext = temp_data.get("ext", "jpg")
        url = SETU_TEMP_FORMAT.format(p_id=p_id, ext=ext)
        setu = MessageSegment.image(url)
        return setu, title, p_id

    @staticmethod
    async def tag_setu(tag: str) -> tuple:
        """
        指定tag涩图.
        """
        url = LOLICON_URL + f"?tag={tag}"
        res = await request.get(url)
        data: dict = await res.json()

        temp_data: dict = data.get("data", list())[0]
        if not temp_data:
            is_ok = False
        is_ok = True

        title = temp_data.get("title", "木陰のねこ")
        p_id = temp_data.get("pid", 88124144)
        ext = temp_data.get("ext", "jpg")
        url = SETU_TEMP_FORMAT.format(p_id=p_id, ext=ext)
        setu = MessageSegment.image(url)
        return setu, title, p_id, is_ok

    @staticmethod
    async def scheduler() -> str:
        """
        每隔指定时间随机抽取一个群发送涩图.
        格式:
            是{tag}哦~❤
            {setu}
        """
        res = await request.get(LOLICON_URL)
        data: dict = await res.json()
        temp_data: dict = data.get("data", list())[0]

        p_id = temp_data.get("pid", 88124144)
        tag = choice(temp_data.get("tags", ["女孩子"]))
        ext = temp_data.get("ext", "jpg")

        url = SETU_TEMP_FORMAT.format(p_id=p_id, ext=ext)
        setu = MessageSegment.image(url)
        repo = SCHEDULER_FORMAT.format(tag=tag, setu=setu)
        return repo