akari-bot/modules/arcaea/info_official.py

"""
This service utilizes API functionality provided by and with permission from lowiro. It is not affiliated with or endorsed by lowiro.
"""
import os
import traceback
from datetime import datetime

from config import Config
from core.elements import Plain
from core.logger import Logger
from core.utils import get_url

assets_path = os.path.abspath('./assets/arcaea')
apiurl = Config('arcapi_official_url')
headers = {'Authorization': Config('arcapi_official_token')}
headers_botarcapi = {"User-Agent": Config('botarcapi_agent')}
botarcapi_url = Config('botarcapi_url')


async def get_songinfo(songid):
    try:
        get_songinfo_from_botarcapi = await get_url(f'{botarcapi_url}song/info?songid={songid}',
                                                    headers=headers_botarcapi, status_code=200, fmt='json')
        return get_songinfo_from_botarcapi['content']
    except Exception:
        traceback.print_exc()
        return False


async def get_info_official(usercode):
    try:
        getuserinfo_json = await get_url(f'{apiurl}user/{usercode}', headers=headers, status_code=200, fmt='json')
    except ValueError as e:
        Logger.info(f'[{usercode}] {e}')
        return {'success': False, 'msg': '查询失败。'}
    except Exception:
        traceback.print_exc()
        return {'success': False, 'msg': '查询失败。'}
    getuserinfo = getuserinfo_json['data']
    username = getuserinfo['display_name']
    potential = getuserinfo['potential']
    if potential is not None:
        potential = int(potential) / 100
    else:
        potential = '--'
    recent = getuserinfo["last_played_song"]
    if recent is None:
        return [Plain('此用户无游玩记录。')]
    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'
    score = recent['score']
    ptt = realptt = '???'
    trackname = recent['song_id']
    songinfo = await get_songinfo(recent['song_id'])
    if songinfo:
        s = songinfo['difficulties'][recent['difficulty']]
        trackname = s['name_en']
        realptt = s['rating'] / 10
        ptt = realptt
        if score >= 10000000:
            ptt += 2
        elif score >= 9800000:
            ptt += 1 + (score - 9800000) / 200000
        else:
            ptt += (score - 9500000) / 300000

    shiny_pure = recent['shiny_pure_count']
    pure = recent['pure_count']
    far = recent['far_count']
    lost = recent['lost_count']
    time_played = datetime.fromtimestamp(recent['time_played'] / 1000)
    result = {'success': True, 'msg': f'{username} (Ptt: {potential})的最近游玩记录：\n'
                                      f'{trackname} ({difficulty})\n'
                                      f'Score: {score}\n'
                                      f'Pure: {pure} ({shiny_pure})\n'
                                      f'Far: {far}\n'
                                      f'Lost: {lost}\n'
                                      f'Potential: {realptt} -> {ptt}\n'
                                      f'Time: {time_played.strftime("%Y-%m-%d %H:%M:%S")}(UTC+8)'}
    return result