mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-10-06 20:29:59 +08:00
update 0.0.6.2
This commit is contained in:
HibiKi
Regular → Executable
Regular → Executable
+65
-65
@@ -1,65 +1,65 @@
|
||||
from configs.path_config import IMAGE_PATH
|
||||
from pathlib import Path
|
||||
|
||||
|
||||
SIGN_RESOURCE_PATH = Path(IMAGE_PATH) / 'sign' / 'sign_res'
|
||||
SIGN_TODAY_CARD_PATH = Path(IMAGE_PATH) / 'sign' / 'today_card'
|
||||
SIGN_BORDER_PATH = Path(SIGN_RESOURCE_PATH) / 'border'
|
||||
SIGN_BACKGROUND_PATH = Path(SIGN_RESOURCE_PATH) / 'background'
|
||||
|
||||
SIGN_BORDER_PATH.mkdir(exist_ok=True, parents=True)
|
||||
SIGN_BACKGROUND_PATH.mkdir(exist_ok=True, parents=True)
|
||||
|
||||
|
||||
lik2relation = {
|
||||
'0': '路人',
|
||||
'1': '陌生',
|
||||
'2': '初识',
|
||||
'3': '普通',
|
||||
'4': '熟悉',
|
||||
'5': '信赖',
|
||||
'6': '相知',
|
||||
'7': '厚谊',
|
||||
'8': '亲密'
|
||||
}
|
||||
|
||||
level2attitude = {
|
||||
'0': '排斥',
|
||||
'1': '警惕',
|
||||
'2': '可以交流',
|
||||
'3': '一般',
|
||||
'4': '是个好人',
|
||||
'5': '好朋友',
|
||||
'6': '可以分享小秘密',
|
||||
'7': '喜欢',
|
||||
'8': '恋人'
|
||||
}
|
||||
|
||||
weekdays = {
|
||||
1: 'Mon',
|
||||
2: 'Tue',
|
||||
3: 'Wed',
|
||||
4: 'Thu',
|
||||
5: 'Fri',
|
||||
6: 'Sat',
|
||||
7: 'Sun'
|
||||
}
|
||||
|
||||
lik2level = {
|
||||
9999: '9',
|
||||
400: '8',
|
||||
270: '7',
|
||||
200: '6',
|
||||
140: '5',
|
||||
90: '4',
|
||||
50: '3',
|
||||
25: '2',
|
||||
10: '1',
|
||||
0: '0'
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
from configs.path_config import IMAGE_PATH
|
||||
from pathlib import Path
|
||||
|
||||
|
||||
SIGN_RESOURCE_PATH = Path(IMAGE_PATH) / 'sign' / 'sign_res'
|
||||
SIGN_TODAY_CARD_PATH = Path(IMAGE_PATH) / 'sign' / 'today_card'
|
||||
SIGN_BORDER_PATH = Path(SIGN_RESOURCE_PATH) / 'border'
|
||||
SIGN_BACKGROUND_PATH = Path(SIGN_RESOURCE_PATH) / 'background'
|
||||
|
||||
SIGN_BORDER_PATH.mkdir(exist_ok=True, parents=True)
|
||||
SIGN_BACKGROUND_PATH.mkdir(exist_ok=True, parents=True)
|
||||
|
||||
|
||||
lik2relation = {
|
||||
'0': '路人',
|
||||
'1': '陌生',
|
||||
'2': '初识',
|
||||
'3': '普通',
|
||||
'4': '熟悉',
|
||||
'5': '信赖',
|
||||
'6': '相知',
|
||||
'7': '厚谊',
|
||||
'8': '亲密'
|
||||
}
|
||||
|
||||
level2attitude = {
|
||||
'0': '排斥',
|
||||
'1': '警惕',
|
||||
'2': '可以交流',
|
||||
'3': '一般',
|
||||
'4': '是个好人',
|
||||
'5': '好朋友',
|
||||
'6': '可以分享小秘密',
|
||||
'7': '喜欢',
|
||||
'8': '恋人'
|
||||
}
|
||||
|
||||
weekdays = {
|
||||
1: 'Mon',
|
||||
2: 'Tue',
|
||||
3: 'Wed',
|
||||
4: 'Thu',
|
||||
5: 'Fri',
|
||||
6: 'Sat',
|
||||
7: 'Sun'
|
||||
}
|
||||
|
||||
lik2level = {
|
||||
9999: '9',
|
||||
400: '8',
|
||||
270: '7',
|
||||
200: '6',
|
||||
140: '5',
|
||||
90: '4',
|
||||
50: '3',
|
||||
25: '2',
|
||||
10: '1',
|
||||
0: '0'
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
Regular → Executable
+39
-43
@@ -4,7 +4,6 @@ from models.group_member_info import GroupInfoUser
|
||||
from models.bag_user import BagUser
|
||||
from configs.config import NICKNAME
|
||||
from nonebot.adapters.cqhttp import MessageSegment
|
||||
from asyncio.exceptions import TimeoutError
|
||||
from utils.image_utils import CreateImg, CreateMat
|
||||
from services.db_context import db
|
||||
from .utils import get_card, SIGN_TODAY_CARD_PATH
|
||||
@@ -12,9 +11,9 @@ from typing import Optional
|
||||
from services.log import logger
|
||||
from .random_event import random_event
|
||||
from utils.data_utils import init_rank
|
||||
from utils.utils import get_user_avatar
|
||||
from io import BytesIO
|
||||
import random
|
||||
import aiohttp
|
||||
import math
|
||||
import asyncio
|
||||
import secrets
|
||||
@@ -142,47 +141,44 @@ async def _pst(users: list, impressions: list, groups: list):
|
||||
width = 10
|
||||
idx = 0
|
||||
A = CreateImg(1740, 3300, color="#FFE4C4")
|
||||
async with aiohttp.ClientSession() as session:
|
||||
for _ in range(count):
|
||||
col_img = CreateImg(550, 3300, 550, 100, color="#FFE4C4")
|
||||
for _ in range(33 if int(lens / 33) >= 1 else lens % 33 - 1):
|
||||
idx += 1
|
||||
if idx > 100:
|
||||
break
|
||||
impression = max(impressions)
|
||||
index = impressions.index(impression)
|
||||
user = users[index]
|
||||
group = groups[index]
|
||||
impressions.pop(index)
|
||||
users.pop(index)
|
||||
groups.pop(index)
|
||||
try:
|
||||
user_name = (
|
||||
await GroupInfoUser.get_member_info(user, group)
|
||||
).user_name
|
||||
except AttributeError:
|
||||
user_name = f"我名字呢？"
|
||||
user_name = user_name if len(user_name) < 11 else user_name[:10] + "..."
|
||||
try:
|
||||
async with session.get(
|
||||
f"http://q1.qlogo.cn/g?b=qq&nk={user}&s=160", timeout=5
|
||||
) as response:
|
||||
ava = CreateImg(
|
||||
50, 50, background=BytesIO(await response.read())
|
||||
)
|
||||
except TimeoutError:
|
||||
ava = CreateImg(50, 50, color="white")
|
||||
ava.circle()
|
||||
bk = CreateImg(550, 100, color="#FFE4C4", font_size=30)
|
||||
font_w, font_h = bk.getsize(f"{idx}")
|
||||
bk.text((5, int((100 - font_h) / 2)), f"{idx}.")
|
||||
bk.paste(ava, (55, int((100 - 50) / 2)), True)
|
||||
bk.text((120, int((100 - font_h) / 2)), f"{user_name}")
|
||||
bk.text((460, int((100 - font_h) / 2)), f"[{impression:.2f}]")
|
||||
col_img.paste(bk)
|
||||
A.paste(col_img, (width, 0))
|
||||
lens -= 33
|
||||
width += 580
|
||||
for _ in range(count):
|
||||
col_img = CreateImg(550, 3300, 550, 100, color="#FFE4C4")
|
||||
for _ in range(33 if int(lens / 33) >= 1 else lens % 33 - 1):
|
||||
idx += 1
|
||||
if idx > 100:
|
||||
break
|
||||
impression = max(impressions)
|
||||
index = impressions.index(impression)
|
||||
user = users[index]
|
||||
group = groups[index]
|
||||
impressions.pop(index)
|
||||
users.pop(index)
|
||||
groups.pop(index)
|
||||
try:
|
||||
user_name = (
|
||||
await GroupInfoUser.get_member_info(user, group)
|
||||
).user_name
|
||||
except AttributeError:
|
||||
user_name = f"我名字呢？"
|
||||
user_name = user_name if len(user_name) < 11 else user_name[:10] + "..."
|
||||
ava = await get_user_avatar(user)
|
||||
if ava:
|
||||
ava = CreateImg(
|
||||
50, 50, background=BytesIO(ava)
|
||||
)
|
||||
else:
|
||||
ava = CreateImg(50, 50, color="white")
|
||||
ava.circle()
|
||||
bk = CreateImg(550, 100, color="#FFE4C4", font_size=30)
|
||||
font_w, font_h = bk.getsize(f"{idx}")
|
||||
bk.text((5, int((100 - font_h) / 2)), f"{idx}.")
|
||||
bk.paste(ava, (55, int((100 - 50) / 2)), True)
|
||||
bk.text((120, int((100 - font_h) / 2)), f"{user_name}")
|
||||
bk.text((460, int((100 - font_h) / 2)), f"[{impression:.2f}]")
|
||||
col_img.paste(bk)
|
||||
A.paste(col_img, (width, 0))
|
||||
lens -= 33
|
||||
width += 580
|
||||
W = CreateImg(1740, 3700, color="#FFE4C4", font_size=130)
|
||||
W.paste(A, (0, 260))
|
||||
font_w, font_h = W.getsize(f"{NICKNAME}的好感度总榜")
|
||||
|
||||
Regular → Executable
+33
-33
@@ -1,33 +1,33 @@
|
||||
from configs.config import Config
|
||||
import random
|
||||
|
||||
|
||||
PROB_DATA = None
|
||||
|
||||
|
||||
def random_event(impression: float) -> 'Union[str, int], str':
|
||||
"""
|
||||
签到随机事件
|
||||
:param impression: 好感度
|
||||
:return: 额外奖励 和 类型
|
||||
"""
|
||||
global PROB_DATA
|
||||
if not PROB_DATA:
|
||||
PROB_DATA = {
|
||||
Config.get_config("sign_in", "SIGN_CARD3_PROB"): '好感度双倍加持卡Ⅲ',
|
||||
Config.get_config("sign_in", "SIGN_CARD2_PROB"): '好感度双倍加持卡Ⅱ',
|
||||
Config.get_config("sign_in", "SIGN_CARD1_PROB"): '好感度双倍加持卡Ⅰ'
|
||||
}
|
||||
rand = random.random() - impression / 1000
|
||||
for prob in PROB_DATA.keys():
|
||||
if rand <= prob:
|
||||
return PROB_DATA[prob], 'props'
|
||||
gold = random.randint(1, random.randint(1, int(1 if impression < 1 else impression)))
|
||||
max_sign_gold = Config.get_config("sign_in", "MAX_SIGN_GOLD")
|
||||
gold = max_sign_gold if gold > max_sign_gold else gold
|
||||
return gold, 'gold'
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
from configs.config import Config
|
||||
import random
|
||||
|
||||
|
||||
PROB_DATA = None
|
||||
|
||||
|
||||
def random_event(impression: float) -> 'Union[str, int], str':
|
||||
"""
|
||||
签到随机事件
|
||||
:param impression: 好感度
|
||||
:return: 额外奖励 和 类型
|
||||
"""
|
||||
global PROB_DATA
|
||||
if not PROB_DATA:
|
||||
PROB_DATA = {
|
||||
Config.get_config("sign_in", "SIGN_CARD3_PROB"): '好感度双倍加持卡Ⅲ',
|
||||
Config.get_config("sign_in", "SIGN_CARD2_PROB"): '好感度双倍加持卡Ⅱ',
|
||||
Config.get_config("sign_in", "SIGN_CARD1_PROB"): '好感度双倍加持卡Ⅰ'
|
||||
}
|
||||
rand = random.random() - impression / 1000
|
||||
for prob in PROB_DATA.keys():
|
||||
if rand <= prob:
|
||||
return PROB_DATA[prob], 'props'
|
||||
gold = random.randint(1, random.randint(1, int(1 if impression < 1 else impression)))
|
||||
max_sign_gold = Config.get_config("sign_in", "MAX_SIGN_GOLD")
|
||||
gold = max_sign_gold if gold > max_sign_gold else gold
|
||||
return gold, 'gold'
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
Regular → Executable
+352
-352
@@ -1,352 +1,352 @@
|
||||
from .config import (
|
||||
SIGN_RESOURCE_PATH,
|
||||
SIGN_TODAY_CARD_PATH,
|
||||
SIGN_BORDER_PATH,
|
||||
SIGN_BACKGROUND_PATH,
|
||||
lik2level,
|
||||
lik2relation,
|
||||
level2attitude,
|
||||
weekdays,
|
||||
)
|
||||
from models.sign_group_user import SignGroupUser
|
||||
from models.group_member_info import GroupInfoUser
|
||||
from nonebot.adapters.cqhttp import MessageSegment
|
||||
from utils.utils import get_user_avatar
|
||||
from utils.image_utils import CreateImg
|
||||
from utils.message_builder import image
|
||||
from configs.config import NICKNAME
|
||||
from pathlib import Path
|
||||
from datetime import datetime
|
||||
from typing import Optional, List
|
||||
from nonebot import Driver
|
||||
from io import BytesIO
|
||||
import asyncio
|
||||
import random
|
||||
import nonebot
|
||||
import os
|
||||
|
||||
|
||||
driver: Driver = nonebot.get_driver()
|
||||
|
||||
|
||||
@driver.on_startup
|
||||
async def init_image():
|
||||
SIGN_RESOURCE_PATH.mkdir(parents=True, exist_ok=True)
|
||||
SIGN_TODAY_CARD_PATH.mkdir(exist_ok=True, parents=True)
|
||||
await GroupInfoUser.add_member_info(114514, 114514, "", datetime.min, 0)
|
||||
_u = await GroupInfoUser.get_member_info(114514, 114514)
|
||||
if _u.uid is None:
|
||||
await _u.update(uid=0).apply()
|
||||
generate_progress_bar_pic()
|
||||
clear_sign_data_pic()
|
||||
|
||||
|
||||
async def get_card(
|
||||
user: "SignGroupUser",
|
||||
nickname: str,
|
||||
add_impression: Optional[float],
|
||||
gold: Optional[int],
|
||||
gift: str,
|
||||
is_double: bool = False,
|
||||
is_card_view: bool = False,
|
||||
) -> MessageSegment:
|
||||
user_id = user.user_qq
|
||||
date = datetime.now().date()
|
||||
_type = "view" if is_card_view else "sign"
|
||||
card_file = (
|
||||
Path(SIGN_TODAY_CARD_PATH)
|
||||
/ f"{user_id}_{user.belonging_group}_{_type}_{date}.png"
|
||||
)
|
||||
if card_file.exists():
|
||||
return image(
|
||||
f"{user_id}_{user.belonging_group}_{_type}_{date}.png", "sign/today_card"
|
||||
)
|
||||
else:
|
||||
if add_impression == -1:
|
||||
card_file = (
|
||||
Path(SIGN_TODAY_CARD_PATH)
|
||||
/ f"{user_id}_{user.belonging_group}_view_{date}.png"
|
||||
)
|
||||
if card_file.exists():
|
||||
return image(
|
||||
f"{user_id}_{user.belonging_group}_view_{date}.png",
|
||||
"sign/today_card",
|
||||
)
|
||||
is_card_view = True
|
||||
ava = BytesIO(await get_user_avatar(user_id))
|
||||
uid = await GroupInfoUser.get_group_member_uid(
|
||||
user.user_qq, user.belonging_group
|
||||
)
|
||||
impression_list = None
|
||||
if is_card_view:
|
||||
_, impression_list, _ = await SignGroupUser.get_all_impression(
|
||||
user.belonging_group
|
||||
)
|
||||
return await asyncio.get_event_loop().run_in_executor(
|
||||
None,
|
||||
_generate_card,
|
||||
user,
|
||||
nickname,
|
||||
user_id,
|
||||
add_impression,
|
||||
gold,
|
||||
gift,
|
||||
uid,
|
||||
ava,
|
||||
impression_list,
|
||||
is_double,
|
||||
is_card_view,
|
||||
)
|
||||
|
||||
|
||||
def _generate_card(
|
||||
user: "SignGroupUser",
|
||||
nickname: str,
|
||||
user_id: int,
|
||||
impression: Optional[float],
|
||||
gold: Optional[int],
|
||||
gift: str,
|
||||
uid: str,
|
||||
ava_bytes: BytesIO,
|
||||
impression_list: List[float],
|
||||
is_double: bool = False,
|
||||
is_card_view: bool = False,
|
||||
) -> MessageSegment:
|
||||
ava_bk = CreateImg(140, 140, is_alpha=True)
|
||||
ava_border = CreateImg(
|
||||
140,
|
||||
140,
|
||||
background=SIGN_BORDER_PATH / "ava_border_01.png",
|
||||
)
|
||||
ava = CreateImg(102, 102, background=ava_bytes)
|
||||
ava.circle()
|
||||
ava_bk.paste(ava, center_type="center")
|
||||
ava_bk.paste(ava_border, alpha=True, center_type="center")
|
||||
|
||||
info_img = CreateImg(250, 150, color=(255, 255, 255, 0), font_size=15)
|
||||
level, next_impression, previous_impression = get_level_and_next_impression(
|
||||
user.impression
|
||||
)
|
||||
info_img.text((0, 0), f"· 好感度等级：{level} [{lik2relation[level]}]")
|
||||
info_img.text((0, 20), f"· {NICKNAME}对你的态度：{level2attitude[level]}")
|
||||
info_img.text((0, 40), f"· 距离升级还差 {next_impression - user.impression:.2f} 好感度")
|
||||
|
||||
bar_bk = CreateImg(220, 20, background=SIGN_RESOURCE_PATH / "bar_white.png")
|
||||
bar = CreateImg(220, 20, background=SIGN_RESOURCE_PATH / "bar.png")
|
||||
bar_bk.paste(
|
||||
bar,
|
||||
(
|
||||
-int(
|
||||
220
|
||||
* (
|
||||
(next_impression - user.impression)
|
||||
/ (next_impression - previous_impression)
|
||||
)
|
||||
),
|
||||
0,
|
||||
),
|
||||
True,
|
||||
)
|
||||
font_size = 30
|
||||
if "好感度双倍加持卡" in gift:
|
||||
font_size = 20
|
||||
gift_border = CreateImg(
|
||||
270,
|
||||
100,
|
||||
background=SIGN_BORDER_PATH / "gift_border_02.png",
|
||||
font_size=font_size,
|
||||
)
|
||||
gift_border.text((0, 0), gift, center_type="center")
|
||||
|
||||
bk = CreateImg(
|
||||
876,
|
||||
424,
|
||||
background=SIGN_BACKGROUND_PATH
|
||||
/ random.choice(os.listdir(SIGN_BACKGROUND_PATH)),
|
||||
font_size=25,
|
||||
)
|
||||
A = CreateImg(876, 274, background=SIGN_RESOURCE_PATH / "white.png")
|
||||
line = CreateImg(2, 180, color="black")
|
||||
A.transparent(2)
|
||||
A.paste(ava_bk, (25, 80), True)
|
||||
A.paste(line, (200, 70))
|
||||
|
||||
nickname_img = CreateImg(
|
||||
0,
|
||||
0,
|
||||
plain_text=nickname,
|
||||
color=(255, 255, 255, 0),
|
||||
font_size=50,
|
||||
font_color=(255, 255, 255),
|
||||
)
|
||||
if uid:
|
||||
uid = f"{uid}".rjust(12, "0")
|
||||
uid = uid[:4] + " " + uid[4:8] + " " + uid[8:]
|
||||
else:
|
||||
uid = "XXXX XXXX XXXX"
|
||||
uid_img = CreateImg(
|
||||
0,
|
||||
0,
|
||||
plain_text=f"UID: {uid}",
|
||||
color=(255, 255, 255, 0),
|
||||
font_size=30,
|
||||
font_color=(255, 255, 255),
|
||||
)
|
||||
sign_day_img = CreateImg(
|
||||
0,
|
||||
0,
|
||||
plain_text=f"{user.checkin_count}",
|
||||
color=(255, 255, 255, 0),
|
||||
font_size=40,
|
||||
font_color=(211, 64, 33),
|
||||
)
|
||||
lik_text1_img = CreateImg(
|
||||
0, 0, plain_text="当前", color=(255, 255, 255, 0), font_size=20
|
||||
)
|
||||
lik_text2_img = CreateImg(
|
||||
0,
|
||||
0,
|
||||
plain_text=f"好感度：{user.impression:.2f}",
|
||||
color=(255, 255, 255, 0),
|
||||
font_size=30,
|
||||
)
|
||||
watermark = CreateImg(
|
||||
0,
|
||||
0,
|
||||
plain_text=f"{NICKNAME}@{datetime.now().year}",
|
||||
color=(255, 255, 255, 0),
|
||||
font_size=15,
|
||||
font_color=(155, 155, 155),
|
||||
)
|
||||
today_data = CreateImg(300, 300, color=(255, 255, 255, 0), font_size=20)
|
||||
if is_card_view:
|
||||
today_sign_text_img = CreateImg(
|
||||
0, 0, plain_text="", color=(255, 255, 255, 0), font_size=30
|
||||
)
|
||||
if impression_list:
|
||||
impression_list.sort(reverse=True)
|
||||
index = impression_list.index(user.impression)
|
||||
rank_img = CreateImg(
|
||||
0,
|
||||
0,
|
||||
plain_text=f"* 此群好感排名第 {index + 1} 位",
|
||||
color=(255, 255, 255, 0),
|
||||
font_size=30,
|
||||
)
|
||||
A.paste(rank_img, ((A.w - rank_img.w - 10), 20), True)
|
||||
today_data.text(
|
||||
(0, 0),
|
||||
f"上次签到日期：{'从未' if user.checkin_time_last == datetime.min else user.checkin_time_last.date()}",
|
||||
)
|
||||
today_data.text((0, 25), f"总金币：{gold}")
|
||||
today_data.text(
|
||||
(0, 50),
|
||||
f"色图概率：{(70 + user.impression if user.impression < 100 else 100):.2f}%",
|
||||
)
|
||||
today_data.text((0, 75), f"开箱次数：{(20 + int(user.impression / 3))}")
|
||||
_type = "view"
|
||||
else:
|
||||
A.paste(gift_border, (570, 140), True)
|
||||
today_sign_text_img = CreateImg(
|
||||
0, 0, plain_text="今日签到", color=(255, 255, 255, 0), font_size=30
|
||||
)
|
||||
if is_double:
|
||||
today_data.text((0, 0), f"好感度 + {impression / 2:.2f} × 2")
|
||||
else:
|
||||
today_data.text((0, 0), f"好感度 + {impression:.2f}")
|
||||
today_data.text((0, 25), f"金币 + {gold}")
|
||||
_type = "sign"
|
||||
current_date = datetime.now()
|
||||
week = current_date.isoweekday()
|
||||
data = current_date.date()
|
||||
hour = current_date.hour
|
||||
minute = current_date.minute
|
||||
second = current_date.second
|
||||
data_img = CreateImg(
|
||||
0,
|
||||
0,
|
||||
plain_text=f"时间：{data} {weekdays[week]} {hour}:{minute}:{second}",
|
||||
color=(255, 255, 255, 0),
|
||||
font_size=20,
|
||||
)
|
||||
bk.paste(nickname_img, (30, 15), True)
|
||||
bk.paste(uid_img, (30, 85), True)
|
||||
bk.paste(A, (0, 150), alpha=True)
|
||||
bk.text((30, 167), "Accumulative check-in for")
|
||||
_x = bk.getsize("Accumulative check-in for")[0] + sign_day_img.w + 45
|
||||
bk.paste(sign_day_img, (346, 158), True)
|
||||
bk.text((_x, 167), "days")
|
||||
bk.paste(data_img, (220, 370), True)
|
||||
bk.paste(lik_text1_img, (220, 240), True)
|
||||
bk.paste(lik_text2_img, (262, 234), True)
|
||||
bk.paste(bar_bk, (225, 275), True)
|
||||
bk.paste(info_img, (220, 305), True)
|
||||
bk.paste(today_sign_text_img, (550, 180), True)
|
||||
bk.paste(today_data, (580, 220), True)
|
||||
bk.paste(watermark, (15, 400), True)
|
||||
bk.save(
|
||||
SIGN_TODAY_CARD_PATH / f"{user_id}_{user.belonging_group}_{_type}_{data}.png"
|
||||
)
|
||||
return image(
|
||||
f"{user_id}_{user.belonging_group}_{_type}_{data}.png", "sign/today_card"
|
||||
)
|
||||
|
||||
|
||||
def generate_progress_bar_pic():
|
||||
bg_2 = (254, 1, 254)
|
||||
bg_1 = (0, 245, 246)
|
||||
|
||||
bk = CreateImg(1000, 50, is_alpha=True)
|
||||
img_x = CreateImg(50, 50, color=bg_2)
|
||||
img_x.circle()
|
||||
img_x.crop((25, 0, 50, 50))
|
||||
img_y = CreateImg(50, 50, color=bg_1)
|
||||
img_y.circle()
|
||||
img_y.crop((0, 0, 25, 50))
|
||||
A = CreateImg(950, 50)
|
||||
width, height = A.size
|
||||
|
||||
step_r = (bg_2[0] - bg_1[0]) / width
|
||||
step_g = (bg_2[1] - bg_1[1]) / width
|
||||
step_b = (bg_2[2] - bg_1[2]) / width
|
||||
|
||||
for y in range(0, width):
|
||||
bg_r = round(bg_1[0] + step_r * y)
|
||||
bg_g = round(bg_1[1] + step_g * y)
|
||||
bg_b = round(bg_1[2] + step_b * y)
|
||||
for x in range(0, height):
|
||||
A.point((y, x), fill=(bg_r, bg_g, bg_b))
|
||||
bk.paste(img_y, (0, 0), True)
|
||||
bk.paste(A, (25, 0))
|
||||
bk.paste(img_x, (975, 0), True)
|
||||
bk.save(SIGN_RESOURCE_PATH / "bar.png")
|
||||
|
||||
A = CreateImg(950, 50)
|
||||
bk = CreateImg(1000, 50, is_alpha=True)
|
||||
img_x = CreateImg(50, 50)
|
||||
img_x.circle()
|
||||
img_x.crop((25, 0, 50, 50))
|
||||
img_y = CreateImg(50, 50)
|
||||
img_y.circle()
|
||||
img_y.crop((0, 0, 25, 50))
|
||||
bk.paste(img_y, (0, 0), True)
|
||||
bk.paste(A, (25, 0))
|
||||
bk.paste(img_x, (975, 0), True)
|
||||
bk.save(SIGN_RESOURCE_PATH / "bar_white.png")
|
||||
|
||||
|
||||
def get_level_and_next_impression(impression: float):
|
||||
if impression == 0:
|
||||
return lik2level[10], 10, 0
|
||||
keys = list(lik2level.keys())
|
||||
for i in range(len(keys)):
|
||||
if impression > keys[i]:
|
||||
return lik2level[keys[i]], keys[i - 1], keys[i]
|
||||
return lik2level[10], 10, 0
|
||||
|
||||
|
||||
def clear_sign_data_pic():
|
||||
date = datetime.now().date()
|
||||
for file in os.listdir(SIGN_TODAY_CARD_PATH):
|
||||
if str(date) not in file:
|
||||
os.remove(SIGN_TODAY_CARD_PATH / file)
|
||||
from .config import (
|
||||
SIGN_RESOURCE_PATH,
|
||||
SIGN_TODAY_CARD_PATH,
|
||||
SIGN_BORDER_PATH,
|
||||
SIGN_BACKGROUND_PATH,
|
||||
lik2level,
|
||||
lik2relation,
|
||||
level2attitude,
|
||||
weekdays,
|
||||
)
|
||||
from models.sign_group_user import SignGroupUser
|
||||
from models.group_member_info import GroupInfoUser
|
||||
from nonebot.adapters.cqhttp import MessageSegment
|
||||
from utils.utils import get_user_avatar
|
||||
from utils.image_utils import CreateImg
|
||||
from utils.message_builder import image
|
||||
from configs.config import NICKNAME
|
||||
from pathlib import Path
|
||||
from datetime import datetime
|
||||
from typing import Optional, List
|
||||
from nonebot import Driver
|
||||
from io import BytesIO
|
||||
import asyncio
|
||||
import random
|
||||
import nonebot
|
||||
import os
|
||||
|
||||
|
||||
driver: Driver = nonebot.get_driver()
|
||||
|
||||
|
||||
@driver.on_startup
|
||||
async def init_image():
|
||||
SIGN_RESOURCE_PATH.mkdir(parents=True, exist_ok=True)
|
||||
SIGN_TODAY_CARD_PATH.mkdir(exist_ok=True, parents=True)
|
||||
await GroupInfoUser.add_member_info(114514, 114514, "", datetime.min, 0)
|
||||
_u = await GroupInfoUser.get_member_info(114514, 114514)
|
||||
if _u.uid is None:
|
||||
await _u.update(uid=0).apply()
|
||||
generate_progress_bar_pic()
|
||||
clear_sign_data_pic()
|
||||
|
||||
|
||||
async def get_card(
|
||||
user: "SignGroupUser",
|
||||
nickname: str,
|
||||
add_impression: Optional[float],
|
||||
gold: Optional[int],
|
||||
gift: str,
|
||||
is_double: bool = False,
|
||||
is_card_view: bool = False,
|
||||
) -> MessageSegment:
|
||||
user_id = user.user_qq
|
||||
date = datetime.now().date()
|
||||
_type = "view" if is_card_view else "sign"
|
||||
card_file = (
|
||||
Path(SIGN_TODAY_CARD_PATH)
|
||||
/ f"{user_id}_{user.belonging_group}_{_type}_{date}.png"
|
||||
)
|
||||
if card_file.exists():
|
||||
return image(
|
||||
f"{user_id}_{user.belonging_group}_{_type}_{date}.png", "sign/today_card"
|
||||
)
|
||||
else:
|
||||
if add_impression == -1:
|
||||
card_file = (
|
||||
Path(SIGN_TODAY_CARD_PATH)
|
||||
/ f"{user_id}_{user.belonging_group}_view_{date}.png"
|
||||
)
|
||||
if card_file.exists():
|
||||
return image(
|
||||
f"{user_id}_{user.belonging_group}_view_{date}.png",
|
||||
"sign/today_card",
|
||||
)
|
||||
is_card_view = True
|
||||
ava = BytesIO(await get_user_avatar(user_id))
|
||||
uid = await GroupInfoUser.get_group_member_uid(
|
||||
user.user_qq, user.belonging_group
|
||||
)
|
||||
impression_list = None
|
||||
if is_card_view:
|
||||
_, impression_list, _ = await SignGroupUser.get_all_impression(
|
||||
user.belonging_group
|
||||
)
|
||||
return await asyncio.get_event_loop().run_in_executor(
|
||||
None,
|
||||
_generate_card,
|
||||
user,
|
||||
nickname,
|
||||
user_id,
|
||||
add_impression,
|
||||
gold,
|
||||
gift,
|
||||
uid,
|
||||
ava,
|
||||
impression_list,
|
||||
is_double,
|
||||
is_card_view,
|
||||
)
|
||||
|
||||
|
||||
def _generate_card(
|
||||
user: "SignGroupUser",
|
||||
nickname: str,
|
||||
user_id: int,
|
||||
impression: Optional[float],
|
||||
gold: Optional[int],
|
||||
gift: str,
|
||||
uid: str,
|
||||
ava_bytes: BytesIO,
|
||||
impression_list: List[float],
|
||||
is_double: bool = False,
|
||||
is_card_view: bool = False,
|
||||
) -> MessageSegment:
|
||||
ava_bk = CreateImg(140, 140, is_alpha=True)
|
||||
ava_border = CreateImg(
|
||||
140,
|
||||
140,
|
||||
background=SIGN_BORDER_PATH / "ava_border_01.png",
|
||||
)
|
||||
ava = CreateImg(102, 102, background=ava_bytes)
|
||||
ava.circle()
|
||||
ava_bk.paste(ava, center_type="center")
|
||||
ava_bk.paste(ava_border, alpha=True, center_type="center")
|
||||
|
||||
info_img = CreateImg(250, 150, color=(255, 255, 255, 0), font_size=15)
|
||||
level, next_impression, previous_impression = get_level_and_next_impression(
|
||||
user.impression
|
||||
)
|
||||
info_img.text((0, 0), f"· 好感度等级：{level} [{lik2relation[level]}]")
|
||||
info_img.text((0, 20), f"· {NICKNAME}对你的态度：{level2attitude[level]}")
|
||||
info_img.text((0, 40), f"· 距离升级还差 {next_impression - user.impression:.2f} 好感度")
|
||||
|
||||
bar_bk = CreateImg(220, 20, background=SIGN_RESOURCE_PATH / "bar_white.png")
|
||||
bar = CreateImg(220, 20, background=SIGN_RESOURCE_PATH / "bar.png")
|
||||
bar_bk.paste(
|
||||
bar,
|
||||
(
|
||||
-int(
|
||||
220
|
||||
* (
|
||||
(next_impression - user.impression)
|
||||
/ (next_impression - previous_impression)
|
||||
)
|
||||
),
|
||||
0,
|
||||
),
|
||||
True,
|
||||
)
|
||||
font_size = 30
|
||||
if "好感度双倍加持卡" in gift:
|
||||
font_size = 20
|
||||
gift_border = CreateImg(
|
||||
270,
|
||||
100,
|
||||
background=SIGN_BORDER_PATH / "gift_border_02.png",
|
||||
font_size=font_size,
|
||||
)
|
||||
gift_border.text((0, 0), gift, center_type="center")
|
||||
|
||||
bk = CreateImg(
|
||||
876,
|
||||
424,
|
||||
background=SIGN_BACKGROUND_PATH
|
||||
/ random.choice(os.listdir(SIGN_BACKGROUND_PATH)),
|
||||
font_size=25,
|
||||
)
|
||||
A = CreateImg(876, 274, background=SIGN_RESOURCE_PATH / "white.png")
|
||||
line = CreateImg(2, 180, color="black")
|
||||
A.transparent(2)
|
||||
A.paste(ava_bk, (25, 80), True)
|
||||
A.paste(line, (200, 70))
|
||||
|
||||
nickname_img = CreateImg(
|
||||
0,
|
||||
0,
|
||||
plain_text=nickname,
|
||||
color=(255, 255, 255, 0),
|
||||
font_size=50,
|
||||
font_color=(255, 255, 255),
|
||||
)
|
||||
if uid:
|
||||
uid = f"{uid}".rjust(12, "0")
|
||||
uid = uid[:4] + " " + uid[4:8] + " " + uid[8:]
|
||||
else:
|
||||
uid = "XXXX XXXX XXXX"
|
||||
uid_img = CreateImg(
|
||||
0,
|
||||
0,
|
||||
plain_text=f"UID: {uid}",
|
||||
color=(255, 255, 255, 0),
|
||||
font_size=30,
|
||||
font_color=(255, 255, 255),
|
||||
)
|
||||
sign_day_img = CreateImg(
|
||||
0,
|
||||
0,
|
||||
plain_text=f"{user.checkin_count}",
|
||||
color=(255, 255, 255, 0),
|
||||
font_size=40,
|
||||
font_color=(211, 64, 33),
|
||||
)
|
||||
lik_text1_img = CreateImg(
|
||||
0, 0, plain_text="当前", color=(255, 255, 255, 0), font_size=20
|
||||
)
|
||||
lik_text2_img = CreateImg(
|
||||
0,
|
||||
0,
|
||||
plain_text=f"好感度：{user.impression:.2f}",
|
||||
color=(255, 255, 255, 0),
|
||||
font_size=30,
|
||||
)
|
||||
watermark = CreateImg(
|
||||
0,
|
||||
0,
|
||||
plain_text=f"{NICKNAME}@{datetime.now().year}",
|
||||
color=(255, 255, 255, 0),
|
||||
font_size=15,
|
||||
font_color=(155, 155, 155),
|
||||
)
|
||||
today_data = CreateImg(300, 300, color=(255, 255, 255, 0), font_size=20)
|
||||
if is_card_view:
|
||||
today_sign_text_img = CreateImg(
|
||||
0, 0, plain_text="", color=(255, 255, 255, 0), font_size=30
|
||||
)
|
||||
if impression_list:
|
||||
impression_list.sort(reverse=True)
|
||||
index = impression_list.index(user.impression)
|
||||
rank_img = CreateImg(
|
||||
0,
|
||||
0,
|
||||
plain_text=f"* 此群好感排名第 {index + 1} 位",
|
||||
color=(255, 255, 255, 0),
|
||||
font_size=30,
|
||||
)
|
||||
A.paste(rank_img, ((A.w - rank_img.w - 10), 20), True)
|
||||
today_data.text(
|
||||
(0, 0),
|
||||
f"上次签到日期：{'从未' if user.checkin_time_last == datetime.min else user.checkin_time_last.date()}",
|
||||
)
|
||||
today_data.text((0, 25), f"总金币：{gold}")
|
||||
today_data.text(
|
||||
(0, 50),
|
||||
f"色图概率：{(70 + user.impression if user.impression < 100 else 100):.2f}%",
|
||||
)
|
||||
today_data.text((0, 75), f"开箱次数：{(20 + int(user.impression / 3))}")
|
||||
_type = "view"
|
||||
else:
|
||||
A.paste(gift_border, (570, 140), True)
|
||||
today_sign_text_img = CreateImg(
|
||||
0, 0, plain_text="今日签到", color=(255, 255, 255, 0), font_size=30
|
||||
)
|
||||
if is_double:
|
||||
today_data.text((0, 0), f"好感度 + {impression / 2:.2f} × 2")
|
||||
else:
|
||||
today_data.text((0, 0), f"好感度 + {impression:.2f}")
|
||||
today_data.text((0, 25), f"金币 + {gold}")
|
||||
_type = "sign"
|
||||
current_date = datetime.now()
|
||||
week = current_date.isoweekday()
|
||||
data = current_date.date()
|
||||
hour = current_date.hour
|
||||
minute = current_date.minute
|
||||
second = current_date.second
|
||||
data_img = CreateImg(
|
||||
0,
|
||||
0,
|
||||
plain_text=f"时间：{data} {weekdays[week]} {hour}:{minute}:{second}",
|
||||
color=(255, 255, 255, 0),
|
||||
font_size=20,
|
||||
)
|
||||
bk.paste(nickname_img, (30, 15), True)
|
||||
bk.paste(uid_img, (30, 85), True)
|
||||
bk.paste(A, (0, 150), alpha=True)
|
||||
bk.text((30, 167), "Accumulative check-in for")
|
||||
_x = bk.getsize("Accumulative check-in for")[0] + sign_day_img.w + 45
|
||||
bk.paste(sign_day_img, (346, 158), True)
|
||||
bk.text((_x, 167), "days")
|
||||
bk.paste(data_img, (220, 370), True)
|
||||
bk.paste(lik_text1_img, (220, 240), True)
|
||||
bk.paste(lik_text2_img, (262, 234), True)
|
||||
bk.paste(bar_bk, (225, 275), True)
|
||||
bk.paste(info_img, (220, 305), True)
|
||||
bk.paste(today_sign_text_img, (550, 180), True)
|
||||
bk.paste(today_data, (580, 220), True)
|
||||
bk.paste(watermark, (15, 400), True)
|
||||
bk.save(
|
||||
SIGN_TODAY_CARD_PATH / f"{user_id}_{user.belonging_group}_{_type}_{data}.png"
|
||||
)
|
||||
return image(
|
||||
f"{user_id}_{user.belonging_group}_{_type}_{data}.png", "sign/today_card"
|
||||
)
|
||||
|
||||
|
||||
def generate_progress_bar_pic():
|
||||
bg_2 = (254, 1, 254)
|
||||
bg_1 = (0, 245, 246)
|
||||
|
||||
bk = CreateImg(1000, 50, is_alpha=True)
|
||||
img_x = CreateImg(50, 50, color=bg_2)
|
||||
img_x.circle()
|
||||
img_x.crop((25, 0, 50, 50))
|
||||
img_y = CreateImg(50, 50, color=bg_1)
|
||||
img_y.circle()
|
||||
img_y.crop((0, 0, 25, 50))
|
||||
A = CreateImg(950, 50)
|
||||
width, height = A.size
|
||||
|
||||
step_r = (bg_2[0] - bg_1[0]) / width
|
||||
step_g = (bg_2[1] - bg_1[1]) / width
|
||||
step_b = (bg_2[2] - bg_1[2]) / width
|
||||
|
||||
for y in range(0, width):
|
||||
bg_r = round(bg_1[0] + step_r * y)
|
||||
bg_g = round(bg_1[1] + step_g * y)
|
||||
bg_b = round(bg_1[2] + step_b * y)
|
||||
for x in range(0, height):
|
||||
A.point((y, x), fill=(bg_r, bg_g, bg_b))
|
||||
bk.paste(img_y, (0, 0), True)
|
||||
bk.paste(A, (25, 0))
|
||||
bk.paste(img_x, (975, 0), True)
|
||||
bk.save(SIGN_RESOURCE_PATH / "bar.png")
|
||||
|
||||
A = CreateImg(950, 50)
|
||||
bk = CreateImg(1000, 50, is_alpha=True)
|
||||
img_x = CreateImg(50, 50)
|
||||
img_x.circle()
|
||||
img_x.crop((25, 0, 50, 50))
|
||||
img_y = CreateImg(50, 50)
|
||||
img_y.circle()
|
||||
img_y.crop((0, 0, 25, 50))
|
||||
bk.paste(img_y, (0, 0), True)
|
||||
bk.paste(A, (25, 0))
|
||||
bk.paste(img_x, (975, 0), True)
|
||||
bk.save(SIGN_RESOURCE_PATH / "bar_white.png")
|
||||
|
||||
|
||||
def get_level_and_next_impression(impression: float):
|
||||
if impression == 0:
|
||||
return lik2level[10], 10, 0
|
||||
keys = list(lik2level.keys())
|
||||
for i in range(len(keys)):
|
||||
if impression > keys[i]:
|
||||
return lik2level[keys[i]], keys[i - 1], keys[i]
|
||||
return lik2level[10], 10, 0
|
||||
|
||||
|
||||
def clear_sign_data_pic():
|
||||
date = datetime.now().date()
|
||||
for file in os.listdir(SIGN_TODAY_CARD_PATH):
|
||||
if str(date) not in file:
|
||||
os.remove(SIGN_TODAY_CARD_PATH / file)
|
||||
|
||||
Reference in New Issue
Block a user