mirror of
https://github.com/xuthus83/LittlePaimon.git
synced 2024-10-21 16:27:15 +08:00
240 lines
9.3 KiB
Python
240 lines
9.3 KiB
Python
from utils.auth_util import get_headers, get_sign_headers, get_use_cookie, get_own_cookie, check_retcode
|
||
from utils.db_util import update_cookie_cache
|
||
from utils.decorator import cache
|
||
from utils import aiorequests
|
||
import datetime
|
||
import re
|
||
from asyncio import sleep
|
||
|
||
|
||
@cache(ttl=datetime.timedelta(minutes=10))
|
||
async def get_abyss_data(user_id, uid, schedule_type="1", use_cache=True):
|
||
server_id = "cn_qd01" if uid[0] == '5' else "cn_gf01"
|
||
url = "https://api-takumi-record.mihoyo.com/game_record/app/genshin/api/spiralAbyss"
|
||
params = {
|
||
"schedule_type": schedule_type,
|
||
"role_id": uid,
|
||
"server": server_id}
|
||
while True:
|
||
cookie = await get_use_cookie(user_id, uid=uid, action='查询深渊')
|
||
if not cookie:
|
||
return '现在派蒙没有可以用的cookie哦，可能是:\n1.公共cookie全都达到了每日30次上限\n2.公共池全都失效了或没有cookie\n让管理员使用 添加公共ck 吧!'
|
||
headers = get_headers(q=f'role_id={uid}&schedule_type={schedule_type}&server={server_id}',
|
||
cookie=cookie['cookie'])
|
||
|
||
resp = await aiorequests.get(url=url, headers=headers, params=params)
|
||
data = resp.json()
|
||
check = await check_retcode(data, cookie, uid)
|
||
if check == '私人cookie达到了每日30次查询上限':
|
||
return check
|
||
elif check:
|
||
return data
|
||
|
||
|
||
async def get_daily_note_data(uid):
|
||
server_id = "cn_qd01" if uid[0] == '5' else "cn_gf01"
|
||
url = "https://api-takumi-record.mihoyo.com/game_record/app/genshin/api/dailyNote"
|
||
cookie = await get_own_cookie(uid, action='查询实时便签')
|
||
if not cookie:
|
||
return f'你的uid{uid}没有绑定对应的cookie,使用ysb绑定才能用实时便签哦!'
|
||
await update_cookie_cache(cookie['cookie'], uid, 'uid')
|
||
headers = get_headers(q=f'role_id={uid}&server={server_id}', cookie=cookie['cookie'])
|
||
params = {
|
||
"server": server_id,
|
||
"role_id": uid
|
||
}
|
||
resp = await aiorequests.get(url=url, headers=headers, params=params)
|
||
data = resp.json()
|
||
if await check_retcode(data, cookie, uid):
|
||
return data
|
||
else:
|
||
return f'你的uid{uid}的cookie已过期,需要重新绑定哦!'
|
||
|
||
|
||
@cache(ttl=datetime.timedelta(hours=1))
|
||
async def get_player_card_data(user_id, uid, use_cache=True):
|
||
server_id = "cn_qd01" if uid[0] == '5' else "cn_gf01"
|
||
url = "https://api-takumi-record.mihoyo.com/game_record/app/genshin/api/index"
|
||
params = {
|
||
"server": server_id,
|
||
"role_id": uid
|
||
}
|
||
while True:
|
||
cookie = await get_use_cookie(user_id, uid=uid, action='查询原神卡片')
|
||
if not cookie:
|
||
return '现在派蒙没有可以用的cookie哦，可能是:\n1.公共cookie全都达到了每日30次上限\n2.公共池全都失效了或没有cookie\n让管理员使用 添加公共ck 吧!'
|
||
headers = get_headers(q=f'role_id={uid}&server={server_id}', cookie=cookie['cookie'])
|
||
resp = await aiorequests.get(url=url, headers=headers, params=params)
|
||
data = resp.json()
|
||
check = await check_retcode(data, cookie, uid)
|
||
if check == '私人cookie达到了每日30次查询上限':
|
||
return check
|
||
elif check:
|
||
return data
|
||
|
||
|
||
@cache(ttl=datetime.timedelta(hours=1))
|
||
async def get_chara_detail_data(user_id, uid, use_cache=True):
|
||
server_id = "cn_qd01" if uid[0] == '5' else "cn_gf01"
|
||
json_data = {
|
||
"server": server_id,
|
||
"role_id": uid,
|
||
"character_ids": []
|
||
}
|
||
url = 'https://api-takumi-record.mihoyo.com/game_record/app/genshin/api/character'
|
||
while True:
|
||
cookie = await get_use_cookie(user_id, uid=uid, action='查询角色详情')
|
||
if not cookie:
|
||
return '现在派蒙没有可以用的cookie哦，可能是:\n1.公共cookie全都达到了每日30次上限\n2.公共池全都失效了或没有cookie\n让管理员使用 添加公共ck 吧!'
|
||
headers = get_headers(b=json_data, cookie=cookie['cookie'])
|
||
resp = await aiorequests.post(url=url, headers=headers, json=json_data)
|
||
data = resp.json()
|
||
check = await check_retcode(data, cookie, uid)
|
||
if check == '私人cookie达到了每日30次查询上限':
|
||
return check
|
||
elif check:
|
||
return data
|
||
|
||
|
||
@cache(ttl=datetime.timedelta(hours=1))
|
||
async def get_chara_skill_data(uid, chara_id, use_cache=True):
|
||
server_id = "cn_qd01" if uid[0] == '5' else "cn_gf01"
|
||
url = 'https://api-takumi.mihoyo.com/event/e20200928calculate/v1/sync/avatar/detail'
|
||
cookie = await get_own_cookie(uid, action='查询角色天赋')
|
||
if not cookie:
|
||
return None
|
||
await update_cookie_cache(cookie['cookie'], uid, 'uid')
|
||
headers = get_headers(q=f'uid={uid}®ion={server_id}&avatar_id={chara_id}', cookie=cookie['cookie'])
|
||
params = {
|
||
"region": server_id,
|
||
"uid": uid,
|
||
"avatar_id": chara_id
|
||
}
|
||
resp = await aiorequests.get(url=url, headers=headers, params=params)
|
||
data = resp.json()
|
||
return data
|
||
|
||
|
||
@cache(ttl=datetime.timedelta(hours=1))
|
||
async def get_monthinfo_data(uid, month, use_cache=True):
|
||
server_id = "cn_qd01" if uid[0] == '5' else "cn_gf01"
|
||
url = 'https://hk4e-api.mihoyo.com/event/ys_ledger/monthInfo'
|
||
cookie = await get_own_cookie(uid, action='查询每月札记')
|
||
if not cookie:
|
||
return f'你的uid{uid}没有绑定对应的cookie,使用ysb绑定才能用每月札记哦!'
|
||
await update_cookie_cache(cookie['cookie'], uid, 'uid')
|
||
headers = get_headers(q=f'month={month}&bind_uid={uid}&bind_region={server_id}', cookie=cookie['cookie'])
|
||
params = {
|
||
"month": int(month),
|
||
"bind_uid": uid,
|
||
"bind_region": server_id
|
||
}
|
||
resp = await aiorequests.get(url=url, headers=headers, params=params)
|
||
data = resp.json()
|
||
if await check_retcode(data, cookie, uid):
|
||
return data
|
||
else:
|
||
return f'你的uid{uid}的cookie已过期,需要重新绑定哦!'
|
||
|
||
|
||
async def get_bind_game(cookie):
|
||
finduid = re.search(r'account_id=(\d{6,12})', cookie)
|
||
if not finduid:
|
||
finduid = re.search(r'ltuid=(\d{6,12})', cookie)
|
||
if not finduid:
|
||
return None, None
|
||
uid = finduid.group(1)
|
||
url = 'https://api-takumi-record.mihoyo.com/game_record/card/wapi/getGameRecordCard'
|
||
headers = get_headers(q=f'uid={uid}', cookie=cookie)
|
||
params = {
|
||
"uid": uid
|
||
}
|
||
resp = await aiorequests.get(url=url, headers=headers, params=params)
|
||
data = resp.json()
|
||
return data, uid
|
||
|
||
|
||
# 获取今日签到信息
|
||
async def get_sign_info(uid):
|
||
server_id = "cn_qd01" if uid[0] == '5' else "cn_gf01"
|
||
url = 'https://api-takumi.mihoyo.com/event/bbs_sign_reward/info'
|
||
cookie = await get_own_cookie(uid, action='查询米游社签到')
|
||
if not cookie:
|
||
return f'你的uid{uid}没有绑定对应的cookie,使用ysb绑定才能用米游社签到哦!'
|
||
headers = {
|
||
'x-rpc-app_version': '2.11.1',
|
||
'x-rpc-client_type': '5',
|
||
'Origin': 'https://webstatic.mihoyo.com',
|
||
'Referer': 'https://webstatic.mihoyo.com/',
|
||
'Cookie': cookie['cookie'],
|
||
'User-Agent': 'Mozilla/5.0 (iPhone; CPU iPhone OS 13_2_3 like Mac OS '
|
||
'X) AppleWebKit/605.1.15 (KHTML, like Gecko) miHoYoBBS/2.11.1',
|
||
|
||
}
|
||
params = {
|
||
'act_id': 'e202009291139501',
|
||
'region': server_id,
|
||
'uid': uid
|
||
}
|
||
resp = await aiorequests.get(url=url, headers=headers, params=params)
|
||
data = resp.json()
|
||
if await check_retcode(data, cookie, uid):
|
||
return data
|
||
else:
|
||
return f'你的uid{uid}的cookie已过期,需要重新绑定哦!'
|
||
|
||
|
||
# 执行签到操作
|
||
async def sign(uid):
|
||
server_id = "cn_qd01" if uid[0] == '5' else "cn_gf01"
|
||
url = 'https://api-takumi.mihoyo.com/event/bbs_sign_reward/sign'
|
||
cookie = await get_own_cookie(uid, action='米游社签到')
|
||
if not cookie:
|
||
return f'你的uid{uid}没有绑定对应的cookie,使用ysb绑定才能用米游社签到哦!'
|
||
headers = get_sign_headers(cookie['cookie'])
|
||
json_data = {
|
||
'act_id': 'e202009291139501',
|
||
'uid': uid,
|
||
'region': server_id
|
||
}
|
||
resp = await aiorequests.post(url=url, headers=headers, json=json_data)
|
||
try:
|
||
data = resp.json()
|
||
except:
|
||
return resp.read()
|
||
if await check_retcode(data, cookie, uid):
|
||
return data
|
||
else:
|
||
return f'你的uid{uid}的cookie已过期,需要重新绑定哦!'
|
||
|
||
|
||
# 获取签到奖励列表
|
||
async def get_sign_list():
|
||
url = 'https://api-takumi.mihoyo.com/event/bbs_sign_reward/home'
|
||
headers = {
|
||
'x-rpc-app_version': '2.11.1',
|
||
'User-Agent': 'Mozilla/5.0 (iPhone; CPU iPhone OS 13_2_3 like Mac OS X) AppleWebKit/605.1.15 ('
|
||
'KHTML, like Gecko) miHoYoBBS/2.11.1',
|
||
'x-rpc-client_type': '5',
|
||
'Referer': 'https://webstatic.mihoyo.com/'
|
||
}
|
||
params = {
|
||
'act_id': 'e202009291139501'
|
||
}
|
||
resp = await aiorequests.get(url=url, headers=headers, params=params)
|
||
data = resp.json()
|
||
return data
|
||
|
||
|
||
async def get_enka_data(uid):
|
||
for _ in range(3):
|
||
try:
|
||
url = f'https://enka.shinshin.moe/u/{uid}/__data.json'
|
||
resp = await aiorequests.get(url=url)
|
||
data = resp.json()
|
||
return data
|
||
except Exception:
|
||
await sleep(1.5)
|
||
|
||
|