mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
536 lines
20 KiB
Python
536 lines
20 KiB
Python
import random
|
||
import time
|
||
from datetime import datetime, timedelta
|
||
|
||
from apscheduler.jobstores.base import JobLookupError
|
||
from nonebot.adapters import Bot
|
||
from nonebot_plugin_alconna import At, UniMessage
|
||
from nonebot_plugin_apscheduler import scheduler
|
||
from pydantic import BaseModel
|
||
|
||
from zhenxun.configs.config import BotConfig, Config
|
||
from zhenxun.models.group_member_info import GroupInfoUser
|
||
from zhenxun.models.user_console import UserConsole
|
||
from zhenxun.utils.enum import GoldHandle
|
||
from zhenxun.utils.exception import InsufficientGold
|
||
from zhenxun.utils.image_utils import BuildImage, BuildMat, MatType, text2image
|
||
from zhenxun.utils.message import MessageUtils
|
||
from zhenxun.utils.platform import PlatformUtils
|
||
|
||
from .model import RussianUser
|
||
|
||
base_config = Config.get("russian")
|
||
|
||
|
||
class Russian(BaseModel):
|
||
|
||
at_user: str | None
|
||
"""指定决斗对象"""
|
||
player1: tuple[str, str]
|
||
"""玩家1id, 昵称"""
|
||
player2: tuple[str, str] | None = None
|
||
"""玩家2id, 昵称"""
|
||
money: int
|
||
"""金额"""
|
||
bullet_num: int
|
||
"""子弹数"""
|
||
bullet_arr: list[int] = []
|
||
"""子弹排列"""
|
||
bullet_index: int = 0
|
||
"""当前子弹下标"""
|
||
next_user: str = ""
|
||
"""下一个开枪用户"""
|
||
time: float = time.time()
|
||
"""创建时间"""
|
||
win_user: str | None = None
|
||
"""胜利者"""
|
||
|
||
|
||
class RussianManage:
|
||
|
||
def __init__(self) -> None:
|
||
self._data: dict[str, Russian] = {}
|
||
|
||
def __check_is_timeout(self, group_id: str) -> bool:
|
||
"""检查决斗是否超时
|
||
|
||
参数:
|
||
group_id: 群组id
|
||
|
||
返回:
|
||
bool: 是否超时
|
||
"""
|
||
if russian := self._data.get(group_id):
|
||
if russian.time + 30 < time.time():
|
||
return True
|
||
return False
|
||
|
||
def __random_bullet(self, num: int) -> list[int]:
|
||
"""随机排列子弹
|
||
|
||
参数:
|
||
num: 子弹数量
|
||
|
||
返回:
|
||
list[int]: 子弹排列数组
|
||
"""
|
||
bullet_list = [0, 0, 0, 0, 0, 0, 0]
|
||
for i in random.sample([0, 1, 2, 3, 4, 5, 6], num):
|
||
bullet_list[i] = 1
|
||
return bullet_list
|
||
|
||
def __remove_job(self, group_id: str):
|
||
"""移除定时任务
|
||
|
||
参数:
|
||
group_id: 群组id
|
||
"""
|
||
try:
|
||
scheduler.remove_job(f"russian_job_{group_id}")
|
||
except JobLookupError:
|
||
pass
|
||
|
||
def __build_job(
|
||
self, bot: Bot, group_id: str, is_add: bool = False, platform: str | None = None
|
||
):
|
||
"""移除定时任务和构建新定时任务
|
||
|
||
参数:
|
||
bot: Bot
|
||
group_id: 群组id
|
||
is_add: 是否添加新定时任务.
|
||
platform: 平台
|
||
"""
|
||
self.__remove_job(group_id)
|
||
if is_add:
|
||
date = datetime.now() + timedelta(seconds=31)
|
||
scheduler.add_job(
|
||
self.__auto_end_game,
|
||
"date",
|
||
run_date=date.replace(microsecond=0),
|
||
id=f"russian_job_{group_id}",
|
||
args=[bot, group_id, platform],
|
||
)
|
||
|
||
async def __auto_end_game(self, bot: Bot, group_id: str, platform: str):
|
||
"""自动结束对决
|
||
|
||
参数:
|
||
bot: Bot
|
||
group_id: 群组id
|
||
platform: 平台
|
||
"""
|
||
result = await self.settlement(group_id, None, platform)
|
||
if result:
|
||
await PlatformUtils.send_message(bot, None, group_id, result)
|
||
|
||
async def add_russian(self, bot: Bot, group_id: str, rus: Russian) -> UniMessage:
|
||
"""添加决斗
|
||
|
||
参数:
|
||
bot: Bot
|
||
group_id: 群组id
|
||
rus: Russian
|
||
|
||
返回:
|
||
UniMessage: 返回消息
|
||
"""
|
||
russian = self._data.get(group_id)
|
||
if russian:
|
||
if russian.time + 30 < time.time():
|
||
if not russian.player2:
|
||
return MessageUtils.build_message(
|
||
f"现在是 {russian.player1[1]} 发起的对决, 请接受对决或等待决斗超时..."
|
||
)
|
||
else:
|
||
return MessageUtils.build_message(
|
||
f"{russian.player1[1]} 和 {russian.player2[1]}的对决还未结束！"
|
||
)
|
||
return MessageUtils.build_message(
|
||
f"现在是 {russian.player1[1]} 发起的对决\n请等待比赛结束后再开始下一轮..."
|
||
)
|
||
max_money = base_config.get("MAX_RUSSIAN_BET_GOLD")
|
||
if rus.money > max_money:
|
||
return MessageUtils.build_message(f"太多了！单次金额不能超过{max_money}！")
|
||
user = await UserConsole.get_user(rus.player1[0])
|
||
if user.gold < rus.money:
|
||
return MessageUtils.build_message("你没有足够的钱支撑起这场挑战")
|
||
rus.bullet_arr = self.__random_bullet(rus.bullet_num)
|
||
self._data[group_id] = rus
|
||
message_list: list[str | At] = []
|
||
if rus.at_user:
|
||
user = await GroupInfoUser.get_or_none(
|
||
user_id=rus.at_user, group_id=group_id
|
||
)
|
||
message_list = [
|
||
f"{rus.player1[1]} 向",
|
||
At(flag="user", target=rus.at_user),
|
||
f"发起了决斗！请 {user.user_name if user else rus.at_user} 在30秒内回复‘接受对决’ or ‘拒绝对决’，超时此次决斗作废！",
|
||
]
|
||
else:
|
||
message_list = [
|
||
"若30秒内无人接受挑战则此次对决作废【首次游玩请at我发送 ’帮助俄罗斯轮盘‘ 来查看命令】"
|
||
]
|
||
result = (
|
||
"咔 " * rus.bullet_num
|
||
+ f"装填完毕\n挑战金额：{rus.money}\n第一枪的概率为：{float(rus.bullet_num) / 7.0 * 100:.2f}%\n"
|
||
)
|
||
|
||
message_list.insert(0, result)
|
||
self.__build_job(bot, group_id, True)
|
||
return MessageUtils.build_message(message_list) # type: ignore
|
||
|
||
async def accept(
|
||
self, bot: Bot, group_id: str, user_id: str, uname: str
|
||
) -> UniMessage:
|
||
"""接受对决
|
||
|
||
参数:
|
||
bot: Bot
|
||
group_id: 群组id
|
||
user_id: 用户id
|
||
uname: 用户名称
|
||
|
||
返回:
|
||
Text | MessageFactory: 返回消息
|
||
"""
|
||
if russian := self._data.get(group_id):
|
||
if russian.at_user and russian.at_user != user_id:
|
||
return MessageUtils.build_message("又不是找你决斗，你接受什么啊！气！")
|
||
if russian.player2:
|
||
return MessageUtils.build_message(
|
||
"当前决斗已被其他玩家接受！请等待下局对决！"
|
||
)
|
||
if russian.player1[0] == user_id:
|
||
return MessageUtils.build_message("你发起的对决，你接受什么啊！气！")
|
||
user = await UserConsole.get_user(user_id)
|
||
if user.gold < russian.money:
|
||
return MessageUtils.build_message("你没有足够的钱来接受这场挑战...")
|
||
russian.player2 = (user_id, uname)
|
||
russian.next_user = russian.player1[0]
|
||
self.__build_job(bot, group_id, True)
|
||
return MessageUtils.build_message(
|
||
[
|
||
"决斗已经开始！请",
|
||
At(flag="user", target=russian.player1[0]),
|
||
"先开枪！",
|
||
]
|
||
)
|
||
return MessageUtils.build_message(
|
||
"目前没有进行的决斗，请发送 装弹 开启决斗吧！"
|
||
)
|
||
|
||
def refuse(self, group_id: str, user_id: str, uname: str) -> UniMessage:
|
||
"""拒绝决斗
|
||
|
||
参数:
|
||
group_id: 群组id
|
||
user_id: 用户id
|
||
uname: 用户名称
|
||
|
||
返回:
|
||
Text | MessageFactory: 返回消息
|
||
"""
|
||
if russian := self._data.get(group_id):
|
||
if russian.at_user:
|
||
if russian.at_user != user_id:
|
||
return MessageUtils.build_message(
|
||
"又不是找你决斗，你拒绝什么啊！气！"
|
||
)
|
||
del self._data[group_id]
|
||
self.__remove_job(group_id)
|
||
return MessageUtils.build_message(
|
||
[
|
||
At(flag="user", target=russian.player1[0]),
|
||
f"{uname}拒绝了你的对决！",
|
||
]
|
||
)
|
||
return MessageUtils.build_message("当前决斗并没有指定对手，无法拒绝哦！")
|
||
return MessageUtils.build_message(
|
||
"目前没有进行的决斗，请发送 装弹 开启决斗吧！"
|
||
)
|
||
|
||
async def shoot(
|
||
self, bot: Bot, group_id: str, user_id: str, uname: str, platform: str
|
||
) -> tuple[UniMessage, UniMessage | None]:
|
||
"""开枪
|
||
|
||
参数:
|
||
bot: Bot
|
||
group_id: 群组id
|
||
user_id: 用户id
|
||
uname: 用户名称
|
||
platform: 平台
|
||
|
||
返回:
|
||
Text | MessageFactory: 返回消息
|
||
"""
|
||
if russian := self._data.get(group_id):
|
||
if not russian.player2:
|
||
return (
|
||
MessageUtils.build_message("当前还没有玩家接受对决，无法开枪..."),
|
||
None,
|
||
)
|
||
if user_id not in [russian.player1[0], russian.player2[0]]:
|
||
"""非玩家1和玩家2发送开枪"""
|
||
return (
|
||
MessageUtils.build_message(
|
||
random.choice(
|
||
[
|
||
f"不要打扰 {russian.player1[1]} 和 {russian.player2[1]} 的决斗啊！",
|
||
f"给我好好做好一个观众！不然{BotConfig.self_nickname}就要生气了",
|
||
f"不要捣乱啊baka{uname}！",
|
||
]
|
||
)
|
||
),
|
||
None,
|
||
)
|
||
if user_id != russian.next_user:
|
||
"""相同玩家连续开枪"""
|
||
return (
|
||
MessageUtils.build_message(
|
||
f"你的左轮不是连发的！该 {russian.player2[1]} 开枪了!"
|
||
),
|
||
None,
|
||
)
|
||
if russian.bullet_arr[russian.bullet_index] == 1:
|
||
"""去世"""
|
||
result = MessageUtils.build_message(
|
||
random.choice(
|
||
[
|
||
'"嘭！"，你直接去世了',
|
||
"眼前一黑，你直接穿越到了异世界...(死亡)",
|
||
"终究还是你先走一步...",
|
||
]
|
||
)
|
||
)
|
||
settle = await self.settlement(group_id, user_id, platform)
|
||
return result, settle
|
||
else:
|
||
"""存活"""
|
||
p = (
|
||
(russian.bullet_index + russian.bullet_num + 1)
|
||
/ len(russian.bullet_arr)
|
||
* 100
|
||
)
|
||
result = (
|
||
random.choice(
|
||
[
|
||
"呼呼，没有爆裂的声响，你活了下来",
|
||
"虽然黑洞洞的枪口很恐怖，但好在没有子弹射出来，你活下来了",
|
||
'"咔"，你没死，看来运气不错',
|
||
]
|
||
)
|
||
+ f"\n下一枪中弹的概率: {p:.2f}%, 轮到 "
|
||
)
|
||
next_user = (
|
||
russian.player2[0]
|
||
if russian.next_user == russian.player1[0]
|
||
else russian.player1[0]
|
||
)
|
||
russian.next_user = next_user
|
||
russian.bullet_index += 1
|
||
self.__build_job(bot, group_id, True)
|
||
return (
|
||
MessageUtils.build_message(
|
||
[result, At(flag="user", target=next_user), " 了!"]
|
||
),
|
||
None,
|
||
)
|
||
return (
|
||
MessageUtils.build_message("目前没有进行的决斗，请发送 装弹 开启决斗吧！"),
|
||
None,
|
||
)
|
||
|
||
async def settlement(
|
||
self, group_id: str, user_id: str | None, platform: str | None = None
|
||
) -> UniMessage:
|
||
"""结算
|
||
|
||
参数:
|
||
group_id: 群组id
|
||
user_id: 用户id
|
||
platform: 平台
|
||
|
||
返回:
|
||
Text | MessageFactory: 返回消息
|
||
"""
|
||
if russian := self._data.get(group_id):
|
||
if not russian.player2:
|
||
if self.__check_is_timeout(group_id):
|
||
del self._data[group_id]
|
||
return MessageUtils.build_message(
|
||
"规定时间内还未有人接受决斗，当前决斗过期..."
|
||
)
|
||
return MessageUtils.build_message("决斗还未开始,，无法结算哦...")
|
||
if user_id and user_id not in [russian.player1[0], russian.player2[0]]:
|
||
return MessageUtils.build_message(f"吃瓜群众不要捣乱！黄牌警告！")
|
||
if not self.__check_is_timeout(group_id):
|
||
return MessageUtils.build_message(
|
||
f"{russian.player1[1]} 和 {russian.player2[1]} 比赛并未超时，请继续比赛..."
|
||
)
|
||
win_user = None
|
||
lose_user = None
|
||
if win_user:
|
||
russian.next_user = (
|
||
russian.player1[0]
|
||
if win_user == russian.player2[0]
|
||
else russian.player2[0]
|
||
)
|
||
if russian.next_user != russian.player1[0]:
|
||
win_user = russian.player1
|
||
lose_user = russian.player2
|
||
else:
|
||
win_user = russian.player2
|
||
lose_user = russian.player1
|
||
if win_user and lose_user:
|
||
rand = 0
|
||
if russian.money > 10:
|
||
rand = random.randint(0, 5)
|
||
fee = int(russian.money * float(rand) / 100)
|
||
fee = 1 if fee < 1 and rand != 0 else fee
|
||
else:
|
||
fee = 0
|
||
winner = await RussianUser.add_count(win_user[0], group_id, "win")
|
||
loser = await RussianUser.add_count(lose_user[0], group_id, "lose")
|
||
await RussianUser.money(
|
||
win_user[0], group_id, "win", russian.money - fee
|
||
)
|
||
await RussianUser.money(lose_user[0], group_id, "lose", russian.money)
|
||
await UserConsole.add_gold(
|
||
win_user[0], russian.money - fee, "russian", platform
|
||
)
|
||
try:
|
||
await UserConsole.reduce_gold(
|
||
lose_user[0],
|
||
russian.money,
|
||
GoldHandle.PLUGIN,
|
||
"russian",
|
||
platform,
|
||
)
|
||
except InsufficientGold:
|
||
if u := await UserConsole.get_user(lose_user[0]):
|
||
u.gold = 0
|
||
await u.save(update_fields=["gold"])
|
||
result = [
|
||
"这场决斗是 ",
|
||
At(flag="user", target=win_user[0]),
|
||
" 胜利了!",
|
||
]
|
||
image = await text2image(
|
||
f"结算：\n"
|
||
f"\t胜者：{win_user[1]}\n"
|
||
f"\t赢取金币：{russian.money - fee}\n"
|
||
f"\t累计胜场：{winner.win_count}\n"
|
||
f"\t累计赚取金币：{winner.make_money}\n"
|
||
f"-------------------\n"
|
||
f"\t败者：{lose_user[1]}\n"
|
||
f"\t输掉金币：{russian.money}\n"
|
||
f"\t累计败场：{loser.fail_count}\n"
|
||
f"\t累计输掉金币：{loser.lose_money}\n"
|
||
f"-------------------\n"
|
||
f"哼哼，{BotConfig.self_nickname}从中收取了 {float(rand)}%({fee}金币) 作为手续费！\n"
|
||
f"子弹排列：{russian.bullet_arr}",
|
||
padding=10,
|
||
color="#f9f6f2",
|
||
)
|
||
self.__remove_job(group_id)
|
||
result.append(image)
|
||
del self._data[group_id]
|
||
return MessageUtils.build_message(result)
|
||
return MessageUtils.build_message("赢家和输家获取错误...")
|
||
return MessageUtils.build_message("比赛并没有开始...无法结算...")
|
||
|
||
async def __get_x_index(self, users: list[RussianUser], group_id: str):
|
||
uid_list = [u.user_id for u in users]
|
||
group_user_list = await GroupInfoUser.filter(
|
||
user_id__in=uid_list, group_id=group_id
|
||
).all()
|
||
group_user = {gu.user_id: gu.user_name for gu in group_user_list}
|
||
data = []
|
||
for uid in uid_list:
|
||
if uid in group_user:
|
||
data.append(group_user[uid])
|
||
else:
|
||
data.append(uid)
|
||
return data
|
||
|
||
async def rank(
|
||
self, user_id: str, group_id: str, rank_type: str, num: int
|
||
) -> BuildImage | str:
|
||
x_index = []
|
||
data = []
|
||
title = ""
|
||
x_name = ""
|
||
if rank_type == "win":
|
||
users = (
|
||
await RussianUser.filter(group_id=group_id, win_count__not=0)
|
||
.order_by("win_count")
|
||
.limit(num)
|
||
)
|
||
x_index = await self.__get_x_index(users, group_id)
|
||
data = [u.win_count for u in users]
|
||
title = "胜场排行"
|
||
x_name = "场次"
|
||
if rank_type == "lose":
|
||
users = (
|
||
await RussianUser.filter(group_id=group_id, fail_count__not=0)
|
||
.order_by("fail_count")
|
||
.limit(num)
|
||
)
|
||
x_index = await self.__get_x_index(users, group_id)
|
||
data = [u.fail_count for u in users]
|
||
title = "败场排行"
|
||
x_name = "场次"
|
||
if rank_type == "a":
|
||
users = (
|
||
await RussianUser.filter(group_id=group_id, make_money__not=0)
|
||
.order_by("make_money")
|
||
.limit(num)
|
||
)
|
||
x_index = await self.__get_x_index(users, group_id)
|
||
data = [u.make_money for u in users]
|
||
title = "欧洲人排行"
|
||
x_name = "金币"
|
||
if rank_type == "b":
|
||
users = (
|
||
await RussianUser.filter(group_id=group_id, lose_money__not=0)
|
||
.order_by("lose_money")
|
||
.limit(num)
|
||
)
|
||
x_index = await self.__get_x_index(users, group_id)
|
||
data = [u.lose_money for u in users]
|
||
title = "慈善家排行"
|
||
x_name = "金币"
|
||
if rank_type == "max_win":
|
||
users = (
|
||
await RussianUser.filter(group_id=group_id, max_winning_streak__not=0)
|
||
.order_by("max_winning_streak")
|
||
.limit(num)
|
||
)
|
||
x_index = await self.__get_x_index(users, group_id)
|
||
data = [u.max_winning_streak for u in users]
|
||
title = "最高连胜排行"
|
||
x_name = "场次"
|
||
if rank_type == "max_lose":
|
||
users = (
|
||
await RussianUser.filter(group_id=group_id, max_losing_streak__not=0)
|
||
.order_by("max_losing_streak")
|
||
.limit(num)
|
||
)
|
||
x_index = await self.__get_x_index(users, group_id)
|
||
data = [u.max_losing_streak for u in users]
|
||
title = "最高连败排行"
|
||
x_name = "场次"
|
||
if not data:
|
||
return "当前数据为空..."
|
||
mat = BuildMat(MatType.BARH)
|
||
mat.x_index = x_index
|
||
mat.data = data # type: ignore
|
||
mat.title = title
|
||
mat.x_name = x_name
|
||
return await mat.build()
|
||
|
||
|
||
russian_manage = RussianManage()
|