mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
352 lines
13 KiB
Python
Executable File
352 lines
13 KiB
Python
Executable File
import asyncio
|
||
import random
|
||
from datetime import datetime
|
||
from typing import Union
|
||
|
||
import pypinyin
|
||
from nonebot.adapters.onebot.v11 import Message
|
||
from PIL import Image
|
||
|
||
from configs.config import Config
|
||
from configs.path_config import IMAGE_PATH
|
||
from models.sign_group_user import SignGroupUser
|
||
from services.log import logger
|
||
from utils.image_utils import BuildImage, alpha2white_pil
|
||
from utils.message_builder import image
|
||
from utils.utils import cn2py
|
||
|
||
from .config import *
|
||
from .models.buff_prices import BuffPrice
|
||
from .models.open_cases_user import OpenCasesUser
|
||
|
||
RESULT_MESSAGE = {
|
||
"BLUE": ["这样看着才舒服", "是自己人，大伙把刀收好", "非常舒适~"],
|
||
"PURPLE": ["还行吧，勉强接受一下下", "居然不是蓝色，太假了", "运气-1-1-1-1-1..."],
|
||
"PINK": ["开始不适....", "你妈妈买菜必涨价！涨三倍！", "你最近不适合出门，真的"],
|
||
"RED": ["已经非常不适", "好兄弟你开的什么箱子啊，一般箱子不是只有蓝色的吗", "开始拿阳寿开箱子了？"],
|
||
"KNIFE": ["你的好运我收到了，你可以去喂鲨鱼了", "最近该吃啥就迟点啥吧，哎，好好的一个人怎么就....哎", "众所周知，欧皇寿命极短."],
|
||
}
|
||
|
||
COLOR2NAME = {"BLUE": "军规", "PURPLE": "受限", "PINK": "保密", "RED": "隐秘", "KNIFE": "罕见"}
|
||
|
||
COLOR2CN = {"BLUE": "蓝", "PURPLE": "紫", "PINK": "粉", "RED": "红", "KNIFE": "金"}
|
||
|
||
|
||
def add_count(user: OpenCasesUser, skin: BuffSkin):
|
||
if skin.color == "BLUE":
|
||
if skin.is_stattrak:
|
||
user.blue_st_count += 1
|
||
else:
|
||
user.blue_count += 1
|
||
elif skin.color == "PURPLE":
|
||
if skin.is_stattrak:
|
||
user.purple_st_count += 1
|
||
else:
|
||
user.purple_count += 1
|
||
elif skin.color == "PINK":
|
||
if skin.is_stattrak:
|
||
user.pink_st_count += 1
|
||
else:
|
||
user.pink_count += 1
|
||
elif skin.color == "RED":
|
||
if skin.is_stattrak:
|
||
user.red_st_count += 1
|
||
else:
|
||
user.red_count += 1
|
||
elif skin.color == "KNIFE":
|
||
if skin.is_stattrak:
|
||
user.knife_st_count += 1
|
||
else:
|
||
user.knife_count += 1
|
||
user.today_open_total += 1
|
||
user.total_count += 1
|
||
user.make_money += skin.skin_price
|
||
user.spend_money += 17
|
||
|
||
|
||
async def get_user_max_count(user_qq: int, group_id: int) -> int:
|
||
"""获取用户每日最大开箱次数
|
||
|
||
Args:
|
||
user_qq (int): 用户id
|
||
group_id (int): 群号
|
||
|
||
Returns:
|
||
int: 最大开箱次数
|
||
"""
|
||
user, _ = await SignGroupUser.get_or_create(user_qq=user_qq, group_id=group_id)
|
||
impression = int(user.impression)
|
||
initial_open_case_count = Config.get_config("open_cases", "INITIAL_OPEN_CASE_COUNT")
|
||
each_impression_add_count = Config.get_config(
|
||
"open_cases", "EACH_IMPRESSION_ADD_COUNT"
|
||
)
|
||
return int(initial_open_case_count + impression / each_impression_add_count) # type: ignore
|
||
|
||
|
||
async def open_case(
|
||
user_qq: int, group_id: int, case_name: str = "狂牙大行动"
|
||
) -> Union[str, Message]:
|
||
"""开箱
|
||
|
||
Args:
|
||
user_qq (int): 用户id
|
||
group_id (int): 群号
|
||
case_name (str, optional): 武器箱名称. Defaults to "狂牙大行动".
|
||
|
||
Returns:
|
||
Union[str, Message]: 回复消息
|
||
"""
|
||
if case_name not in ["狂牙大行动", "突围大行动", "命悬一线", "裂空", "光谱"]:
|
||
return "武器箱未收录"
|
||
logger.debug(f"尝试开启武器箱: {case_name}", "开箱", user_qq, group_id)
|
||
case = cn2py(case_name)
|
||
user = await OpenCasesUser.get_or_none(user_qq=user_qq, group_id=group_id)
|
||
if not user:
|
||
user = await OpenCasesUser.create(
|
||
user_qq=user_qq, group_id=group_id, open_cases_time_last=datetime.now()
|
||
)
|
||
max_count = await get_user_max_count(user_qq, group_id)
|
||
# 一天次数上限
|
||
if user.today_open_total >= max_count:
|
||
return _handle_is_MAX_COUNT()
|
||
skin_list = await random_skin(1, case_name)
|
||
if not skin_list:
|
||
return "未抽取到任何皮肤..."
|
||
skin, rand = skin_list[0]
|
||
rand = str(rand)[:11]
|
||
add_count(user, skin)
|
||
ridicule_result = random.choice(RESULT_MESSAGE[skin.color])
|
||
price_result = skin.skin_price
|
||
if skin.color == "KNIFE":
|
||
user.knifes_name = (
|
||
user.knifes_name
|
||
+ f"{case}||{skin.name}{'（StatTrak™）' if skin.is_stattrak else ''} | {skin.skin_name} ({skin.abrasion}) 磨损：{rand}， 价格：{skin.skin_price},"
|
||
)
|
||
img_path = (
|
||
IMAGE_PATH
|
||
/ "cases"
|
||
/ case
|
||
/ f"{cn2py(skin.name)} - {cn2py(skin.skin_name)}.png"
|
||
)
|
||
logger.info(
|
||
f"开启{case_name}武器箱获得 {skin.name}{'（StatTrak™）' if skin.is_stattrak else ''} | {skin.skin_name} ({skin.abrasion}) 磨损: [{rand}] 价格: {skin.skin_price}",
|
||
"开箱",
|
||
user_qq,
|
||
group_id,
|
||
)
|
||
await user.save()
|
||
over_count = max_count - user.today_open_total
|
||
return (
|
||
f"开启{case_name}武器箱.\n剩余开箱次数:{over_count}.\n"
|
||
+ image(img_path)
|
||
+ "\n"
|
||
+ f"皮肤:[{COLOR2NAME[skin.color]}]{skin.name}{'（StatTrak™）' if skin.is_stattrak else ''} | {skin.skin_name} ({skin.abrasion})\n"
|
||
f"磨损:{rand}\n"
|
||
f"价格:{price_result}\n"
|
||
f":{ridicule_result}"
|
||
)
|
||
|
||
|
||
async def open_multiple_case(
|
||
user_qq: int, group_id: int, case_name: str, num: int = 10
|
||
):
|
||
user, _ = await OpenCasesUser.get_or_create(user_qq=user_qq, group_id=group_id)
|
||
max_count = await get_user_max_count(user_qq, group_id)
|
||
if user.today_open_total >= max_count:
|
||
return _handle_is_MAX_COUNT()
|
||
if max_count - user.today_open_total < num:
|
||
return (
|
||
f"今天开箱次数不足{num}次噢，请单抽试试看（也许单抽运气更好？）"
|
||
f"\n剩余开箱次数:{max_count - user.today_open_total}"
|
||
)
|
||
if num < 5:
|
||
h = 270
|
||
elif num % 5 == 0:
|
||
h = 270 * int(num / 5)
|
||
else:
|
||
h = 270 * int(num / 5) + 270
|
||
case = cn2py(case_name)
|
||
skin_count = {}
|
||
img_list = []
|
||
skin_list = await random_skin(num, case_name)
|
||
if not skin_list:
|
||
return "未抽取到任何皮肤..."
|
||
total_price = 0
|
||
for skin, rand in skin_list:
|
||
total_price += skin.skin_price
|
||
rand = str(rand)[:11]
|
||
add_count(user, skin)
|
||
color_name = COLOR2CN[skin.color]
|
||
if skin.is_stattrak:
|
||
color_name += "(暗金)"
|
||
if not skin_count.get(color_name):
|
||
skin_count[color_name] = 0
|
||
skin_count[color_name] += 1
|
||
if skin.color == "KNIFE":
|
||
user.knifes_name = (
|
||
user.knifes_name
|
||
+ f"{case}||{skin.name}{'（StatTrak™）' if skin.is_stattrak else ''} | {skin.skin_name} ({skin.abrasion}) 磨损：{rand}， 价格：{skin.skin_price},"
|
||
)
|
||
wImg = BuildImage(200, 270, 200, 200)
|
||
await wImg.apaste(
|
||
alpha2white_pil(
|
||
Image.open(
|
||
IMAGE_PATH
|
||
/ "cases"
|
||
/ case
|
||
/ f"{cn2py(skin.name)} - {cn2py(skin.skin_name)}.png"
|
||
).resize((200, 200), Image.ANTIALIAS)
|
||
),
|
||
(0, 0),
|
||
)
|
||
await wImg.atext(
|
||
(5, 200),
|
||
f"{skin.name}{'（StatTrak™）' if skin.is_stattrak else ''} | {skin.skin_name} ({skin.abrasion})",
|
||
)
|
||
await wImg.atext((5, 220), f"磨损：{rand}")
|
||
await wImg.atext((5, 240), f"价格：{skin.skin_price}")
|
||
img_list.append(wImg)
|
||
logger.info(
|
||
f"开启{case_name}武器箱获得 {skin.name}{'（StatTrak™）' if skin.is_stattrak else ''} | {skin.skin_name} ({skin.abrasion}) 磨损: [{rand}] 价格: {skin.skin_price}",
|
||
"开箱",
|
||
user_qq,
|
||
group_id,
|
||
)
|
||
await user.save()
|
||
markImg = BuildImage(1000, h, 200, 270)
|
||
for img in img_list:
|
||
markImg.paste(img)
|
||
over_count = max_count - user.today_open_total
|
||
result = ""
|
||
for color_name in skin_count:
|
||
result += f"[{color_name}:{skin_count[color_name]}] "
|
||
return (
|
||
f"开启{case_name}武器箱\n剩余开箱次数：{over_count}\n"
|
||
+ image(markImg.pic2bs4())
|
||
+ "\n"
|
||
+ result[:-1]
|
||
+ f"\n总获取金额：{total_price:.2f}\n总花费：{17 * num}"
|
||
)
|
||
|
||
|
||
def _handle_is_MAX_COUNT() -> str:
|
||
return f"今天已达开箱上限了喔，明天再来吧\n(提升好感度可以增加每日开箱数 #疯狂暗示)"
|
||
|
||
|
||
async def total_open_statistics(user_qq: int, group: int) -> str:
|
||
user, _ = await OpenCasesUser.get_or_create(user_qq=user_qq, group_id=group)
|
||
return (
|
||
f"开箱总数：{user.total_count}\n"
|
||
f"今日开箱：{user.today_open_total}\n"
|
||
f"蓝色军规：{user.blue_count}\n"
|
||
f"蓝色暗金：{user.blue_st_count}\n"
|
||
f"紫色受限：{user.purple_count}\n"
|
||
f"紫色暗金：{user.purple_st_count}\n"
|
||
f"粉色保密：{user.pink_count}\n"
|
||
f"粉色暗金：{user.pink_st_count}\n"
|
||
f"红色隐秘：{user.red_count}\n"
|
||
f"红色暗金：{user.red_st_count}\n"
|
||
f"金色罕见：{user.knife_count}\n"
|
||
f"金色暗金：{user.knife_st_count}\n"
|
||
f"花费金额：{user.spend_money}\n"
|
||
f"获取金额：{user.make_money:.2f}\n"
|
||
f"最后开箱日期：{user.open_cases_time_last.date()}"
|
||
)
|
||
|
||
|
||
async def group_statistics(group: int):
|
||
user_list = await OpenCasesUser.filter(group_id=group).all()
|
||
# lan zi fen hong jin pricei
|
||
uplist = [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0.0, 0, 0]
|
||
for user in user_list:
|
||
uplist[0] += user.blue_count
|
||
uplist[1] += user.blue_st_count
|
||
uplist[2] += user.purple_count
|
||
uplist[3] += user.purple_st_count
|
||
uplist[4] += user.pink_count
|
||
uplist[5] += user.pink_st_count
|
||
uplist[6] += user.red_count
|
||
uplist[7] += user.red_st_count
|
||
uplist[8] += user.knife_count
|
||
uplist[9] += user.knife_st_count
|
||
uplist[10] += user.make_money
|
||
uplist[11] += user.total_count
|
||
uplist[12] += user.today_open_total
|
||
return (
|
||
f"群开箱总数：{uplist[11]}\n"
|
||
f"群今日开箱：{uplist[12]}\n"
|
||
f"蓝色军规：{uplist[0]}\n"
|
||
f"蓝色暗金：{uplist[1]}\n"
|
||
f"紫色受限：{uplist[2]}\n"
|
||
f"紫色暗金：{uplist[3]}\n"
|
||
f"粉色保密：{uplist[4]}\n"
|
||
f"粉色暗金：{uplist[5]}\n"
|
||
f"红色隐秘：{uplist[6]}\n"
|
||
f"红色暗金：{uplist[7]}\n"
|
||
f"金色罕见：{uplist[8]}\n"
|
||
f"金色暗金：{uplist[9]}\n"
|
||
f"花费金额：{uplist[11] * 17}\n"
|
||
f"获取金额：{uplist[10]:.2f}"
|
||
)
|
||
|
||
|
||
async def my_knifes_name(user_id: int, group: int):
|
||
user, _ = await OpenCasesUser.get_or_create(user_qq=user_id, group_id=group)
|
||
knifes_name = user.knifes_name
|
||
if knifes_name:
|
||
knifes_list = knifes_name[:-1].split(",")
|
||
length = len(knifes_list)
|
||
if length < 5:
|
||
h = 600
|
||
w = length * 540
|
||
elif length % 5 == 0:
|
||
h = 600 * int(length / 5)
|
||
w = 540 * 5
|
||
else:
|
||
h = 600 * int(length / 5) + 600
|
||
w = 540 * 5
|
||
A = await asyncio.get_event_loop().run_in_executor(
|
||
None, _pst_my_knife, w, h, knifes_list
|
||
)
|
||
return image(b64=A.pic2bs4())
|
||
else:
|
||
return "您木有开出金色级别的皮肤喔"
|
||
|
||
|
||
def _pst_my_knife(w, h, knifes_list):
|
||
A = BuildImage(w, h, 540, 600)
|
||
for knife in knifes_list:
|
||
case = knife.split("||")[0]
|
||
knife = knife.split("||")[1]
|
||
name = knife[: knife.find("(")].strip()
|
||
itype = knife[knife.find("(") + 1 : knife.find(")")].strip()
|
||
mosun = knife[knife.find("磨损：") + 3 : knife.rfind("价格：")].strip()
|
||
if mosun[-1] == "," or mosun[-1] == "，":
|
||
mosun = mosun[:-1]
|
||
price = knife[knife.find("价格：") + 3 :]
|
||
skin_name = ""
|
||
for i in pypinyin.pinyin(
|
||
name.replace("|", "-").replace("（StatTrak™）", "").strip(),
|
||
style=pypinyin.NORMAL,
|
||
):
|
||
skin_name += "".join(i)
|
||
knife_img = BuildImage(470, 600, 470, 470, font_size=20)
|
||
knife_img.paste(
|
||
alpha2white_pil(
|
||
Image.open(IMAGE_PATH / f"cases" / case / f"{skin_name}.png").resize(
|
||
(470, 470), Image.ANTIALIAS
|
||
)
|
||
),
|
||
(0, 0),
|
||
)
|
||
knife_img.text((5, 500), f"\t{name}({itype})")
|
||
knife_img.text((5, 530), f"\t磨损：{mosun}")
|
||
knife_img.text((5, 560), f"\t价格：{price}")
|
||
A.paste(knife_img)
|
||
return A
|
||
|
||
|
||
# G3SG1（StatTrak™） | 血腥迷彩 (战痕累累)
|
||
# G3SG1（StatTrak™） | 血腥迷彩 (战痕累累)
|
||
# G3SG1（StatTrak™） | 血腥迷彩 (战痕累累)
|