季落 008c066828
深渊查询完善 (#38)
* 深渊查询完善

* 🚨 `pre-commit-ci`修复格式错误

* 深渊查询完善

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-05-18 22:27:26 +08:00

335 lines
11 KiB
Python

import copy
import time
import uuid
import random
import asyncio
from string import digits, ascii_letters
from typing import Dict, Union, Optional, cast
from gsuid_core.utils.api.mys_api import _MysApi
from gsuid_core.utils.api.mys.models import MysSign, SignInfo, SignList
from gsuid_core.utils.api.mys.tools import (
random_hex,
generate_os_ds,
get_web_ds_token,
)
from .api import srdbsqla
from ..sruid_utils.api.mys.api import _API
from ..sruid_utils.api.mys.models import (
GachaLog,
AbyssData,
RoleIndex,
AvatarInfo,
MonthlyAward,
DailyNoteData,
RoleBasicInfo,
)
RECOGNIZE_SERVER = {
'1': 'prod_gf_cn',
'2': 'prod_gf_cn',
# '5': 'cn_qd01',
# '6': 'os_usa',
# '7': 'os_euro',
# '8': 'os_asia',
# '9': 'os_cht',
}
class MysApi(_MysApi):
device_id = uuid.uuid4().hex
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
asyncio.run(self.get_fp())
self._HEADER['x-rpc-device_id'] = self.device_id
self._HEADER['x-rpc-page'] = '3.1.3_#/rpg'
async def create_qrcode_url(self) -> Union[Dict, int]:
device_id: str = ''.join(random.choices(ascii_letters + digits, k=64))
app_id: str = '8'
data = await self._mys_request(
_API['CREATE_QRCODE'],
'POST',
header={},
data={'app_id': app_id, 'device': device_id},
)
if isinstance(data, Dict):
url: str = data['data']['url']
ticket = url.split('ticket=')[1]
return {
'app_id': app_id,
'ticket': ticket,
'device': device_id,
'url': url,
}
return data
async def get_daily_data(self, uid: str) -> Union[DailyNoteData, int]:
data = await self.simple_mys_req(
'STAR_RAIL_NOTE_URL', uid, header=self._HEADER
)
if isinstance(data, Dict):
data = cast(DailyNoteData, data['data'])
return data
async def get_role_index(self, uid: str) -> Union[RoleIndex, int]:
data = await self.simple_mys_req(
'STAR_RAIL_INDEX_URL', uid, header=self._HEADER
)
if isinstance(data, Dict):
data = cast(RoleIndex, data['data'])
return data
async def get_gacha_log_by_link_in_authkey(
self,
authkey: str,
gacha_type: str = '11',
page: int = 1,
end_id: str = '0',
) -> Union[int, GachaLog]:
# server_id = 'cn_qd01' if
# uid[0] == '5' else 'cn_gf01'
server_id = 'prod_gf_cn'
data = await self._mys_request(
url=self.MAPI['STAR_RAIL_GACHA_LOG_URL'],
method='GET',
header=self._HEADER,
params={
'authkey_ver': '1',
'sign_type': '2',
'auth_appid': 'webview_gacha',
'default_gacha_type': 11,
'gacha_id': 'dbebc8d9fbb0d4ffa067423482ce505bc5ea',
'timestamp': str(int(time.time())),
'lang': 'zh-cn',
'plat_type': 'pc',
'region': server_id,
'authkey': authkey,
'game_biz': 'hkrpg_cn',
'gacha_type': gacha_type,
'page': page,
'size': '5',
'end_id': end_id,
},
)
if isinstance(data, Dict):
data = cast(GachaLog, data['data'])
return data
async def get_avatar_info(
self, uid: str, avatar_id: int, need_wiki: bool = False
) -> Union[AvatarInfo, int]:
data = await self.simple_mys_req(
'STAR_RAIL_AVATAR_INFO_URL',
uid,
params={
'id': avatar_id,
'need_wiki': 'true' if need_wiki else 'false',
'role_id': uid,
'server': RECOGNIZE_SERVER.get(str(uid)[0], 'prod_gf_cn'),
},
header=self._HEADER,
)
if isinstance(data, Dict):
data = cast(AvatarInfo, data['data'])
return data
async def get_sign_list(self, uid) -> Union[SignList, int]:
# is_os = self.check_os(uid)
is_os = False
if is_os:
params = {
'act_id': 'e202304121516551',
'lang': 'zh-cn',
}
else:
params = {
'act_id': 'e202304121516551',
'lang': 'zh-cn',
}
data = await self._mys_req_get(
'STAR_RAIL_SIGN_LIST_URL', is_os, params
)
if isinstance(data, Dict):
data = cast(SignList, data['data'])
return data
async def get_sign_info(self, uid) -> Union[SignInfo, int]:
# server_id = RECOGNIZE_SERVER.get(str(uid)[0])
# is_os = self.check_os(uid)
is_os = False
if is_os:
# TODO
params = {
'act_id': 'e202304121516551',
'lang': 'zh-cn',
'region': 'prod_gf_cn',
'uid': uid,
}
header = {
'DS': generate_os_ds(),
}
else:
params = {
'act_id': 'e202304121516551',
'lang': 'zh-cn',
'region': 'prod_gf_cn',
'uid': uid,
}
header = self._HEADER
data = await self._mys_req_get(
'STAR_RAIL_SIGN_INFO_URL', is_os, params, header
)
if isinstance(data, Dict):
data = cast(SignInfo, data['data'])
return data
async def get_srspiral_abyss_info(
self,
uid: str,
schedule_type='1',
ck: Optional[str] = None,
) -> Union[AbyssData, int]:
server_id = self.RECOGNIZE_SERVER.get(uid[0])
data = await self.simple_mys_req(
'CHALLENGE_INFO_URL',
uid,
params={
'isPrev': 'true',
'need_all': 'true',
'role_id': uid,
'schedule_type': schedule_type,
'server': server_id,
},
cookie=ck,
header=self._HEADER,
)
# print(data)
if isinstance(data, Dict):
data = cast(AbyssData, data['data'])
return data
async def mys_sign(
self, uid, header={}, server_id='cn_gf01'
) -> Union[MysSign, int]:
ck = await self.get_ck(uid, 'OWNER')
if ck is None:
return -51
if int(str(uid)[0]) < 6:
HEADER = copy.deepcopy(self._HEADER)
HEADER['Cookie'] = ck
HEADER['x-rpc-app_version'] = '2.44.1'
HEADER['x-rpc-client_type'] = '5'
HEADER['X_Requested_With'] = 'com.mihoyo.hyperion'
HEADER['DS'] = get_web_ds_token(True)
HEADER['Referer'] = 'https://webstatic.mihoyo.com'
HEADER.update(header)
data = await self._mys_request(
url=_API['STAR_RAIL_SIGN_URL'],
method='POST',
header=HEADER,
data={
'act_id': 'e202304121516551',
'region': 'prod_gf_cn',
'uid': uid,
'lang': 'zh-cn',
},
)
else:
data = -99
if isinstance(data, Dict):
data = cast(MysSign, data['data'])
return data
async def get_award(self, sr_uid, month) -> Union[MonthlyAward, int]:
server_id = RECOGNIZE_SERVER.get(str(sr_uid)[0])
ck = await self.get_ck(sr_uid, 'OWNER')
if ck is None:
return -51
if int(str(sr_uid)[0]) < 6:
HEADER = copy.deepcopy(self._HEADER)
HEADER['Cookie'] = ck
HEADER['DS'] = get_web_ds_token(True)
data = await self._mys_request(
url=_API['STAR_RAIL_MONTH_INFO_URL'],
method='GET',
header=HEADER,
params={'uid': sr_uid, 'region': server_id, 'month': month},
)
else:
HEADER = copy.deepcopy(self._HEADER_OS)
HEADER['Cookie'] = ck
HEADER['DS'] = generate_os_ds()
data = await self._mys_request(
url=_API['STAR_RAIL_MONTH_INFO_URL'],
method='GET',
header=HEADER,
params={'uid': sr_uid, 'region': server_id, 'month': month},
use_proxy=True,
)
if isinstance(data, Dict):
data = cast(MonthlyAward, data['data'])
return data
async def get_role_basic_info(
self, sr_uid: str
) -> Union[RoleBasicInfo, int]:
data = await self.simple_mys_req(
'STAR_RAIL_ROLE_BASIC_INFO_URL', sr_uid, header=self._HEADER
)
if isinstance(data, Dict):
data = cast(RoleBasicInfo, data['data'])
return data
def generate_seed(self, length: int):
characters = '0123456789abcdef'
result = ''.join(random.choices(characters, k=length))
return result
async def get_fp(self):
seed_id = self.generate_seed(16)
seed_time = str(int(time.time() * 1000))
ext_fields = f'{{"userAgent":"{self._HEADER["User-Agent"]}",\
"browserScreenSize":281520,"maxTouchPoints":5,\
"isTouchSupported":true,"browserLanguage":"zh-CN","browserPlat":"iPhone",\
"browserTimeZone":"Asia/Shanghai","webGlRender":"Apple GPU",\
"webGlVendor":"Apple Inc.",\
"numOfPlugins":0,"listOfPlugins":"unknown","screenRatio":3,"deviceMemory":"unknown",\
"hardwareConcurrency":"4","cpuClass":"unknown","ifNotTrack":"unknown","ifAdBlock":0,\
"hasLiedResolution":1,"hasLiedOs":0,"hasLiedBrowser":0}}'
body = {
'seed_id': seed_id,
'device_id': self.device_id,
'platform': '5',
'seed_time': seed_time,
'ext_fields': ext_fields,
'app_name': 'account_cn',
'device_fp': '38d7ee834d1e9',
}
HEADER = copy.deepcopy(self._HEADER)
res = await self._mys_request(
url=_API['GET_FP_URL'],
method='POST',
header=HEADER,
data=body,
)
if res["retcode"] != 0:
print("获取fp连接失败")
print(res)
self._HEADER['x-rpc-device_fp'] = random_hex(13).lower()
elif res["data"]["code"] != 200:
print("获取fp参数不正确")
print(res["data"]["msg"])
self._HEADER['x-rpc-device_fp'] = random_hex(13).lower()
else:
self._HEADER['x-rpc-device_fp'] = res["data"]["device_fp"]
mys_api = MysApi()
mys_api.dbsqla = srdbsqla
mys_api.MAPI.update(_API)
mys_api.is_sr = True
mys_api.RECOGNIZE_SERVER = RECOGNIZE_SERVER