update 0.0.6.2

plugins/gold_redbag/data_source.py

@@ -0,0 +1,87 @@
+from models.bag_user import BagUser
+from utils.utils import is_number, get_local_proxy, get_user_avatar
+from utils.image_utils import CreateImg
+from utils.user_agent import get_user_agent
+from configs.path_config import IMAGE_PATH
+from .model import RedbagUser
+import random
+import os
+from io import BytesIO
+import asyncio
+
+
+# 检查金币数量合法性，并添加记录数据
+async def check_gold(user_id: int, group_id: int, amount: str):
+    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: int, group_id: int, amount: int):
+    await BagUser.add_gold(user_id, group_id, amount)
+
+
+# 开红包
+async def open_redbag(user_id: int, group_id: int, 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 = CreateImg(0, 0, font_size=38, background=f'{IMAGE_PATH}/prts/redbag_2/{random_redbag}')
+    ava = CreateImg(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: int, 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: int, send_user_nickname: str, amount: int, text: str):
+    send_user_nickname += '的红包'
+    amount = str(amount)
+    head = CreateImg(1000, 980, font_size=30, background=f'{IMAGE_PATH}/prts/redbag_12.png')
+    size = CreateImg(0, 0, font_size=50).getsize(send_user_nickname)
+    # QQ头像
+    ava_bk = CreateImg(100 + size[0], 66, color='white', font_size=50)
+    ava = CreateImg(66, 66, 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))
+    # 金额
+    size = CreateImg(0, 0, font_size=150).getsize(amount)
+    price = CreateImg(size[0], size[1], font_size=150)
+    price.text((0, 0), amount, fill=(209, 171, 108))
+    # 金币中文
+    head.paste(price, (int((1000 - size[0]) / 2) - 50, 460))
+    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()
+
+
+