PamGram/modules/apihelper/artifact.py

from base64 import b64encode
from random import choice

import httpx


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


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


class ArtifactOcrRate:
    OCR_URL = "https://api.genshin.pub/api/v1/app/ocr"
    RATE_URL = "https://api.genshin.pub/api/v1/relic/rate"
    HEADERS = {
        'authority': 'api.genshin.pub',
        'accept': 'application/json, text/plain, */*',
        'accept-language': 'zh-CN,zh;q=0.9,zh-Hans;q=0.8,und;q=0.7,en;q=0.6,zh-Hant;q=0.5,ja;q=0.4',
        'content-type': 'application/json;charset=UTF-8',
        'dnt': '1',
        'origin': 'https://genshin.pub',
        'referer': 'https://genshin.pub/',
        'sec-ch-ua': '"Chromium";v="104", " Not A;Brand";v="99"',
        'sec-ch-ua-mobile': '?0',
        'sec-ch-ua-platform': '"Windows"',
        'sec-fetch-dest': 'empty',
        'sec-fetch-mode': 'cors',
        'sec-fetch-site': 'same-site',
        'sec-gpc': '1',
        'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/104.0.5112.115 Safari/537.36',
    }

    def __init__(self):
        self.client = httpx.AsyncClient(headers=self.HEADERS)

    async def close(self):
        await self.client.aclose()

    async def get_artifact_attr(self, photo_byte):
        b64_str = b64encode(photo_byte).decode()
        req = await self.client.post(self.OCR_URL, json={"image": b64_str}, timeout=8)
        return req

    async def rate_artifact(self, artifact_attr: dict):
        req = await self.client.post(self.RATE_URL, json=artifact_attr, timeout=8)
        return req