akari-bot/modules/ptt/__init__.py

import os

from PIL import Image, ImageDraw, ImageFont

from core.component import on_command
from core.elements import Image as Img
from core.builtins.message import MessageSession
from core.utils.cache import random_cache_path

assets_path = os.path.abspath('./assets/arcaea')

p = on_command('ptt',
               developers=['OasisAkari'])


@p.handle('<potential> {生成一张Arcaea Potential图片}')
async def pttimg(msg: MessageSession):
    ptt = msg.parsed_msg['<potential>']
    # ptt
    if ptt == '--':
        ptt = -1
    else:
        try:
            ptt = float(ptt)
        except ValueError:
            await msg.finish('发生错误：potential 必须为 ≥0.00 且 ≤99.99 的数字。')
    if ptt >= 13.00:
        pttimg = 7
    elif ptt >= 12.50:
        pttimg = 6
    elif ptt >= 12.00:
        pttimg = 5
    elif ptt >= 11.00:
        pttimg = 4
    elif ptt >= 10.00:
        pttimg = 3
    elif ptt >= 7.00:
        pttimg = 2
    elif ptt >= 3.50:
        pttimg = 1
    elif ptt >= 0:
        pttimg = 0
    else:
        pttimg = 'off'
    pttimgr = Image.open(f'{assets_path}/ptt/rating_{str(pttimg)}.png')
    ptttext = Image.new("RGBA", (119, 119))
    font1 = ImageFont.truetype(os.path.abspath(f'{assets_path}/Fonts/Exo-SemiBold.ttf'), 49)
    font2 = ImageFont.truetype(os.path.abspath(f'{assets_path}/Fonts/Exo-SemiBold.ttf'), 33)
    if ptt >= 0 and ptt <= 99.99:
        rawptt = str(ptt).split('.')
        if len(rawptt) < 2:
            ptt1 = rawptt[0]
            ptt2 = '00'
        else:
            ptt1 = rawptt[0]
            ptt2 = rawptt[1][:2]
            if len(ptt2) < 2:
                ptt2 += '0'
        ptttext_width, ptttext_height = ptttext.size
        font1_width, font1_height = font1.getsize(ptt1 + '.')
        font2_width, font2_height = font2.getsize(ptt2)
        print(font1_width, font1_height)
        print(font2_width, font2_height)
        pttimg = Image.new("RGBA", (font1_width + font2_width + 6, font1_height + 6))
        drawptt = ImageDraw.Draw(pttimg)
        stroke_color = '#52495d'
        if int(ptt1) >= 13:
            stroke_color = '#81122F'
        drawptt.text((0, 0), ptt1 + '.', 'white', font=font1, stroke_width=3, stroke_fill=stroke_color)
        print(int(int(font1_height) - int(font2_height)))
        drawptt.text((font1_width, int(int(font1_height) - int(font2_height))), ptt2, 'white', font=font2, stroke_width=3, stroke_fill=stroke_color)
    elif ptt == -1:
        ptt = '--'
        ptttext_width, ptttext_height = ptttext.size
        font1_width, font1_height = font1.getsize(ptt)
        pttimg = Image.new("RGBA", (font1_width + 6, font1_height + 6))
        drawptt = ImageDraw.Draw(pttimg)
        drawptt.text((0, 0), ptt, 'white', font=font1, stroke_width=3, stroke_fill='#52495d')
    else:
        return await msg.finish('发生错误：potential 必须为 ≥0.00 且 ≤99.99 的数字。')
    pttimg_width, pttimg_height = pttimg.size
    ptttext.alpha_composite(pttimg,
                            (int((ptttext_width - pttimg_width) / 2), int((ptttext_height - pttimg_height) / 2) - 11))
    ptttext = ptttext.resize(pttimgr.size)
    pttimgr.alpha_composite(ptttext, (0, 0))
    savepath = random_cache_path() + '.png'
    pttimgr.save(savepath)
    await msg.finish([Img(path=savepath)])