PagerMaid_Plugins_Pyro/hitokoto/main.py

from pagermaid.listener import listener
from pagermaid.enums import Message, AsyncClient
from pagermaid.utils import lang


@listener(command="hitokoto", description=lang("hitokoto_des"))
async def hitokoto(request: AsyncClient, message: Message):
    hitokoto_while = 1
    hitokoto_json = None
    try:
        hitokoto_json = (
            await request.get("https://v1.hitokoto.cn/?charset=utf-8")
        ).json()
    except ValueError:
        while hitokoto_while < 10:
            hitokoto_while += 1
            try:
                hitokoto_json = (
                    await request.get("https://v1.hitokoto.cn/?charset=utf-8")
                ).json()
                break
            except Exception:
                continue
        if not hitokoto_json:
            return
    hitokoto_type = {
        "a": lang("hitokoto_type_anime"),
        "b": lang("hitokoto_type_manga"),
        "c": lang("hitokoto_type_game"),
        "d": lang("hitokoto_type_article"),
        "e": lang("hitokoto_type_original"),
        "f": lang("hitokoto_type_web"),
        "g": lang("hitokoto_type_other"),
        "h": lang("hitokoto_type_movie"),
        "i": lang("hitokoto_type_poem"),
        "j": lang("hitokoto_type_netease_music"),
        "k": lang("hitokoto_type_philosophy"),
        "l": lang("hitokoto_type_meme"),
    }
    add = ""
    if works := hitokoto_json["from"]:
        add += f"《{works}》"
    if works_type := hitokoto_json["type"]:
        add += f"（{hitokoto_type[works_type]}）"
    if from_who := hitokoto_json["from_who"]:
        add += f"{from_who}"
    await message.edit(f"{hitokoto_json['hitokoto']} {add}")