61 lines
2.2 KiB
Python
61 lines
2.2 KiB
Python
from io import BytesIO
|
||
from os import sep
|
||
import requests, yaml, re
|
||
from json.decoder import JSONDecodeError
|
||
from PIL import Image
|
||
from defs.character import repl
|
||
from defs.weapons import headers
|
||
|
||
|
||
def get_url(name: str):
|
||
res = requests.get(url=f'https://api.minigg.cn/artifacts?query={name}', headers=headers)
|
||
if res.text == "undefined\n":
|
||
raise JSONDecodeError("", "", 0)
|
||
py_dict = yaml.safe_load(re.sub(r'\[? *(, *)+\]?', repl, res.text))
|
||
return py_dict
|
||
|
||
|
||
def gen_artifacts(data: dict):
|
||
base_img = Image.new(mode="RGBA", size=(1280, 256), color=(255, 255, 255))
|
||
for index, value in enumerate(data):
|
||
img = Image.open(BytesIO(requests.get(data[value]).content))
|
||
base_img.paste(img, (256 * index, 0))
|
||
|
||
jpg_img = Image.new('RGB', size=(1280, 256), color=(255, 255, 255))
|
||
jpg_img.paste(base_img, (0, 0), mask=base_img)
|
||
jpg_img.save(f'temp{sep}artifacts.jpg', format='JPEG', subsampling=0, quality=90)
|
||
|
||
|
||
async def get_artifacts(name: str):
|
||
artifacts_im = '''<b>{}</b>
|
||
【稀有度】：{}
|
||
【2件套】：{}
|
||
【4件套】：{}
|
||
【{}】：{}
|
||
【{}】：{}
|
||
【{}】：{}
|
||
【{}】：{}
|
||
【{}】：{}
|
||
'''
|
||
try:
|
||
data = get_url(name)
|
||
except JSONDecodeError:
|
||
return f"没有找到该武器,派蒙也米有办法！是不是名字错了？", None
|
||
try:
|
||
star = ""
|
||
for i in data["rarity"]:
|
||
star = star + i + "星、"
|
||
star = star[:-1]
|
||
im = artifacts_im.format(data["name"], star, data["2pc"], data["4pc"], data["flower"]["name"],
|
||
data["flower"]["description"],
|
||
data["plume"]["name"], data["plume"]["description"], data["sands"]["name"],
|
||
data["sands"]["description"],
|
||
data["goblet"]["name"], data["goblet"]["description"], data["circlet"]["name"],
|
||
data["circlet"]["description"])
|
||
gen_artifacts(data['images'])
|
||
return im, f'temp{sep}artifacts.jpg'
|
||
except KeyError:
|
||
return f"没有找到该武器,派蒙也米有办法！是不是名字错了？", None
|
||
except:
|
||
return f"没有找到该武器,派蒙也米有办法！是不是名字错了？", None
|