summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorKyomotoi <[email protected]>2022-07-22 19:44:55 +0800
committerKyomotoi <[email protected]>2022-07-22 19:44:55 +0800
commit873288a6b0a8ce1724bf936b2e264cdcdce5b783 (patch)
tree1c133ee506fb198bd621911f9cabe70078558da5
parent7a77b742541a9015818aff5328b783f63dbf2986 (diff)
downloadATRI-873288a6b0a8ce1724bf936b2e264cdcdce5b783.tar.gz
ATRI-873288a6b0a8ce1724bf936b2e264cdcdce5b783.tar.bz2
ATRI-873288a6b0a8ce1724bf936b2e264cdcdce5b783.zip
✨ 完善插件
-rw-r--r--ATRI/plugins/anti_effort/image_dealer.py56
1 files changed, 56 insertions, 0 deletions
diff --git a/ATRI/plugins/anti_effort/image_dealer.py b/ATRI/plugins/anti_effort/image_dealer.py
new file mode 100644
index 0000000..82139c4
--- /dev/null
+++ b/ATRI/plugins/anti_effort/image_dealer.py
@@ -0,0 +1,56 @@
+from pathlib import Path
+from datetime import timedelta, date
+from PIL import Image, ImageFont, ImageDraw
+
+
+PLUGIN_DIR = Path(".") / "data" / "plugins" / "anti_effort"
+TEMP_PATH = Path(".") / "data" / "temp"
+PLUGIN_DIR.mkdir(parents=True, exist_ok=True)
+
+
+def image_dealer(user_img: bytes, user_nickname: str, coding_time: str):
+ user_p = str((TEMP_PATH / f"{user_nickname}-ae.png").absolute())
+
+ with open(user_p, "wb") as w:
+ w.write(user_img)
+
+ bg_size = 1088, 2048
+ bg = Image.new("RGBA", bg_size, color=(0, 0, 0, 0))
+
+ xb_file = PLUGIN_DIR / "xb-bg-0.png"
+ xb_frame = Image.open(xb_file).convert("RGBA")
+ user_i = Image.open(user_p).convert("RGBA").resize((448, 448), Image.ANTIALIAS)
+
+ _, _, _, a = xb_frame.split()
+
+ today = date.today()
+ yd = str(today - timedelta(days=1))
+
+ font_file = str(PLUGIN_DIR / "hwxw.ttf")
+ font_user_nickname = ImageFont.truetype(font_file, 60)
+ font_coding_time = ImageFont.truetype(font_file, 100)
+ font_date = ImageFont.truetype(font_file, 20)
+
+ bg.paste(user_i, (351, 551))
+ bg.paste(xb_frame, (0, 0), a)
+
+ text_nick_size = font_user_nickname.getsize(user_nickname)
+ text_nick_x = int((bg_size[0] - text_nick_size[0] + 68) / 2)
+
+ text_time_size = font_coding_time.getsize(coding_time)
+ text_time_x = int((bg_size[0] - text_time_size[0] + 68) / 2)
+
+ img = ImageDraw.Draw(bg)
+ img.text(
+ (text_nick_x, 1076),
+ user_nickname,
+ font=font_user_nickname,
+ fill=(255, 201, 72),
+ )
+ img.text(
+ (text_time_x, 1564), coding_time, font=font_coding_time, fill=(255, 201, 72)
+ )
+ img.text((405, 1156), yd, font=font_date, fill=(255, 201, 72))
+
+ bg.save(user_p)
+ return user_p