zhenxun_bot/plugins/genshin/almanac/alc.py

from PIL import Image, ImageDraw, ImageFont
from io import BytesIO

import os
import json
import random
import base64
import time

FILE_PATH = os.path.dirname(__file__)
FONT_PATH = os.path.join(FILE_PATH, "汉仪文黑.ttf")

data = {}  # configs.json里的数据

almanac_data = {
    # 生成的黄历base64字符串和黄历更新日期
    "date": "",
    "almanac_base64_str": ""
}

chinese = {"0": "", "1": "一", "2": "二", "3": "三", "4": "四", "5": "五", "6": "六", "7": "七", "8": "八", "9": "九"}


def month_to_chinese(month: str):
    # 把日期数字转成中文数字
    m = int(month)
    if m < 10:
        return chinese[month[-1]]
    elif m < 20:
        return "十" + chinese[month[-1]]
    else:
        return chinese[month[0]] + "十" + chinese[month[-1]]


def load_data():
    # 载入config.json文件的数据
    global data
    with open(os.path.join(FILE_PATH, 'config.json'), 'r', encoding='UTF-8') as f:
        data = json.load(f)

    almanac_data["date"] = ""
    almanac_data["almanac_base64_str"] = ""


load_data()


def seed_random_list(seed: str, l: list):
    # 使用随机种子随机选择列表中的元素，相同的种子和列表将返回同样的输出
    seed = seed + str(l)
    random.seed(seed)
    index = random.random() * len(l)
    return l[int(index)]


def generate_almanac():
    # 生成黄历图片，然后转换成base64保存到 almanac_data["almanac_base64_str"]

    seed = time.strftime("%Y-%m-%d")
    offset = 1
    today_luck = []
    l = list(data.keys())

    while len(today_luck) < 6:
        # 随机6个不同的运势放到 today_luck
        r = seed_random_list(str(offset) + seed, l)
        if r in today_luck:
            offset += 1
        else:
            today_luck.append(r)

    back = Image.open(os.path.join(FILE_PATH, "back.png"))

    year = time.strftime("%Y")
    month = month_to_chinese(time.strftime("%m")) + "月"
    day = month_to_chinese(time.strftime("%d")) + "日"

    draw = ImageDraw.Draw(back)
    draw.text((118, 165), year, fill="#8d7650ff", font=ImageFont.truetype(FONT_PATH, size=30), anchor="mm",
              align="center")
    draw.text((260, 165), day, fill="#f7f8f2ff", font=ImageFont.truetype(FONT_PATH, size=35), anchor="mm",
              align="center")
    draw.text((410, 165), month, fill="#8d7650ff", font=ImageFont.truetype(FONT_PATH, size=30), anchor="mm",
              align="center")

    buff = Image.new("RGBA", (325, 160))
    debuff = Image.new("RGBA", (325, 160))

    buff_draw = ImageDraw.Draw(buff)
    debuff_draw = ImageDraw.Draw(debuff)

    for i in range(3):
        buff_name = today_luck[i]
        debuff_name = today_luck[(i + 3)]

        buff_effect = seed_random_list(seed, data[buff_name]["buff"])
        debuff_effect = seed_random_list(seed, data[debuff_name]["debuff"])

        buff_draw.text((0, i * 53), buff_name, fill="#756141ff", font=ImageFont.truetype(FONT_PATH, size=25))
        debuff_draw.text((0, i * 53), debuff_name, fill="#756141ff", font=ImageFont.truetype(FONT_PATH, size=25))

        buff_draw.text((0, i * 53 + 28), buff_effect, fill="#b5b3acff", font=ImageFont.truetype(FONT_PATH, size=19))
        debuff_draw.text((0, i * 53 + 28), debuff_effect, fill="#b5b3acff", font=ImageFont.truetype(FONT_PATH, size=19))

    back.paste(buff, (150, 230), buff)
    back.paste(debuff, (150, 400), debuff)

    bio = BytesIO()
    back.save(bio, format='PNG')
    base64_str = base64.b64encode(bio.getvalue()).decode()

    almanac_data["date"] = time.strftime("%Y-%m-%d")
    almanac_data["almanac_base64_str"] = 'base64://' + base64_str


def get_almanac_base64_str():
    # if almanac_data["date"] == time.strftime("%Y-%m-%d"):
    #     return almanac_data["almanac_base64_str"]
    # else:
    #     generate_almanac()
    #     return almanac_data["almanac_base64_str"]
    generate_almanac()
    return almanac_data["almanac_base64_str"]