mirror of
https://github.com/KimigaiiWuyi/GenshinUID.git
synced 2025-05-31 04:30:29 +08:00
60 lines
2.2 KiB
Python
60 lines
2.2 KiB
Python
from typing import Optional
|
||
|
||
from ..db_operation.db_operation import cache_db, error_db
|
||
from ..mhy_api.get_mhy_data import get_info, get_spiral_abyss_info
|
||
|
||
|
||
class GetCookies:
|
||
def __init__(self) -> None:
|
||
self.useable_cookies: Optional[str] = None
|
||
self.uid: Optional[str] = None
|
||
self.raw_data: Optional[dict] = None
|
||
|
||
async def get_useable_cookies(self, uid: str):
|
||
self.uid = uid
|
||
while True:
|
||
self.useable_cookies = str(await cache_db(uid))
|
||
if self.useable_cookies == '':
|
||
return '绑定记录不存在。'
|
||
elif self.useable_cookies == '没有可以使用的Cookies！':
|
||
return '没有可以使用的Cookies！'
|
||
await self.get_uid_data()
|
||
msg = await self.check_cookies_useable()
|
||
if isinstance(msg, str):
|
||
return msg
|
||
else:
|
||
if msg:
|
||
return None
|
||
|
||
async def get_uid_data(self):
|
||
self.raw_data = await get_info(self.uid, self.useable_cookies)
|
||
|
||
async def get_spiral_abyss_data(self, schedule_type: str = '1'):
|
||
return await get_spiral_abyss_info(
|
||
self.uid, self.useable_cookies, schedule_type
|
||
)
|
||
|
||
async def check_cookies_useable(self):
|
||
if self.raw_data:
|
||
if self.raw_data['retcode'] != 0:
|
||
if self.raw_data['retcode'] == 10001:
|
||
if self.useable_cookies:
|
||
await error_db(self.useable_cookies, 'error')
|
||
return False
|
||
elif self.raw_data['retcode'] == 10101:
|
||
if self.useable_cookies:
|
||
await error_db(self.useable_cookies, 'limit30')
|
||
return False
|
||
elif self.raw_data['retcode'] == 10102:
|
||
return '当前查询id已经设置了隐私，无法查询！'
|
||
else:
|
||
return (
|
||
'Api报错，返回内容为：\r\n'
|
||
+ str(self.raw_data)
|
||
+ '\r\n出现这种情况可能的UID输入错误 or 不存在'
|
||
)
|
||
else:
|
||
return True
|
||
else:
|
||
return '没有可以使用的Cookies！'
|