generate.py

import math
import asyncio
import datetime
import base64
from io import BytesIO
from .event import *
from .draw import *

server_name = {
    'cn': '国服',
    'tw': '台服',
    'jp': '日服',
}

def im2base64str(im):
    io = BytesIO()
    im.save(io, 'png')
    base64_str = f"base64://{base64.b64encode(io.getvalue()).decode()}"
    return base64_str

async def generate_day_schedule(server = 'cn'):
    if server == 'cn':
        events = await get_events('cn', 0, 7)
        eventsb = await get_events('cnb', 0, 7)
        if len(events) < len(eventsb):
            events = eventsb
    else:
        events = await get_events(server, 0, 7)

    has_prediction = False
    title_len = 25
    for event in events:
        if event['start_days'] > 0:
            has_prediction = True
        title_len = max(title_len, len(event['title']) + 5)
    if has_prediction:
        im = create_image(len(events) + 2, title_len)
    else:
        im = create_image(len(events) + 1, title_len)

    title = f'公主连结{server_name[server]}活动'
    pcr_now = get_pcr_now(0)
    draw_title(im, 0, title, pcr_now.strftime('%Y/%m/%d'), '正在进行')

    if len(events) == 0:
        draw_item(im, 1, 1, '无数据', 0)
    i = 1
    for event in events:
        if event['start_days'] <= 0:
            draw_item(im, i, event['type'], event['title'], event['left_days'])
            i += 1
    if has_prediction:
        draw_title(im, i, right = '即将开始')
        for event in events:
            if event['start_days'] > 0:
                i += 1
                draw_item(im, i, event['type'], event['title'], -event['start_days'])
    return im