summaryrefslogtreecommitdiff
path: root/ATRI/plugins/setu/data_source.py
blob: 47f76b62e74492058f35053b586cc70194731124 (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
import base64

# from pathlib import Path
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"
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)
        url = temp_data["urls"].get("original", "ignore")

        setu = MessageSegment.image(url, timeout=114514)
        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)
        url = temp_data["urls"].get("original", "ignore")
        setu = MessageSegment.image(url, timeout=114514)
        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]

        tag = choice(temp_data.get("tags", ["女孩子"]))

        url = temp_data["urls"].get("original", "ignore")
        setu = MessageSegment.image(url, timeout=114514)
        repo = SCHEDULER_FORMAT.format(tag=tag, setu=setu)
        return repo