mirror of
https://github.com/PaiGramTeam/PaiGram.git
synced 2024-11-25 09:37:30 +00:00
🔥 Delete Artifact Rate
This commit is contained in:
parent
4ce60123de
commit
15bc2b2724
GPG Key ID:
4CBB3F4FA8C85659
@ -1,104 +0,0 @@
|
|||||||
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
|
|
||||||
@ -1,147 +0,0 @@
|
|||||||
from typing import Optional, Tuple
|
|
||||||
|
|
||||||
from telegram import File, InlineKeyboardButton, InlineKeyboardMarkup, Update
|
|
||||||
from telegram.constants import ChatAction, ParseMode
|
|
||||||
from telegram.ext import CallbackContext, ConversationHandler, filters
|
|
||||||
from telegram.helpers import escape_markdown
|
|
||||||
|
|
||||||
from core.baseplugin import BasePlugin
|
|
||||||
from core.plugin import Plugin, conversation, handler
|
|
||||||
from modules.apihelper.artifact import ArtifactOcrRate, get_comment, get_format_sub_item
|
|
||||||
from utils.decorators.error import error_callable
|
|
||||||
from utils.decorators.restricts import restricts
|
|
||||||
from utils.log import logger
|
|
||||||
|
|
||||||
COMMAND_RESULT = 1
|
|
||||||
|
|
||||||
|
|
||||||
class ArtifactRate(Plugin.Conversation, BasePlugin.Conversation):
|
|
||||||
"""圣遗物评分"""
|
|
||||||
|
|
||||||
STAR_KEYBOARD = [
|
|
||||||
[InlineKeyboardButton(f"{i}", callback_data=f"artifact_ocr_rate_data|star|{i}") for i in range(1, 6)]
|
|
||||||
]
|
|
||||||
|
|
||||||
LEVEL_KEYBOARD = [
|
|
||||||
[
|
|
||||||
InlineKeyboardButton(f"{i * 5 + j}", callback_data=f"artifact_ocr_rate_data|level|{i * 5 + j}")
|
|
||||||
for j in range(1, 6)
|
|
||||||
]
|
|
||||||
for i in range(0, 4)
|
|
||||||
]
|
|
||||||
|
|
||||||
def __init__(self):
|
|
||||||
self.artifact_rate = ArtifactOcrRate()
|
|
||||||
|
|
||||||
async def get_rate(self, artifact_attr: dict) -> str:
|
|
||||||
rate_result_req = await self.artifact_rate.rate_artifact(artifact_attr)
|
|
||||||
if rate_result_req.status_code != 200:
|
|
||||||
if rate_result_req.status_code == 400:
|
|
||||||
artifact_attr = rate_result_req.json()
|
|
||||||
return artifact_attr.get("message", "API请求错误")
|
|
||||||
return "API请求错误，请稍后再试"
|
|
||||||
rate_result = rate_result_req.json()
|
|
||||||
return (
|
|
||||||
"*圣遗物评分结果*\n"
|
|
||||||
f"主属性：{escape_markdown(artifact_attr['main_item']['name'], version=2)}\n"
|
|
||||||
f"{escape_markdown(get_format_sub_item(artifact_attr), version=2)}"
|
|
||||||
f"`--------------------`\n"
|
|
||||||
f"总分：{escape_markdown(rate_result['total_percent'], version=2)}\n"
|
|
||||||
f"主词条：{escape_markdown(rate_result['main_percent'], version=2)}\n"
|
|
||||||
f"副词条：{escape_markdown(rate_result['sub_percent'], version=2)}\n"
|
|
||||||
f"`--------------------`\n"
|
|
||||||
f"{escape_markdown(get_comment(rate_result['total_percent']), version=2)}\n"
|
|
||||||
"_评分、识图均来自 genshin\\.pub_"
|
|
||||||
)
|
|
||||||
|
|
||||||
@conversation.entry_point
|
|
||||||
@handler.command(command="artifact_rate", filters=filters.ChatType.PRIVATE, block=True)
|
|
||||||
@handler.message(filters=filters.Regex(r"^圣遗物评分(.*)"), block=True)
|
|
||||||
@handler.message(filters=filters.CaptionRegex(r"^圣遗物评分(.*)"), block=True)
|
|
||||||
@error_callable
|
|
||||||
@restricts(return_data=ConversationHandler.END)
|
|
||||||
async def command_start(self, update: Update, context: CallbackContext) -> int:
|
|
||||||
message = update.effective_message
|
|
||||||
user = update.effective_user
|
|
||||||
logger.info(f"用户 {user.full_name}[{user.id}] 圣遗物评分命令请求")
|
|
||||||
context.user_data["artifact_attr"] = None
|
|
||||||
photo_file: Optional[File] = None
|
|
||||||
if message is None:
|
|
||||||
return ConversationHandler.END
|
|
||||||
message_data = message if message.reply_to_message is None else message.reply_to_message
|
|
||||||
|
|
||||||
if message_data.photo is not None and len(message_data.photo) >= 1:
|
|
||||||
photo_file = await message_data.photo[-1].get_file() # 草 居然第一张是预览图我人都麻了
|
|
||||||
elif message_data.document is not None:
|
|
||||||
document = message_data.document
|
|
||||||
if "image" not in document.mime_type:
|
|
||||||
await message.reply_text("错误的图片类型")
|
|
||||||
return ConversationHandler.END
|
|
||||||
if document.file_size >= 5242880:
|
|
||||||
await message.reply_text("图片太大啦")
|
|
||||||
return ConversationHandler.END
|
|
||||||
photo_file = await document.get_file()
|
|
||||||
if photo_file is None:
|
|
||||||
await message.reply_text("图呢？")
|
|
||||||
return ConversationHandler.END
|
|
||||||
photo_byte = await photo_file.download_as_bytearray()
|
|
||||||
artifact_attr_req = await self.artifact_rate.get_artifact_attr(photo_byte)
|
|
||||||
if artifact_attr_req.status_code != 200:
|
|
||||||
if artifact_attr_req.status_code == 400:
|
|
||||||
artifact_attr = artifact_attr_req.json()
|
|
||||||
await message.reply_text(artifact_attr.get("message", "API请求错误，请稍后再试"))
|
|
||||||
return ConversationHandler.END
|
|
||||||
await message.reply_text("API请求错误，请稍后再试")
|
|
||||||
return ConversationHandler.END
|
|
||||||
artifact_attr = artifact_attr_req.json()
|
|
||||||
context.user_data["artifact_attr"] = artifact_attr
|
|
||||||
if artifact_attr.get("star") is None:
|
|
||||||
await message.reply_text("无法识别圣遗物星级，请选择圣遗物星级", reply_markup=InlineKeyboardMarkup(self.STAR_KEYBOARD))
|
|
||||||
return COMMAND_RESULT
|
|
||||||
if artifact_attr.get("level") is None:
|
|
||||||
await message.reply_text("无法识别圣遗物等级，请选择圣遗物等级", reply_markup=InlineKeyboardMarkup(self.LEVEL_KEYBOARD))
|
|
||||||
return COMMAND_RESULT
|
|
||||||
reply_message = await message.reply_text("识图成功！\n" "正在评分中...")
|
|
||||||
rate_text = await self.get_rate(artifact_attr)
|
|
||||||
await reply_message.edit_text(rate_text, parse_mode=ParseMode.MARKDOWN_V2)
|
|
||||||
return ConversationHandler.END
|
|
||||||
|
|
||||||
@conversation.state(state=COMMAND_RESULT)
|
|
||||||
@handler.callback_query()
|
|
||||||
@error_callable
|
|
||||||
async def command_result(self, update: Update, context: CallbackContext) -> int:
|
|
||||||
query = update.callback_query
|
|
||||||
artifact_attr = context.user_data.get("artifact_attr")
|
|
||||||
await query.answer()
|
|
||||||
if artifact_attr is None:
|
|
||||||
await query.edit_message_text("数据错误，请重新发送图片")
|
|
||||||
return ConversationHandler.END
|
|
||||||
|
|
||||||
def get_callback_data(callback_query_data: str) -> Tuple[str, int]:
|
|
||||||
_data = callback_query_data.split("|")
|
|
||||||
_key_name = _data[1]
|
|
||||||
try:
|
|
||||||
_value = int(_data[2])
|
|
||||||
except ValueError:
|
|
||||||
_value = -1
|
|
||||||
return _key_name, _value
|
|
||||||
|
|
||||||
await query.message.reply_chat_action(ChatAction.TYPING)
|
|
||||||
key_name, value = get_callback_data(query.data)
|
|
||||||
if key_name == "level":
|
|
||||||
artifact_attr["level"] = value
|
|
||||||
elif key_name == "star":
|
|
||||||
artifact_attr["star"] = value
|
|
||||||
else:
|
|
||||||
await query.edit_message_text("数据错误，请重新发送图片")
|
|
||||||
return ConversationHandler.END
|
|
||||||
if artifact_attr.get("level") is None:
|
|
||||||
await query.edit_message_text("无法识别圣遗物等级，请选择圣遗物等级", reply_markup=InlineKeyboardMarkup(self.LEVEL_KEYBOARD))
|
|
||||||
return COMMAND_RESULT
|
|
||||||
if artifact_attr.get("star") is None:
|
|
||||||
await query.edit_message_text("无法识别圣遗物星级，请选择圣遗物星级", reply_markup=InlineKeyboardMarkup(self.STAR_KEYBOARD))
|
|
||||||
return COMMAND_RESULT
|
|
||||||
await query.edit_message_text("正在评分中...")
|
|
||||||
rate_text = await self.get_rate(artifact_attr)
|
|
||||||
await query.edit_message_text(rate_text, parse_mode=ParseMode.MARKDOWN_V2)
|
|
||||||
return ConversationHandler.END
|
|
||||||
@ -1,48 +0,0 @@
|
|||||||
import logging
|
|
||||||
|
|
||||||
import aiofiles
|
|
||||||
import pytest
|
|
||||||
import pytest_asyncio
|
|
||||||
from flaky import flaky
|
|
||||||
|
|
||||||
from modules.apihelper.artifact import ArtifactOcrRate
|
|
||||||
|
|
||||||
LOGGER = logging.getLogger(__name__)
|
|
||||||
|
|
||||||
|
|
||||||
@pytest_asyncio.fixture
|
|
||||||
async def artifact_rate():
|
|
||||||
_artifact_rate = ArtifactOcrRate()
|
|
||||||
yield _artifact_rate
|
|
||||||
await _artifact_rate.close()
|
|
||||||
|
|
||||||
|
|
||||||
# noinspection PyShadowingNames
|
|
||||||
@pytest.mark.asyncio
|
|
||||||
class TestArtifactOcrRate:
|
|
||||||
@staticmethod
|
|
||||||
@flaky(3, 1)
|
|
||||||
async def test_rate_artifact(artifact_rate):
|
|
||||||
artifact_attr = {
|
|
||||||
"name": "翠绿的猎人之冠",
|
|
||||||
"pos": "理之冠",
|
|
||||||
"star": 5,
|
|
||||||
"level": 20,
|
|
||||||
"main_item": {"type": "cr", "name": "暴击率", "value": "31.1%"},
|
|
||||||
"sub_item": [
|
|
||||||
{"type": "hp", "name": "生命值", "value": "9.3%"},
|
|
||||||
{"type": "df", "name": "防御力", "value": "46"},
|
|
||||||
{"type": "atk", "name": "攻击力", "value": "49"},
|
|
||||||
{"type": "cd", "name": "暴击伤害", "value": "10.9%"},
|
|
||||||
],
|
|
||||||
}
|
|
||||||
assert await artifact_rate.rate_artifact(artifact_attr)
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
@flaky(3, 1)
|
|
||||||
async def test_ocr_artifact(artifact_rate):
|
|
||||||
async with aiofiles.open("tests/data/test_artifact.jpg", mode="rb") as f:
|
|
||||||
photo = await f.read()
|
|
||||||
data = await artifact_rate.get_artifact_attr(photo)
|
|
||||||
LOGGER.info(data.text)
|
|
||||||
assert data.status_code == 200
|
|
||||||
Loading…
Reference in New Issue
Block a user