summaryrefslogtreecommitdiff
path: root/ATRI/plugins/setu.py
blob: 315803b39029dfe3c07a1e7ed0d0e7e8cefbee95 (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
# -*- coding:utf-8 -*-
import time
import json
import nonebot
from nonebot import on_command, CommandSession

from ATRI.modules import response # type: ignore


bot = nonebot.get_bot()
master = bot.config.SUPERUSERS
apikey = bot.config.LOLICONAPI # type: ignore

URL = 'https://api.lolicon.app/setu/'

SETU_REPLY = """Title: {title}
Pid: {pid}
[CQ:image,file={setu}]
---------------
完成时间:{time}s"""


@on_command('setu', aliases = ['图来'], only_to_me = False)
async def _(session: CommandSession):
    with open(f'ATRI\\plugins\\switch\\switch.json', 'r') as f:
        data = json.load(f)

    if data["setu"] == 0:
        start = time.perf_counter()
        values = {
            "apikey": apikey,
            "r18": "0",
            "num": "1"
        }

        dc = json.loads(response.request_api_params(URL, values))
        end = time.perf_counter()

        await session.send(
            SETU_REPLY.format(
                title=dc["data"][0]["title"],
                pid=dc["data"][0]["pid"],
                setu=dc["data"][0]["url"],
                time = round(end - start, 3)
            )
        )

    else:
        await session.send('该功能已被禁用...')