xuthus/LittlePaimon
1
0
Fork 0
mirror of https://github.com/xuthus83/LittlePaimon.git synced 2024-10-21 16:27:15 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
e67c117348
LittlePaimon/Paimon_Info/draw_daily_note.py

297 lines
17 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

import random
import datetime
# from io import BytesIO
import os
from PIL import Image, ImageDraw, ImageFont
import matplotlib.pyplot as plt
# from aiohttp import ClientSession
from nonebot.adapters.onebot.v11 import MessageSegment
from ..utils.util import pil2b64, get_pic
res_path = os.path.join(os.path.dirname(os.path.dirname(__file__)), 'res')
# def get_font(size):
# return ImageFont.truetype(os.path.join(res_path, 'msyhbd.ttc'), size)
#
#
# def get_odd_time(seconds):
# m, s = divmod(int(seconds), 60)
# h, m = divmod(m, 60)
# return "剩余%02d时%02d分%02d秒" % (h, m, s)
#
#
# async def get_avater_pic(avater_url):
# async with ClientSession() as session:
# res = await session.get(avater_url)
# res = await res.read()
# avater = Image.open(BytesIO(res)).convert("RGBA").resize((60, 60))
# return avater
#
#
# bg_card_color = {'1': '#C3B8A4', '2': '#C3B8A4', '3': '#4C74A7', '4': '#D7B599'}
#
#
# async def draw_daily_note_card(data, uid):
# if not data:
# return '数据出错'
# if data['retcode'] == 10102:
# return '这uid没有在米游社公开信息哦,请到 个人主页-管理 中打开'
# elif data['retcode'] == 10104:
# return 'uid有误哦，检查一下或再手动输入一次uid吧'
# elif data['retcode'] != 0:
# return f'派蒙获取数据失败了，获取状态：\n{data["message"]},{data["retcode"]}'
# data = data['data']
# # 载入所需素材图标
# bg_color = random.choice(list(bg_card_color.items()))
# bg_img = Image.open(os.path.join(res_path, 'daily_note', f'便签背景{bg_color[0]}.png')).convert("RGBA")
# enemy = Image.open(os.path.join(res_path, 'daily_note', '周本.png')).convert("RGBA")
# task = Image.open(os.path.join(res_path, 'daily_note', '委托.png')).convert("RGBA")
# power = Image.open(os.path.join(res_path, 'daily_note', '树脂.png')).convert("RGBA")
# money = Image.open(os.path.join(res_path, 'daily_note', '洞天宝钱.png')).convert("RGBA")
# send_icon = Image.open(os.path.join(res_path, 'daily_note', '派遣背景.png')).convert("RGBA").resize((110, 55))
# send_finish_icon = Image.open(os.path.join(res_path, 'daily_note', '派遣完成.png')).convert("RGBA").resize((55, 55))
# abyss = Image.open(os.path.join(res_path, 'daily_note', '深渊.png')).convert('RGBA').resize((160, 160))
# tran = Image.open(os.path.join(res_path, 'daily_note', '参量.png')).convert('RGBA').resize((40, 40))
# bg_draw = ImageDraw.Draw(bg_img)
#
# bg_draw.text((23, 20), '实时便笺', font=get_font(30), fill='white')
# bg_draw.text((255, 20), 'UID：' + uid, font=get_font(30), fill='white')
# # 树脂
# bg_img.alpha_composite(power, (120, 150))
# bg_draw.text((170, 145), f'{data["current_resin"]}/160', font=get_font(30), fill=bg_color[1])
# if data["current_resin"] == 160:
# bg_draw.text((310, 144), '已回满', font=get_font(30), fill=bg_color[1])
# else:
# recover_time = datetime.datetime.now() + datetime.timedelta(seconds=int(data['resin_recovery_time']))
# # recover_time_day = recover_time.day > datetime.datetime.now().day and '明天' or '今天'
# recover_time_day = '今天' if recover_time.day == datetime.datetime.now().day else '明天'
# recover_time_str = f'将于{recover_time_day}{recover_time.strftime("%H:%M")}回满'
# bg_draw.text((320, 147), recover_time_str, font=get_font(25), fill=bg_color[1])
# # 洞天宝钱
# bg_img.alpha_composite(money, (120, 220))
# bg_draw.text((170, 220), f'{data["current_home_coin"]}/2400', font=get_font(30), fill=bg_color[1])
# if data["current_home_coin"] == 2400:
# bg_draw.text((350, 219), '已存满', font=get_font(30), fill=bg_color[1])
# else:
# recover_time = datetime.datetime.now() + datetime.timedelta(seconds=int(data['home_coin_recovery_time']))
# recover_time_day = recover_time.day - datetime.datetime.now().day
# if recover_time_day == 1:
# recover_time_day_str = '明天'
# elif recover_time_day == 0:
# recover_time_day_str = '今天'
# else:
# recover_time_day_str = str(recover_time.day) + '日'
# recover_time_str = f'将于{recover_time_day_str}{recover_time.strftime("%H:%M")}攒满'
# # recover_time_str = f'将于{recover_time.strftime("%d日%H:%M")}攒满'
# bg_draw.text((360, 222), recover_time_str, font=get_font(25), fill=bg_color[1])
# # 委托
# bg_img.alpha_composite(task, (120, 295))
# bg_draw.text((170, 297), f'{data["finished_task_num"]}/4', font=get_font(30), fill=bg_color[1])
# if data["finished_task_num"] == 4:
# bg_draw.text((247, 297), '已完成', font=get_font(30), fill=bg_color[1])
# else:
# bg_draw.text((245, 298), '未完成', font=get_font(25), fill=bg_color[1])
# # 周本
# bg_img.alpha_composite(enemy, (120, 370))
# bg_draw.text((170, 370), f'{data["remain_resin_discount_num"]}/3', font=get_font(30), fill=bg_color[1])
# if data["remain_resin_discount_num"] == 0:
# bg_draw.text((247, 369), '已完成', font=get_font(30), fill=bg_color[1])
# else:
# bg_draw.text((245, 372), '未完成', font=get_font(25), fill=bg_color[1])
# # 参量质变仪
# bg_img.alpha_composite(tran, (360, 297))
# if not data['transformer']['obtained']:
# bg_draw.text((413, 298), '未获得', font=get_font(25), fill=bg_color[1])
# else:
# if data['transformer']['recovery_time']['reached']:
# bg_draw.text((415, 297), '已可用', font=get_font(30), fill=bg_color[1])
# else:
# bg_draw.text((413, 298), f"{data['transformer']['recovery_time']['Day']}天后", font=get_font(25),
# fill=bg_color[1])
# # 深渊
# abyss_new_month = datetime.datetime.now().month if datetime.datetime.now().day < 16 else datetime.datetime.now().month + 1
# abyss_new_day = 16 if datetime.datetime.now().day < 16 else 1
# abyss_new = datetime.datetime.strptime('2022.' + str(abyss_new_month) + '.' + str(abyss_new_day) + '.04:00',
# '%Y.%m.%d.%H:%M') - datetime.datetime.now()
# abyss_new_str = f'{abyss_new.days + 1}天后刷新' if abyss_new.days <= 8 else '已刷新'
# bg_img.alpha_composite(abyss, (520, 264))
# bg_draw.text((568, 300), '深渊', font=get_font(30), fill=bg_color[1])
# if abyss_new_str == '已刷新':
# bg_draw.text((561, 350), abyss_new_str, font=get_font(25), fill=bg_color[1])
# else:
# bg_draw.text((540, 350), abyss_new_str, font=get_font(25), fill=bg_color[1])
# # 派遣
# h = 430
# if not data['expeditions']:
# bg_draw.text((300, h + 140), '没有派遣信息', font=get_font(30), fill=bg_color[1])
# else:
# for send in data['expeditions']:
# send_avatar = await get_avater_pic(send['avatar_side_icon'])
# send_status = '派遣已完成！' if send['status'] == 'Finished' else get_odd_time(send['remained_time'])
# bg_draw.rectangle((145, h, 645, h + 55), fill=None, outline=bg_color[1], width=3)
# if send['status'] == 'Finished':
# bg_img.alpha_composite(send_finish_icon, (590, h))
# bg_img.alpha_composite(send_icon, (150, h))
# bg_img.alpha_composite(send_avatar, (150, h - 10))
# if send_status == '派遣已完成！':
# bg_draw.text((329, h + 10), send_status, font=get_font(25), fill=bg_color[1])
# else:
# bg_draw.text((300, h + 10), send_status, font=get_font(25), fill=bg_color[1])
# h += 57
# last_finish_second = int(max([s['remained_time'] for s in data['expeditions']]))
# if last_finish_second != 0:
# last_finish_time = datetime.datetime.now() + datetime.timedelta(seconds=last_finish_second)
# last_finish_day = last_finish_time.day > datetime.datetime.now().day and '明天' or '今天'
# last_finish_str = f'将于{last_finish_day}{last_finish_time.strftime("%H:%M")}完成全部派遣'
# bg_draw.text((211, h + 3.5), last_finish_str, font=get_font(30), fill=bg_color[1])
# else:
# bg_draw.text((290, h + 3.5), '派遣已全部完成', font=get_font(30), fill=bg_color[1])
# bg_draw.text((274, 797), 'Created by 惜月の小派蒙', font=get_font(20), fill=bg_color[1])
#
# bg_img = pil2b64(bg_img, 70)
# bg_img = MessageSegment.image(bg_img)
# return bg_img
def get_font(size, font='msyhbd.ttc'):
return ImageFont.truetype(os.path.join(res_path, font), size)
async def draw_ring(per):
per_list = [per, 1 - per]
colors = ['#507bd0', '#FFFFFF']
plt.pie(per_list, startangle=90, colors=colors)
fig, ax = plt.subplots(figsize=(6, 6))
ax.pie(per_list,
wedgeprops={'width': 0.18},
startangle=90,
colors=colors)
plt.savefig('temp.png', transparent=True)
img = Image.open('temp.png').resize((266, 266)).convert('RGBA')
os.remove('temp.png')
return img
async def draw_daily_note_card(data, uid):
if not data:
return '数据出错'
if data['retcode'] == 10102:
return '这uid没有在米游社公开信息哦,请到 个人主页-管理 中打开'
elif data['retcode'] == 10104:
return 'uid有误哦，检查一下或再手动输入一次uid吧'
elif data['retcode'] != 0:
return f'派蒙获取数据失败了，获取状态：\n{data["message"]},{data["retcode"]}'
data = data['data']
circle_img = Image.open(os.path.join(res_path, 'daily_note', '透明圆.png'))
finished_icon = Image.open(os.path.join(res_path, 'daily_note', 'finished.png'))
bg_img = Image.open(os.path.join(res_path, 'daily_note', 'ssbq.png')).convert("RGBA")
bg_draw = ImageDraw.Draw(bg_img)
# uid文字
bg_draw.text((152, 251), f"uid{uid}", fill='#5680d2', font=get_font(60, 'number.ttf'))
# 树脂文字
bg_draw.text((337, 480), f"{data['current_resin']}/160", fill='white', font=get_font(48, 'number.ttf'))
bg_img.alpha_composite(await draw_ring(data['current_resin'] / 160), (98, 369))
if data['current_resin'] == 160:
bg_draw.text((892, 480), f"树脂满了哦~", fill='white', font=get_font(40, '优设标题黑.ttf'))
else:
recover_time = datetime.datetime.now() + datetime.timedelta(seconds=int(data['resin_recovery_time']))
recover_time_day = '今天' if recover_time.day == datetime.datetime.now().day else '明天'
recover_time_str = f'将于{recover_time_day}{recover_time.strftime("%H:%M")}回满'
bg_draw.text((780, 480), recover_time_str, fill='white', font=get_font(40, '优设标题黑.ttf'))
# 宝钱文字
bg_draw.text((337, 701), f"{data['current_home_coin']}/{data['max_home_coin']}", fill='white', font=get_font(48, 'number.ttf'))
bg_img.alpha_composite(await draw_ring(data['current_home_coin'] / data['max_home_coin']), (98, 593))
if data['current_home_coin'] == data['max_home_coin']:
bg_draw.text((820, 701), f"洞天宝钱满了哦~", fill='white', font=get_font(40, '优设标题黑.ttf'))
else:
recover_time = datetime.datetime.now() + datetime.timedelta(seconds=int(data['home_coin_recovery_time']))
recover_time_day = recover_time.day - datetime.datetime.now().day
if recover_time_day == 1:
recover_time_day_str = '明天'
elif recover_time_day == 0:
recover_time_day_str = '今天'
else:
recover_time_day_str = str(recover_time.day) + '日'
recover_time_str = f'将于{recover_time_day_str}{recover_time.strftime("%H:%M")}攒满'
bg_draw.text((762, 701), recover_time_str, fill='white', font=get_font(40, '优设标题黑.ttf'))
# 委托文字
bg_draw.text((337, 924), f"{data['finished_task_num']}/4", fill='white', font=get_font(48, 'number.ttf'))
bg_img.alpha_composite(await draw_ring(data['finished_task_num'] / 4), (98, 816))
if data['finished_task_num'] == 4:
bg_draw.text((750, 924), "今日委托已全部完成~", fill='white', font=get_font(40, '优设标题黑.ttf'))
else:
bg_draw.text((790, 924), "今日委托完成情况", fill='white', font=get_font(40, '优设标题黑.ttf'))
# 质变文字
if data['transformer']['obtained']:
bg_draw.text((337, 1147), f"{7 - data['transformer']['recovery_time']['Day']}/7", fill='white',
font=get_font(48, 'number.ttf'))
bg_img.alpha_composite(await draw_ring((7 - data['transformer']['recovery_time']['Day']) / 7), (98, 1039))
if data['transformer']['recovery_time']['Day'] == 0:
bg_draw.text((465, 1147), "可使用", fill='white', font=get_font(40, '优设标题黑.ttf'))
else:
bg_draw.text((471, 1127), f"{data['transformer']['recovery_time']['Day']}天后", fill='white',
font=get_font(40, '优设标题黑.ttf'))
bg_draw.text((465, 1167), "可使用", fill='white', font=get_font(40, '优设标题黑.ttf'))
else:
bg_draw.text((337, 1143), "未获得", fill='white', font=get_font(48, '优设标题黑.ttf'))
# 周本文字
bg_draw.text((843, 1147), f"{3 - data['remain_resin_discount_num']}/3", fill='white',
font=get_font(48, 'number.ttf'))
bg_img.alpha_composite(await draw_ring((3 - data['remain_resin_discount_num']) / 3), (604, 1039))
if data['remain_resin_discount_num'] == 0:
bg_draw.text((1005, 1147), "已完成", fill='white', font=get_font(40, '优设标题黑.ttf'))
else:
bg_draw.text((977, 1127), f"剩余{data['remain_resin_discount_num']}次", fill='white',
font=get_font(40, '优设标题黑.ttf'))
bg_draw.text((965, 1167), "周本减半", fill='white', font=get_font(40, '优设标题黑.ttf'))
# 深渊文字
abyss_new_month = datetime.datetime.now().month if datetime.datetime.now().day < 16 else datetime.datetime.now().month + 1
abyss_new_day = 16 if datetime.datetime.now().day < 16 else 1
abyss_new = datetime.datetime.strptime('2022.' + str(abyss_new_month) + '.' + str(abyss_new_day) + '.04:00',
'%Y.%m.%d.%H:%M') - datetime.datetime.now()
bg_draw.text((337, 1358), f"{abyss_new.days}/15", fill='white',
font=get_font(48, 'number.ttf'))
bg_draw.text((745, 1358), f"本期深渊还有{abyss_new.days}天结束", fill='white',
font=get_font(40, '优设标题黑.ttf'))
bg_img.alpha_composite(await draw_ring(abyss_new.days / 15), (100, 1249))
# 派遣情况
exp = data['expeditions']
i = 0
for role in exp:
role_avatar = await get_pic(role['avatar_side_icon'], (135, 135), 'RGBA')
bg_img.alpha_composite(role_avatar, (i * 200 + 168, 1537))
bg_img.alpha_composite(await draw_ring(1 - int(role['remained_time']) / 72000), (i * 201 + 101, 1490))
if role['status'] == 'Ongoing':
bg_img.alpha_composite(circle_img, (i * 200 + 172, 1559))
hour = int(role['remained_time']) // 3600
bg_draw.text((i * 200 + 212, 1580), f"{hour}h", fill='white', font=get_font(40, 'number.ttf'))
minute = int(role['remained_time']) % 3600 // 60
bg_draw.text((i * 200 + 197, 1620), f"{minute}m", fill='white', font=get_font(40, 'number.ttf'))
else:
bg_img.alpha_composite(finished_icon, (i * 200 + 191, 1576))
i += 1
bg_draw.text((1220, 1580), "派遣全部", fill="#5680d2", font=get_font(40, '优设标题黑.ttf'))
bg_draw.text((1220, 1620), "完成时间", fill="#5680d2", font=get_font(40, '优设标题黑.ttf'))
max_time = int(max([s['remained_time'] for s in exp]))
if max_time == 0:
bg_draw.text((1410, 1583), "已全部完成~", fill="#5680d2",
font=get_font(60, '优设标题黑.ttf'))
else:
last_finish_time = datetime.datetime.now() + datetime.timedelta(seconds=max_time)
last_finish_day = last_finish_time.day > datetime.datetime.now().day and '明天' or '今天'
last_finish_str = f'{last_finish_day}{last_finish_time.strftime("%H:%M")}'
bg_draw.text((1408, 1588), last_finish_str, fill="#5680d2",
font=get_font(60, '优设标题黑.ttf'))
role_img = Image.open(os.path.join(res_path, 'emoticons', random.choice(os.listdir(os.path.join(res_path, 'emoticons'))))).convert('RGBA')
role_img = role_img.resize((int(role_img.size[0] * 3.5), int(role_img.size[1] * 3.5)), Image.ANTIALIAS)
bg_img.alpha_composite(role_img, (1220, 200))
now = datetime.datetime.now().strftime('%m月%d日%H:%M')
bg_draw.text((554, 1794), 'Created by LittlePaimon·' + now, fill='#5680d2', font=get_font(40, '优设标题黑.ttf'))
bg_img = bg_img.resize((int(bg_img.size[0] * 0.35), int(bg_img.size[1] * 0.35)), Image.ANTIALIAS)
bg_img = pil2b64(bg_img, 70)
bg_img = MessageSegment.image(bg_img)
return bg_img
Reference in New Issue View Git Blame Copy Permalink