89 lines
3.4 KiB
Python
89 lines
3.4 KiB
Python
import asyncio
|
|
import os
|
|
import traceback
|
|
from datetime import datetime
|
|
|
|
from config import Config
|
|
from core.builtins import Plain, Image
|
|
from core.logger import Logger
|
|
from core.utils.http import get_url
|
|
from modules.arcaea.utils import autofix_b30_song_background
|
|
|
|
assets_path = os.path.abspath('./assets/arcaea')
|
|
api_url = Config("botarcapi_url")
|
|
headers = {"Authorization": f'Bearer {Config("botarcapi_token")}'}
|
|
|
|
|
|
async def get_info(msg, usercode):
|
|
try:
|
|
get_ = await get_url(api_url + f"user/info?user_code={usercode}&recent=1&with_song_info=True",
|
|
status_code=200,
|
|
headers=headers,
|
|
fmt='json')
|
|
except ValueError as e:
|
|
return [Plain(msg.locale.t('arcaea.message.failed.errcode') + str(e))]
|
|
except Exception:
|
|
traceback.print_exc()
|
|
return [Plain(msg.locale.t('arcaea.message.failed'))]
|
|
Logger.debug(get_)
|
|
if get_["status"] == 0:
|
|
recent = get_['content']["recent_score"]
|
|
if len(recent) < 0:
|
|
return [Plain(msg.locale.t('arcaea.message.info.result.none'))]
|
|
recent = recent[0]
|
|
difficulty = '???'
|
|
if recent['difficulty'] == 0:
|
|
difficulty = 'PST'
|
|
elif recent['difficulty'] == 1:
|
|
difficulty = 'PRS'
|
|
elif recent['difficulty'] == 2:
|
|
difficulty = 'FTR'
|
|
elif recent['difficulty'] == 3:
|
|
difficulty = 'BYD'
|
|
songinfo = get_['content']['song_info'][0]
|
|
trackname = songinfo['name_en']
|
|
imgpath = f'{assets_path}/jacket/{recent["song_id"]}_{recent["difficulty"]}.jpg'
|
|
if not os.path.exists(imgpath):
|
|
imgpath = f'{assets_path}/jacket/{recent["song_id"]}.jpg'
|
|
realptt = songinfo['rating'] / 10
|
|
ptt = recent['rating']
|
|
score = recent['score']
|
|
shiny_pure = recent['shiny_perfect_count']
|
|
pure = recent['perfect_count']
|
|
far = recent['near_count']
|
|
lost = recent['miss_count']
|
|
username = get_['content']['account_info']['name']
|
|
usrptt = int(get_['content']['account_info']['rating'])
|
|
if usrptt == -1:
|
|
usrptt = '--'
|
|
else:
|
|
usrptt = usrptt / 100
|
|
time_played = datetime.fromtimestamp(recent['time_played'] / 1000)
|
|
result = [
|
|
Plain(
|
|
msg.locale.t(
|
|
'arcaea.message.info.result',
|
|
username=username,
|
|
potential=usrptt,
|
|
trackname=trackname,
|
|
difficulty=difficulty,
|
|
score=score,
|
|
pure=pure,
|
|
shiny_pure=shiny_pure,
|
|
far=far,
|
|
lost=lost,
|
|
realptt=realptt,
|
|
ptt=ptt,
|
|
time_played=time_played.strftime("%Y-%m-%d %H:%M:%S")))]
|
|
if os.path.exists(imgpath):
|
|
result.append(Image(imgpath))
|
|
else:
|
|
asyncio.create_task(autofix_b30_song_background(recent["song_id"],
|
|
byd=False if recent["difficulty"] != 3 else True))
|
|
return result
|
|
|
|
else:
|
|
errcode_string = f"arcaea.errcode.{get_['status']}"
|
|
if locale := msg.locale.t(errcode_string) != errcode_string:
|
|
return Plain(f'{msg.locale.t("arcaea.message.failed.errcode")}{locale}')
|
|
return Plain(msg.locale.t('arcaea.message.failed') + get_)
|