143 lines
7.1 KiB
Python
143 lines
7.1 KiB
Python
import time
|
|
import os
|
|
from re import findall
|
|
from bs4 import BeautifulSoup
|
|
from requests import get
|
|
from io import BytesIO
|
|
from PIL import Image, ImageDraw, ImageFont
|
|
from defs.redis_load import redis_status, redis
|
|
|
|
|
|
def ys_font(size):
|
|
return ImageFont.truetype(f"assets{os.sep}fonts{os.sep}ZhuZiAWan-2.ttc", size=size)
|
|
|
|
|
|
def generate_event():
|
|
# 生成图片到 temp 文件夹
|
|
TEXT_PATH = f"assets{os.sep}event"
|
|
raw_data = get(
|
|
"https://hk4e-api.mihoyo.com/common/hk4e_cn/announcement/api/getAnnList?game=hk4e&game_biz=hk4e_cn&lang=zh-cn"
|
|
"&bundle_id=hk4e_cn&platform=pc®ion=cn_gf01&level=55&uid=100000000").json()
|
|
# now_time = datetime.datetime.now().strftime('%Y-%m-%d')
|
|
# raw_time_data = get(
|
|
# "https://api-takumi.mihoyo.com/event/bbs_activity_calendar/getActList?"
|
|
# "time={}&game_biz=ys_cn&page=1&tag_id=0".format(now_time)).json()
|
|
raw_time_data = get("https://hk4e-api.mihoyo.com/common/hk4e_cn/announcement/api/"
|
|
"getAnnContent?game=hk4e&game_biz=hk4e_cn&lang=zh-cn&bundle_id=hk4e_cn"
|
|
"&platform=pc®ion=cn_gf01&level=55&uid=100000000").json()
|
|
|
|
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("<[a-zA-Z]+.*?>([\s\S]*?)</[a-zA-Z]*?>", time_data)[0]
|
|
k["time_data"] = time_data
|
|
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("<[a-zA-Z]+.*?>([\s\S]*?)</[a-zA-Z]*?>", s)[0])
|
|
else:
|
|
time_datas.append(s)
|
|
k["time_data"] = "——".join(time_datas)
|
|
else:
|
|
k["time_data"] = time_data
|
|
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("<[a-zA-Z]+.*?>([\s\S]*?)</[a-zA-Z]*?>", start_time)[0]
|
|
end_time = findall("<[a-zA-Z]+.*?>([\s\S]*?)</[a-zA-Z]*?>",
|
|
content_bs.find_all("tr")[1].td.find_all("p")[2].text)[0]
|
|
if "<t class=" in end_time:
|
|
end_time = findall("<[a-zA-Z]+.*?>([\s\S]*?)</[a-zA-Z]*?>", end_time)[0]
|
|
time_data = start_time + "——" + end_time
|
|
k["time_data"] = time_data
|
|
|
|
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 = 900 + ((1 + (len(event_data['normal_event'])+len(event_data['other_event'])))//2)*390 +
|
|
# ((1 + len(event_data['gacha_event']))//2)*533
|
|
base_h = 600 + len(event_data['normal_event']) * (390 + 90) + len(event_data['gacha_event']) * (533 + 90)
|
|
base_img = Image.new(mode="RGB", size=(1080, base_h), color=(237, 217, 195))
|
|
|
|
event1_path = os.path.join(TEXT_PATH, "event_1.png")
|
|
event2_path = os.path.join(TEXT_PATH, "event_2.png")
|
|
# event3_path = os.path.join(TEXT_PATH,"event_3.png")
|
|
event1 = Image.open(event1_path)
|
|
event2 = Image.open(event2_path)
|
|
# event3 = Image.open(event3_path)
|
|
|
|
base_img.paste(event1, (0, 0), event1)
|
|
# base_img.paste(event2,(0,300+((1+len(event_data['normal_event']))//2)*390),event2)
|
|
base_img.paste(event2, (0, len(event_data['normal_event']) * (390 + 90) + 300), event2)
|
|
# base_img.paste(event3,(0,600+((1+len(event_data['normal_event']))//2)*390 + ((1 +
|
|
# len(event_data['gacha_event']))//2)*533),event3)
|
|
|
|
time_img1 = Image.new(mode="RGB", size=(1080, len(event_data['normal_event']) * (390 + 90)),
|
|
color=(237, 130, 116))
|
|
time_img2 = Image.new(mode="RGB", size=(1080, len(event_data['gacha_event']) * (533 + 90)),
|
|
color=(237, 130, 116))
|
|
base_img.paste(time_img1, (0, 300))
|
|
base_img.paste(time_img2, (0, 600 + len(event_data['normal_event']) * (390 + 90)))
|
|
base_draw = ImageDraw.Draw(base_img)
|
|
for index, value in enumerate(event_data['normal_event']):
|
|
img = Image.open(BytesIO(get(value["banner"]).content))
|
|
base_draw.text((540, 300 + 45 + 390 + (390 + 90) * index + 1),
|
|
value["time_data"], (255, 255, 255), ys_font(42),
|
|
anchor="mm")
|
|
# base_img.paste(img,((index%2)*1080,300 + 390*(index//2)))
|
|
base_img.paste(img, (0, 300 + (390 + 90) * index))
|
|
|
|
for index, value in enumerate(event_data['gacha_event']):
|
|
img = Image.open(BytesIO(get(value["banner"]).content))
|
|
base_draw.text((540, 600 + 45 + (390 + 90) * len(event_data['normal_event']) + 533 + index * (533 + 90)),
|
|
value["time_data"], (255, 255, 255), ys_font(42),
|
|
anchor="mm")
|
|
# base_img.paste(img,((index%2)*1080,600 + ((1 + len(event_data['normal_event']))//2)*390 +
|
|
# 533*(index//2)))
|
|
base_img.paste(img, (0, 600 + (390 + 90) * len(event_data['normal_event']) + index * (533 + 90)))
|
|
# for index,value in enumerate(event_data['other_event']):
|
|
# img = Image.open(BytesIO(get(value["banner"]).content))
|
|
# base_img.paste(img,((index%2)*1080,900 + ((1 + len(event_data['normal_event']))//2)*390 +
|
|
# ((1 + len(event_data['gacha_event']))//2)*533 + 390*(index//2)))
|
|
|
|
base_img = base_img.convert('RGB')
|
|
base_img.save(f'temp{os.sep}event.jpg', format='JPEG', subsampling=0, quality=90)
|
|
|
|
# 更新缓存的日程的更新日期
|
|
if redis_status():
|
|
redis.set('event', time.strftime("%Y-%m-%d"))
|
|
|
|
|
|
def get_event_image():
|
|
# 判断是否需要重新生成事件，无 redis 每次生成。
|
|
if redis_status():
|
|
try:
|
|
date = redis.get('event').decode()
|
|
except AttributeError:
|
|
date = None
|
|
if not date == time.strftime("%Y-%m-%d"):
|
|
generate_event()
|
|
return f'temp{os.sep}event.jpg'
|
|
else:
|
|
return redis.get('event_file_id').decode()
|
|
else:
|
|
generate_event()
|
|
return f'temp{os.sep}event.jpg'
|