87 lines
3.8 KiB
Python
87 lines
3.8 KiB
Python
import os.path
|
|
import shutil
|
|
import traceback
|
|
|
|
from core.builtins import Bot, Image
|
|
from core.component import module
|
|
from core.utils.cache import random_cache_path
|
|
from core.utils.http import get_url, download_to_cache
|
|
from .dbutils import PgrBindInfoManager
|
|
from .game_record import parse_game_record
|
|
from .genb19 import drawb19
|
|
from .update import update_assets, p_headers
|
|
|
|
phi = module('phigros', developers=['OasisAkari'], desc='{phigros.help.desc}',
|
|
alias={'p': 'phigros', 'pgr': 'phigros', 'phi': 'phigros'})
|
|
|
|
|
|
@phi.command('bind <sessiontoken> {{phigros.help.bind}}')
|
|
async def _(msg: Bot.MessageSession, sessiontoken: str):
|
|
need_revoke = False
|
|
send_msg = []
|
|
if msg.target.target_from in [
|
|
'QQ|Group',
|
|
'QQ|Guild',
|
|
'Discord|Channel',
|
|
'Telegram|group',
|
|
'Telegram|supergroup',
|
|
'Kook|GROUP']:
|
|
send_msg.append(await msg.send_message(msg.locale.t("phigros.message.bind.warning"), quote=False))
|
|
need_revoke = True
|
|
headers = p_headers.copy()
|
|
headers['X-LC-Session'] = sessiontoken
|
|
get_user_info = await get_url('https://rak3ffdi.cloud.tds1.tapapis.cn/1.1/users/me', headers=headers, fmt='json')
|
|
if 'nickname' in get_user_info:
|
|
bind = PgrBindInfoManager(msg).set_bind_info(sessiontoken=sessiontoken, username=get_user_info['nickname'])
|
|
if bind:
|
|
send_msg.append(await msg.send_message(msg.locale.t("phigros.message.bind.success",
|
|
username=get_user_info['nickname']), quote=False))
|
|
if need_revoke:
|
|
await msg.sleep(15)
|
|
for i in send_msg:
|
|
await i.delete()
|
|
|
|
|
|
@phi.command('unbind {{phigros.help.unbind}}')
|
|
async def _(msg: Bot.MessageSession):
|
|
if PgrBindInfoManager(msg).remove_bind_info():
|
|
await msg.finish(msg.locale.t("phigros.message.unbind.success"))
|
|
|
|
|
|
@phi.command('b19 {{phigros.help.b19}}')
|
|
async def _(msg: Bot.MessageSession):
|
|
if (bind := PgrBindInfoManager(msg).get_bind_info()) is None:
|
|
await msg.finish(msg.locale.t("phigros.message.user_unbound", prefix=msg.prefixes[0]))
|
|
else:
|
|
try:
|
|
headers = p_headers.copy()
|
|
headers['X-LC-Session'] = bind[0]
|
|
get_save_url = await get_url('https://rak3ffdi.cloud.tds1.tapapis.cn/1.1/classes/_GameSave',
|
|
headers=headers,
|
|
fmt='json')
|
|
save_url = get_save_url['results'][0]['gameFile']['url']
|
|
download = await download_to_cache(save_url)
|
|
rd_path = random_cache_path()
|
|
shutil.unpack_archive(download, rd_path)
|
|
game_records = parse_game_record(os.path.join(rd_path, 'gameRecord'))
|
|
sort_by_rks = sorted({f'{level}.{song}':
|
|
game_records[song][level] for song in game_records for level in
|
|
game_records[song]}.items(),
|
|
key=lambda x: x[1]['rks'], reverse=True)
|
|
phi_list = [s for s in sort_by_rks if s[1]['score'] == 1000000]
|
|
b19_data = ([sorted(phi_list, key=lambda x: x[1]['rks'], reverse=True)[0]]
|
|
if phi_list else []) + sort_by_rks[0: 19]
|
|
if len(rks_acc := [i[1]['rks'] for i in b19_data]) < 20:
|
|
rks_acc += [0] * (20 - len(rks_acc))
|
|
await msg.send_message(Image(drawb19(bind[1], round(sum(rks_acc) / len(rks_acc), 2), b19_data)))
|
|
except Exception as e:
|
|
traceback.print_exc()
|
|
await msg.send_message(msg.locale.t("phigros.message.b19.get_failed", err=str(e)))
|
|
|
|
|
|
@phi.command('update', required_superuser=True)
|
|
async def _(msg: Bot.MessageSession):
|
|
if await update_assets():
|
|
await msg.finish(msg.locale.t("success"))
|
|
else:
|
|
await msg.finish(msg.locale.t("failed"))
|