mirror of
https://github.com/KimigaiiWuyi/GenshinUID.git
synced 2025-05-30 04:00:31 +08:00
308 lines
13 KiB
Python
308 lines
13 KiB
Python
from io import BytesIO
|
|
from re import findall
|
|
from typing import List
|
|
from pathlib import Path
|
|
from datetime import datetime
|
|
|
|
from httpx import get
|
|
from bs4 import BeautifulSoup
|
|
from PIL import Image, ImageDraw
|
|
|
|
from .get_event_data import get_genshin_events
|
|
from ..utils.genshin_fonts.genshin_fonts import genshin_font_origin
|
|
|
|
TEXT_PATH = Path(__file__).parent / 'texture2d'
|
|
IMG_PATH = Path(__file__).parent / 'event.jpg'
|
|
|
|
PATTERN = r'<[a-zA-Z]+.*?>([\s\S]*?)</[a-zA-Z]*?>'
|
|
|
|
|
|
async def get_month_and_time(time_data: str) -> List:
|
|
"""
|
|
:说明:
|
|
接收时间字符串`2022/02/09 18:59:59`
|
|
转换为`['02/09', '18:59PM']`
|
|
:参数:
|
|
* time_data (str): 时间字符串。
|
|
:返回:
|
|
* [month, time] (list): ['02/09', '18:59PM']。
|
|
"""
|
|
if '永久开放' in time_data:
|
|
month = time_data[:5]
|
|
time = '永久开放'
|
|
if '更新后' in time_data:
|
|
month = time_data[:5]
|
|
time = '更新后'
|
|
else:
|
|
time_data = time_data.split(' ') # type: ignore
|
|
time_data[0] = time_data[0].replace('-', '/') # type: ignore
|
|
month = time_data[0].split('/', 1)[1]
|
|
time = ':'.join(time_data[1].split(':')[:-1])
|
|
if int(time.split(':')[0]) <= 12:
|
|
time = time + 'AM'
|
|
else:
|
|
time = time + 'PM'
|
|
return [month, time]
|
|
|
|
|
|
async def save_draw_event_img() -> None:
|
|
"""
|
|
:说明:
|
|
绘制原神活动列表图片，存放至同级目录``event.png``。
|
|
"""
|
|
raw_data = await get_genshin_events('List')
|
|
raw_time_data = await get_genshin_events('Content')
|
|
|
|
data = raw_data['data']['list'][1]['list']
|
|
|
|
event_data = {'gacha_event': [], 'normal_event': [], 'other_event': []}
|
|
for k in data:
|
|
for i in raw_time_data['data']['list']:
|
|
if k['title'] in i['title']:
|
|
content_bs = BeautifulSoup(i['content'], 'lxml')
|
|
for index, value in enumerate(content_bs.find_all('p')):
|
|
if value.text == '〓任务开放时间〓':
|
|
time_data = content_bs.find_all('p')[index + 1].text
|
|
if '<t class=' in time_data:
|
|
time_data = findall(
|
|
PATTERN,
|
|
time_data,
|
|
)[0]
|
|
month_start, time_start = await get_month_and_time(
|
|
time_data
|
|
)
|
|
k['start_time'] = [month_start, time_start]
|
|
k['end_time'] = ['更新后', '永久开放']
|
|
elif value.text == '〓活动时间〓':
|
|
time_data = content_bs.find_all('p')[index + 1].text
|
|
if '<t class=' in time_data:
|
|
time_datas = []
|
|
for s in time_data.split(' ~ '):
|
|
if '<t class=' in s:
|
|
time_datas.append(
|
|
findall(
|
|
PATTERN,
|
|
s,
|
|
)[0]
|
|
)
|
|
else:
|
|
time_datas.append(s)
|
|
if ' ' in time_datas[0]:
|
|
(
|
|
month_start,
|
|
time_start,
|
|
) = await get_month_and_time(time_datas[0])
|
|
else:
|
|
month_start, time_start = '版本更新后', '更新后'
|
|
|
|
if ' ' in time_datas[1]:
|
|
month_end, time_end = await get_month_and_time(
|
|
time_datas[1]
|
|
)
|
|
elif '版本结束' in time_datas[1]:
|
|
month_end, time_end = time_datas[1][:5], '结束后'
|
|
else:
|
|
month_end, time_end = '更新后', '永久开放'
|
|
k['start_time'] = [month_start, time_start]
|
|
k['end_time'] = [month_end, time_end]
|
|
elif '活动内容' in time_data:
|
|
for n in range(2, 10):
|
|
time_data = content_bs.find_all('p')[
|
|
index + n
|
|
].text
|
|
if '版本更新后' in time_data:
|
|
time_data_end = content_bs.find_all('p')[
|
|
index + n + 1
|
|
].text
|
|
if '<t class=' in time_data_end:
|
|
time_data_end = findall(
|
|
PATTERN,
|
|
time_data_end,
|
|
)[0]
|
|
(
|
|
month_end,
|
|
time_end,
|
|
) = await get_month_and_time(
|
|
time_data_end
|
|
)
|
|
k['start_time'] = [
|
|
time_data[:5],
|
|
'更新后',
|
|
]
|
|
k['end_time'] = [month_end, time_end]
|
|
else:
|
|
k['start_time'] = [time_data, '维护后']
|
|
k['end_time'] = ['更新后', '永久开放']
|
|
break
|
|
elif '<t class=' in time_data:
|
|
time_data = findall(
|
|
PATTERN,
|
|
time_data,
|
|
)[0]
|
|
(
|
|
month_start,
|
|
time_start,
|
|
) = await get_month_and_time(time_data)
|
|
k['start_time'] = [month_start, time_start]
|
|
time_data_end = content_bs.find_all('p')[
|
|
index + n + 1
|
|
].text
|
|
if '<t class=' in time_data_end:
|
|
time_data_end = findall(
|
|
PATTERN,
|
|
time_data_end,
|
|
)[0]
|
|
(
|
|
month_end,
|
|
time_end,
|
|
) = await get_month_and_time(
|
|
time_data_end
|
|
)
|
|
k['end_time'] = [month_end, time_end]
|
|
elif '版本结束' in time_data_end:
|
|
k['end_time'] = [
|
|
time_data_end[1:6],
|
|
'结束',
|
|
]
|
|
else:
|
|
k['end_time'] = ['更新后', '永久开放']
|
|
break
|
|
else:
|
|
month_start, time_start = await get_month_and_time(
|
|
time_data
|
|
)
|
|
k['start_time'] = [month_start, time_start]
|
|
k['end_time'] = ['更新后', '永久开放']
|
|
elif value.text == '〓祈愿介绍〓':
|
|
start_time = (
|
|
content_bs.find_all('tr')[1]
|
|
.td.find_all('p')[0]
|
|
.text
|
|
)
|
|
if '<t class=' in start_time:
|
|
start_time = findall(
|
|
PATTERN,
|
|
start_time,
|
|
)[0]
|
|
end_time = findall(
|
|
PATTERN,
|
|
content_bs.find_all('tr')[1]
|
|
.td.find_all('p')[2]
|
|
.text,
|
|
)[0]
|
|
if '<t class=' in end_time:
|
|
end_time = findall(
|
|
PATTERN,
|
|
end_time,
|
|
)[0]
|
|
|
|
month_start, time_start = await get_month_and_time(
|
|
start_time
|
|
)
|
|
month_end, time_end = await get_month_and_time(
|
|
end_time
|
|
)
|
|
|
|
k['start_time'] = [month_start, time_start]
|
|
k['end_time'] = [month_end, time_end]
|
|
|
|
if '冒险助力礼包' in k['title'] or '纪行' in k['title']:
|
|
continue
|
|
# if '角色试用' in k['title'] or '传说任务' in k['title']:
|
|
# event_data['other_event'].append(k)
|
|
elif k['tag_label'] == '扭蛋':
|
|
event_data['gacha_event'].append(k)
|
|
elif k['tag_label'] == '活动':
|
|
event_data['normal_event'].append(k)
|
|
|
|
base_h = (
|
|
450
|
|
+ len(event_data['normal_event']) * (270 + 10)
|
|
+ len(event_data['gacha_event']) * (370 + 10)
|
|
)
|
|
base_img = Image.new(
|
|
mode='RGBA', size=(950, base_h), color=(255, 253, 248, 255)
|
|
)
|
|
|
|
text_color = (60, 59, 64)
|
|
event_color = (250, 93, 93)
|
|
gacha_color = (93, 198, 250)
|
|
font_l = genshin_font_origin(52)
|
|
font_m = genshin_font_origin(34)
|
|
font_s = genshin_font_origin(28)
|
|
|
|
now_time = datetime.now().strftime('%Y/%m/%d')
|
|
event_title_path = TEXT_PATH / 'event_title.png'
|
|
event_title = Image.open(event_title_path)
|
|
event_title_draw = ImageDraw.Draw(event_title)
|
|
event_title_draw.text(
|
|
(7, 380), now_time, font=font_l, fill=text_color, anchor='lm'
|
|
)
|
|
base_img.paste(event_title, (0, 0), event_title)
|
|
|
|
for index, value in enumerate(event_data['normal_event']):
|
|
event_img = Image.new(mode='RGBA', size=(950, 280))
|
|
img = Image.open(BytesIO(get(value['banner']).content))
|
|
img = img.resize((745, 270), Image.Resampling.LANCZOS)
|
|
event_img.paste(img, (205, 10))
|
|
event_img_draw = ImageDraw.Draw(event_img)
|
|
|
|
if isinstance(value['start_time'], str):
|
|
value['start_time'] = await get_month_and_time(value['start_time'])
|
|
if isinstance(value['end_time'], str):
|
|
value['end_time'] = await get_month_and_time(value['end_time'])
|
|
event_img_draw.rectangle([(0, 0), (950, 10)], fill=event_color)
|
|
event_img_draw.polygon(
|
|
[(32, 150), (32, 176), (55, 163)], fill=(243, 110, 110)
|
|
)
|
|
event_img_draw.text(
|
|
(8, 83), value['start_time'][0], text_color, font_l, anchor='lm'
|
|
)
|
|
event_img_draw.text(
|
|
(8, 129), value['start_time'][1], text_color, font_s, anchor='lm'
|
|
)
|
|
event_img_draw.text(
|
|
(39, 213), value['end_time'][0], text_color, font_l, anchor='lm'
|
|
)
|
|
event_img_draw.text(
|
|
(39, 256), value['end_time'][1], text_color, font_s, anchor='lm'
|
|
)
|
|
|
|
base_img.paste(event_img, (0, 450 + 280 * index), event_img)
|
|
|
|
for index, value in enumerate(event_data['gacha_event']):
|
|
event_img = Image.new(mode='RGBA', size=(950, 380))
|
|
img = Image.open(BytesIO(get(value['banner']).content))
|
|
img = img.resize((745, 370), Image.Resampling.LANCZOS)
|
|
event_img.paste(img, (205, 10))
|
|
event_img_draw = ImageDraw.Draw(event_img)
|
|
|
|
event_img_draw.rectangle([(0, 0), (950, 10)], fill=gacha_color)
|
|
event_img_draw.rectangle([(8, 45), (58, 75)], fill=gacha_color)
|
|
event_img_draw.text((65, 60), '祈愿', text_color, font_m, anchor='lm')
|
|
event_img_draw.polygon(
|
|
[(32, 250), (32, 276), (55, 263)], fill=(243, 110, 110)
|
|
)
|
|
event_img_draw.text(
|
|
(8, 183), value['start_time'][0], text_color, font_l, anchor='lm'
|
|
)
|
|
event_img_draw.text(
|
|
(8, 229), value['start_time'][1], text_color, font_s, anchor='lm'
|
|
)
|
|
event_img_draw.text(
|
|
(39, 313), value['end_time'][0], text_color, font_l, anchor='lm'
|
|
)
|
|
event_img_draw.text(
|
|
(39, 356), value['end_time'][1], text_color, font_s, anchor='lm'
|
|
)
|
|
|
|
base_img.paste(
|
|
event_img,
|
|
(0, 450 + len(event_data['normal_event']) * 280 + 380 * index),
|
|
event_img,
|
|
)
|
|
|
|
base_img = base_img.convert('RGB')
|
|
base_img.save(IMG_PATH, format='JPEG', subsampling=0, quality=90)
|
|
return
|