Telegram_PaimonBot/plugins/artifact_rate.py

from base64 import b64encode

import requests
from secrets import choice
from pyrogram import Client, filters
from pyrogram.types import Message
from defs.artifact_rate import get_artifact_attr, rate_artifact

from os import remove


def get_format_sub_item(artifact_attr):
    msg = ""
    for i in artifact_attr["sub_item"]:
        msg += f'{i["name"]:\u3000<6} | {i["value"]}\n'
    return msg


def get_yiyan(get_yiyan_num):
    data = {"1": ["破玩意谁能用啊，谁都用不了吧", "喂了吧，这东西做狗粮还能有点用", "抽卡有保底，圣遗物没有下限",
                  "未来可期呢（笑）", "你出门一定很安全", "你是不是得罪米哈游了？", "……宁就是班尼特本特？",
                  "丢人！你给我退出提瓦特(", "不能说很糟糕，只能说特别不好"],
            "2": ["淡如清泉镇的圣水，莫得提升", "你怎么不强化啊？", "嗯嗯嗯好好好可以可以可以挺好挺好（敷衍）",
                  "这就是日常，下一个", "洗洗还能吃（bushi）", "下次一定行……？", "派蒙平静地点了个赞",
                  "不知道该说什么，就当留个纪念吧"],
            "3": ["不能说有质变，只能说有提升", "过渡用的话没啥问题，大概", "再努努力吧", "嗯，差不多能用",
                  "这很合理", "达成成就“合格圣遗物”", "嗯，及格了，过渡用挺好的", "中规中矩，有待提升"],
            "4": ["以普遍理性而论，很好", "算是个很不戳的圣遗物了！", "很好，很有精神！", "再努努力，超越一下自己",
                  "感觉可以戴着它大杀四方了", "这就是大佬背包里的平均水平吧", "先锁上呗，这波不亏", "达成成就“高分圣遗物”",
                  "这波对输出有很大提升啊(认真)", "我也想拥有这种分数的圣遗物(切实)"],
            "5": ["多吃点好的，出门注意安全", "晒吧，欧不可耻，只是可恨", "没啥好说的，让我自闭一会", "达成成就“高分圣遗物”",
                  "怕不是以后开宝箱只能开出卷心菜", "吃了吗？没吃的话，吃我一拳", "我觉得这个游戏有问题", "这合理吗",
                  "这东西没啥用，给我吧（柠檬）", "？？？ ？？？？"]}
    try:
        data_ = int(float(get_yiyan_num))
    except ValueError:
        data_ = 0
    if data_ == 100:
        return choice(data["5"])
    return choice(data[str(data_ // 20 + 1)])


async def artifact_rate_msg(client: Client, message: Message):
    if not message.photo:
        return await message.reply("图呢？\n*请命令将与截图一起发送", quote=True)
    msg = await message.reply("正在下载图片。。。", quote=True)
    path = await message.download()
    with open(path, "rb") as f:
        image_b64 = b64encode(f.read()).decode()
    remove(path)
    try:
        artifact_attr = await get_artifact_attr(image_b64)
    except (requests.exceptions.ConnectionError, requests.exceptions.ReadTimeout):
        return await msg.edit("连接超时")
    if 'err' in artifact_attr.keys():
        err_msg = artifact_attr["full"]["message"]
        return await msg.edit(f"发生了点小错误：\n{err_msg}")
    if 'star' not in artifact_attr:
        msg_ = await message.reply('无法识别圣遗物星级，请手动输入数字（1-5）：', quote=True)
        answer = await client.listen(message.chat.id, filters=filters.user(message.from_user.id))  # noqa
        try:
            artifact_attr['star'] = int(answer.text)
        except ValueError:
            artifact_attr['star'] = 4
        await msg_.delete()
        try:
            await answer.delete()
        except:
            pass
    if 'level' not in artifact_attr:
        msg_ = await message.reply('无法识别圣遗物等级，请手动输入数字（1-20）：', quote=True)
        answer = await client.listen(message.chat.id, filters=filters.user(message.from_user.id))  # noqa
        try:
            artifact_attr['level'] = int(answer.text)
        except ValueError:
            artifact_attr['level'] = 1
        await msg_.delete()
        try:
            await answer.delete()
        except:
            pass
    await msg.edit("识图成功！\n正在评分中...")
    rate_result = await rate_artifact(artifact_attr)
    if 'err' in rate_result.keys():
        err_msg = rate_result["full"]["message"]
        return await msg.edit(f"发生了点小错误：\n{err_msg}")
    format_result = f'圣遗物评分结果：\n' \
                    f'主属性：{artifact_attr["main_item"]["name"]}\n' \
                    f'{get_format_sub_item(artifact_attr)}' \
                    f'`------------------------------`\n' \
                    f'总分：{rate_result["total_percent"]}\n' \
                    f'主词条：{rate_result["main_percent"]}\n' \
                    f'副词条：{rate_result["sub_percent"]}\n' \
                    f'`------------------------------`\n' \
                    f'{get_yiyan(rate_result["total_percent"])}\n' \
                    f'评分、识图均来自 genshin.pub'
    await msg.edit(format_result)