mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
109 lines
3.8 KiB
Python
Executable File
109 lines
3.8 KiB
Python
Executable File
import asyncio
|
|
import os
|
|
import random
|
|
from io import BytesIO
|
|
from typing import Union
|
|
|
|
from configs.path_config import IMAGE_PATH
|
|
from models.bag_user import BagUser
|
|
from utils.image_utils import BuildImage
|
|
from utils.utils import get_user_avatar, is_number
|
|
|
|
from .model import RedbagUser
|
|
|
|
|
|
# 检查金币数量合法性，并添加记录数据
|
|
async def check_gold(user_id: str, group_id: str, amount: Union[str, int]):
|
|
if is_number(amount):
|
|
amount = int(amount)
|
|
user_gold = await BagUser.get_gold(user_id, group_id)
|
|
if amount < 1:
|
|
return False, "小气鬼，要别人倒贴金币给你嘛！"
|
|
if user_gold < amount:
|
|
return False, "没有金币的话请不要发红包..."
|
|
await BagUser.spend_gold(user_id, group_id, amount)
|
|
await RedbagUser.add_redbag_data(user_id, group_id, "send", amount)
|
|
return True, amount
|
|
else:
|
|
return False, "给我好好的输入红包里金币的数量啊喂！"
|
|
|
|
|
|
# 金币退回
|
|
async def return_gold(user_id: str, group_id: str, amount: int):
|
|
await BagUser.add_gold(user_id, group_id, amount)
|
|
|
|
|
|
# 开红包
|
|
async def open_redbag(user_id: str, group_id: str, redbag_data: dict):
|
|
amount = random.choice(redbag_data[group_id]["redbag"])
|
|
redbag_data[group_id]["redbag"].remove(amount)
|
|
redbag_data[group_id]["open_user"].append(user_id)
|
|
redbag_data[group_id]["open_amount"] += amount
|
|
await RedbagUser.add_redbag_data(user_id, group_id, "get", amount)
|
|
await BagUser.add_gold(user_id, group_id, amount)
|
|
return amount, redbag_data
|
|
|
|
|
|
# 随机红包图片
|
|
async def generate_send_redbag_pic(user_id: int, msg: str = "恭喜发财 大吉大利"):
|
|
random_redbag = random.choice(os.listdir(f"{IMAGE_PATH}/prts/redbag_2"))
|
|
redbag = BuildImage(
|
|
0, 0, font_size=38, background=f"{IMAGE_PATH}/prts/redbag_2/{random_redbag}"
|
|
)
|
|
ava = BuildImage(65, 65, background=BytesIO(await get_user_avatar(user_id)))
|
|
await asyncio.get_event_loop().run_in_executor(None, ava.circle)
|
|
redbag.text(
|
|
(int((redbag.size[0] - redbag.getsize(msg)[0]) / 2), 210), msg, (240, 218, 164)
|
|
)
|
|
redbag.paste(ava, (int((redbag.size[0] - ava.size[0]) / 2), 130), True)
|
|
return redbag.pic2bs4()
|
|
|
|
|
|
# 开红包图片
|
|
async def generate_open_redbag_pic(
|
|
user_id: str, send_user_nickname: str, amount: int, text: str
|
|
):
|
|
return await asyncio.create_task(
|
|
_generate_open_redbag_pic(user_id, send_user_nickname, amount, text)
|
|
)
|
|
|
|
|
|
# 开红包图片
|
|
async def _generate_open_redbag_pic(
|
|
user_id: str, send_user_nickname: str, amount: int, text: str
|
|
):
|
|
send_user_nickname += "的红包"
|
|
amount_str = str(amount)
|
|
random_redbag = random.choice(os.listdir(f"{IMAGE_PATH}/prts/redbag_1"))
|
|
head = BuildImage(
|
|
1000,
|
|
980,
|
|
font_size=30,
|
|
background=f"{IMAGE_PATH}/prts/redbag_1/{random_redbag}",
|
|
)
|
|
size = BuildImage(0, 0, font_size=50).getsize(send_user_nickname)
|
|
# QQ头像
|
|
ava_bk = BuildImage(100 + size[0], 66, is_alpha=True, font_size=50)
|
|
ava = BuildImage(
|
|
66, 66, is_alpha=True, background=BytesIO(await get_user_avatar(user_id))
|
|
)
|
|
ava_bk.paste(ava)
|
|
ava_bk.text((100, 7), send_user_nickname)
|
|
# ava_bk.show()
|
|
ava_bk_w, ava_bk_h = ava_bk.size
|
|
head.paste(ava_bk, (int((1000 - ava_bk_w) / 2), 300), alpha=True)
|
|
# 金额
|
|
size = BuildImage(0, 0, font_size=150).getsize(amount_str)
|
|
price = BuildImage(size[0], size[1], is_alpha=True, font_size=150)
|
|
price.text((0, 0), amount_str, fill=(209, 171, 108))
|
|
# 金币中文
|
|
head.paste(price, (int((1000 - size[0]) / 2) - 50, 460), alpha=True)
|
|
head.text(
|
|
(int((1000 - size[0]) / 2 + size[0]) - 50, 500 + size[1] - 70),
|
|
"金币",
|
|
fill=(209, 171, 108),
|
|
)
|
|
# 剩余数量和金额
|
|
head.text((350, 900), text, (198, 198, 198))
|
|
return head.pic2bs4()
|