mirror of
https://github.com/xuthus83/LittlePaimon.git
synced 2024-10-21 16:27:15 +08:00
67 lines
2.0 KiB
Python
67 lines
2.0 KiB
Python
import json
|
||
from urllib import parse
|
||
from aiohttp import ClientSession
|
||
|
||
|
||
def toApi(url):
|
||
spliturl = str(url).split("?")
|
||
if "webstatic-sea" in spliturl[0] or "hk4e-api-os" in spliturl[0]:
|
||
spliturl[0] = "https://hk4e-api-os.mihoyo.com/event/gacha_info/api/getGachaLog"
|
||
else:
|
||
spliturl[0] = "https://hk4e-api.mihoyo.com/event/gacha_info/api/getGachaLog"
|
||
url = "?".join(spliturl)
|
||
return url
|
||
|
||
|
||
def getApi(url, gachaType, size, page, end_id=""):
|
||
parsed = parse.urlparse(url)
|
||
querys = parse.parse_qsl(str(parsed.query))
|
||
param_dict = dict(querys)
|
||
param_dict["size"] = size
|
||
param_dict["gacha_type"] = gachaType
|
||
param_dict["page"] = page
|
||
param_dict["lang"] = "zh-cn"
|
||
param_dict["end_id"] = end_id
|
||
param = parse.urlencode(param_dict)
|
||
path = str(url).split("?")[0]
|
||
api = path + "?" + param
|
||
return api
|
||
|
||
|
||
async def checkApi(url):
|
||
try:
|
||
async with ClientSession() as session:
|
||
r = await session.get(url)
|
||
j = await r.json()
|
||
except Exception as e:
|
||
return f'API请求解析出错：{e}'
|
||
|
||
if not j["data"]:
|
||
if j["message"] == "authkey error":
|
||
return "authkey错误，请重新获取链接给派蒙！"
|
||
elif j["message"] == "authkey timeout":
|
||
return "authkey已过期，请重新获取链接给派蒙！"
|
||
else:
|
||
return f'数据为空，错误代码：{j["message"]}'
|
||
return 'OK'
|
||
|
||
|
||
def getQueryVariable(url, variable):
|
||
query = str(url).split("?")[1]
|
||
vars = query.split("&")
|
||
for v in vars:
|
||
if v.split("=")[0] == variable:
|
||
return v.split("=")[1]
|
||
return ""
|
||
|
||
|
||
async def getGachaInfo(url):
|
||
region = getQueryVariable(url, "region")
|
||
lang = getQueryVariable(url, "lang")
|
||
gachaInfoUrl = "https://webstatic.mihoyo.com/hk4e/gacha_info/{}/items/{}.json".format(region, lang)
|
||
async with ClientSession() as session:
|
||
r = await session.get(gachaInfoUrl)
|
||
s = (await r.read()).decode("utf-8")
|
||
gachaInfo = json.loads(s)
|
||
return gachaInfo
|