mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-10-04 19:29:59 +08:00
modified: basic_plugins/admin_bot_manage/_data_source.py
modified: basic_plugins/admin_bot_manage/admin_config.py modified: basic_plugins/admin_bot_manage/custom_welcome_message.py modified: basic_plugins/admin_bot_manage/timing_task.py modified: basic_plugins/apscheduler/__init__.py modified: basic_plugins/ban/__init__.py modified: basic_plugins/ban/data_source.py modified: basic_plugins/chat_history/chat_message.py modified: basic_plugins/chat_history/chat_message_handle.py modified: basic_plugins/group_handle/__init__.py modified: basic_plugins/hooks/_utils.py modified: basic_plugins/hooks/ban_hook.py modified: basic_plugins/hooks/chkdsk_hook.py modified: basic_plugins/init_plugin_config/__init__.py deleted: basic_plugins/init_plugin_config/init_group_manager.py modified: basic_plugins/invite_manager/__init__.py new file: basic_plugins/invite_manager/utils.py modified: basic_plugins/nickname.py modified: basic_plugins/plugin_shop/__init__.py modified: basic_plugins/plugin_shop/data_source.py modified: basic_plugins/scripts.py modified: basic_plugins/shop/__init__.py modified: basic_plugins/shop/buy.py modified: basic_plugins/shop/gold.py modified: basic_plugins/shop/my_props/__init__.py deleted: basic_plugins/shop/reset_today_gold.py modified: basic_plugins/shop/shop_handle/__init__.py modified: basic_plugins/shop/shop_handle/data_source.py modified: basic_plugins/shop/use/__init__.py modified: basic_plugins/shop/use/data_source.py modified: basic_plugins/super_cmd/__init__.py modified: basic_plugins/super_cmd/bot_friend_group.py modified: basic_plugins/super_cmd/clear_data.py modified: basic_plugins/super_cmd/exec_sql.py modified: basic_plugins/super_cmd/manager_group.py modified: basic_plugins/super_cmd/reload_setting.py modified: basic_plugins/super_cmd/set_admin_permissions.py deleted: basic_plugins/super_cmd/super_task_switch.py modified: basic_plugins/super_cmd/update_friend_group_info.py modified: basic_plugins/super_help/__init__.py modified: basic_plugins/update_info.py modified: configs/config.py modified: configs/utils/__init__.py modified: models/bag_user.py modified: models/ban_user.py modified: models/chat_history.py modified: models/friend_user.py modified: models/goods_info.py modified: models/group_info.py modified: models/group_member_info.py modified: models/level_user.py modified: models/sign_group_user.py modified: models/user_shop_gold_log.py modified: plugins/aconfig/__init__.py modified: plugins/ai/__init__.py modified: plugins/ai/data_source.py modified: plugins/bilibili_sub/__init__.py modified: plugins/bilibili_sub/data_source.py modified: plugins/bilibili_sub/model.py modified: plugins/black_word/__init__.py modified: plugins/black_word/model.py modified: plugins/black_word/utils.py modified: plugins/bt/data_source.py modified: plugins/genshin/almanac/__init__.py modified: plugins/genshin/material_remind/__init__.py modified: plugins/genshin/query_user/_models/__init__.py modified: plugins/genshin/query_user/_utils/__init__.py modified: plugins/genshin/query_user/bind/__init__.py modified: plugins/genshin/query_user/genshin_sign/__init__.py modified: plugins/genshin/query_user/genshin_sign/data_source.py modified: plugins/genshin/query_user/genshin_sign/init_task.py modified: plugins/genshin/query_user/mihoyobbs_sign/__init__.py modified: plugins/genshin/query_user/query_memo/__init__.py modified: plugins/genshin/query_user/query_memo/data_source.py modified: plugins/genshin/query_user/query_role/__init__.py modified: plugins/genshin/query_user/query_role/data_source.py modified: plugins/genshin/query_user/reset_today_query_user_data/__init__.py modified: plugins/genshin/query_user/resin_remind/__init__.py modified: plugins/genshin/query_user/resin_remind/init_task.py modified: plugins/gold_redbag/model.py modified: plugins/image_management/send_image/__init__.py modified: plugins/my_info/__init__.py modified: plugins/open_cases/models/buff_prices.py modified: plugins/open_cases/models/open_cases_user.py modified: plugins/open_cases/open_cases_c.py modified: plugins/open_cases/utils.py modified: plugins/parse_bilibili_json.py modified: plugins/pid_search.py modified: plugins/pix_gallery/__init__.py modified: plugins/pix_gallery/_data_source.py modified: plugins/pix_gallery/_model/omega_pixiv_illusts.py modified: plugins/pix_gallery/_model/pixiv.py modified: plugins/pix_gallery/_model/pixiv_keyword_user.py modified: plugins/pix_gallery/pix_add_keyword.py modified: plugins/pix_gallery/pix_pass_del_keyword.py modified: plugins/pix_gallery/pix_show_info.py modified: plugins/pix_gallery/pix_update.py modified: plugins/pixiv_rank_search/data_source.py modified: plugins/poke/__init__.py modified: plugins/russian/__init__.py modified: plugins/russian/data_source.py modified: plugins/russian/model.py modified: plugins/send_dinggong_voice/__init__.py modified: plugins/send_setu_/_model.py modified: plugins/send_setu_/send_setu/__init__.py modified: plugins/send_setu_/send_setu/data_source.py modified: plugins/send_setu_/update_setu/data_source.py modified: plugins/sign_in/goods_register.py modified: plugins/sign_in/group_user_checkin.py modified: plugins/sign_in/random_event.py modified: plugins/sign_in/utils.py modified: plugins/statistics/_model.py modified: plugins/statistics/statistics_handle.py modified: plugins/statistics/statistics_hook.py modified: plugins/update_picture.py modified: plugins/web_ui/api/request.py modified: plugins/word_bank/_model.py deleted: plugins/word_bank/_old_model.py modified: plugins/word_bank/_rule.py modified: plugins/word_bank/word_handle.py modified: plugins/word_clouds/data_source.py modified: resources/image/sign/sign_res/bar.png modified: resources/image/sign/sign_res/bar_white.png modified: services/db_context.py modified: services/log.py modified: utils/browser.py modified: utils/data_utils.py modified: utils/depends/__init__.py modified: utils/http_utils.py modified: utils/image_utils.py modified: utils/manager/admin_manager.py modified: utils/message_builder.py modified: utils/utils.py
This commit is contained in:
HibiKier
@@ -0,0 +1,192 @@
|
||||
from typing import Optional, Dict, Union
|
||||
|
||||
from nonebot.adapters.onebot.v11 import Message, MessageSegment
|
||||
from pydantic import BaseModel
|
||||
import time
|
||||
|
||||
|
||||
class GameEntry(BaseModel):
|
||||
game_name: str
|
||||
module: str
|
||||
default_msg: str
|
||||
msg_data: Dict[int, Union[str, Message, MessageSegment]]
|
||||
timeout: int # 超时时限
|
||||
anti_concurrency: bool # 是否阻断
|
||||
|
||||
|
||||
class GroupGameStatus(BaseModel):
|
||||
game: GameEntry
|
||||
status: int
|
||||
time: time.time() # 创建时间
|
||||
|
||||
|
||||
class GameManager:
|
||||
def __init__(self):
|
||||
self._data = {}
|
||||
self._status = {}
|
||||
|
||||
def add_game(
|
||||
self,
|
||||
game_name: str,
|
||||
module: str,
|
||||
timeout: int,
|
||||
default_msg: Optional[str] = "游戏还未结束！",
|
||||
msg_data: Dict[int, Union[str, Message, MessageSegment]] = None,
|
||||
anti_concurrency: bool = True,
|
||||
**kwargs,
|
||||
):
|
||||
"""
|
||||
参数:
|
||||
将游戏添加到游戏管理器
|
||||
说明:
|
||||
:param game_name: 游戏名称
|
||||
:param module: 模块名
|
||||
:param timeout: 超时时长
|
||||
:param default_msg: 默认回复消息
|
||||
:param msg_data: 不同状态回复的消息
|
||||
:param anti_concurrency: 是否阻断反并发
|
||||
"""
|
||||
self._data[module] = GameEntry(
|
||||
game_name=game_name,
|
||||
module=module,
|
||||
timeout=timeout,
|
||||
default_msg=default_msg,
|
||||
msg_data=msg_data or {},
|
||||
anti_concurrency=anti_concurrency,
|
||||
**kwargs,
|
||||
)
|
||||
|
||||
def start(self, group_id: int, module: str):
|
||||
"""
|
||||
说明:
|
||||
游戏开始标记
|
||||
参数:
|
||||
:param group_id: 群号
|
||||
:param module: 模块名
|
||||
"""
|
||||
if not self._status.get(group_id):
|
||||
self._status[group_id] = []
|
||||
if module not in [x.game.module for x in self._status[module]]:
|
||||
self._status[group_id].append(
|
||||
GroupGameStatus(game=self._data[module], status=0)
|
||||
)
|
||||
|
||||
def end(self, group_id: int, module: str):
|
||||
"""
|
||||
说明:
|
||||
游戏结束标记
|
||||
参数:
|
||||
:param group_id: 群号
|
||||
:param module: 模块名
|
||||
"""
|
||||
if self._status.get(group_id) and module in [
|
||||
x.game.module for x in self._status[group_id]
|
||||
]:
|
||||
for x in self._status[group_id]:
|
||||
if self._status[group_id][x].game.module == module:
|
||||
self._status[group_id].remove(x)
|
||||
break
|
||||
|
||||
def set_status(self, group_id: int, module: str, status: int):
|
||||
"""
|
||||
说明:
|
||||
设置游戏状态，根据状态发送不同的提示消息 msg_data
|
||||
参数:
|
||||
:param group_id: 群号
|
||||
:param module: 模块名
|
||||
:param status: 状态码
|
||||
"""
|
||||
if self._status.get(group_id) and module in [
|
||||
x.game.module for x in self._status[group_id]
|
||||
]:
|
||||
[x.game.module for x in self._status[group_id] if x.game.module == module][
|
||||
0
|
||||
].status = status
|
||||
|
||||
def check(self, group_id, module: str) -> Optional[str]:
|
||||
"""
|
||||
说明:
|
||||
检查群游戏当前状态并返回提示语句
|
||||
参数:
|
||||
:param group_id: 群号
|
||||
:param module: 模块名
|
||||
"""
|
||||
if module in self._data and self._status.get(group_id):
|
||||
for x in self._status[group_id]:
|
||||
if x.game.anti_concurrency:
|
||||
return f"{x.game.game_name} 还未结束，请等待 {x.game.game_name} 游戏结束！"
|
||||
if self._status.get(group_id) and module in [
|
||||
x.game.module for x in self._status[group_id]
|
||||
]:
|
||||
group_game_status = [
|
||||
x.game.module for x in self._status[group_id] if x.game.module == module
|
||||
][0]
|
||||
if time.time() - group_game_status.time > group_game_status.game.timeout:
|
||||
# 超时结束
|
||||
self.end(group_id, module)
|
||||
else:
|
||||
return (
|
||||
group_game_status.game.msg_data.get(group_game_status.status)
|
||||
or group_game_status.game.default_msg
|
||||
)
|
||||
|
||||
|
||||
game_manager = GameManager()
|
||||
|
||||
|
||||
class Game:
|
||||
"""
|
||||
反并发，游戏重复开始
|
||||
"""
|
||||
def __init__(
|
||||
self,
|
||||
game_name: str,
|
||||
module: str,
|
||||
timeout: int = 60,
|
||||
default_msg: Optional[str] = None,
|
||||
msg_data: Dict[int, Union[str, Message, MessageSegment]] = None,
|
||||
anti_concurrency: bool = True,
|
||||
):
|
||||
"""
|
||||
参数:
|
||||
将游戏添加到游戏管理器
|
||||
说明:
|
||||
:param game_name: 游戏名称
|
||||
:param module: 模块名
|
||||
:param timeout: 超时时长
|
||||
:param default_msg: 默认回复消息
|
||||
:param msg_data: 不同状态回复的消息
|
||||
:param anti_concurrency: 是否阻断反并发
|
||||
"""
|
||||
self.module = module
|
||||
game_manager.add_game(
|
||||
game_name, module, timeout, default_msg, msg_data, anti_concurrency
|
||||
)
|
||||
|
||||
def start(self, group_id: int):
|
||||
"""
|
||||
说明:
|
||||
游戏开始标记
|
||||
参数:
|
||||
:param group_id: 群号
|
||||
"""
|
||||
game_manager.start(group_id, self.module)
|
||||
|
||||
def end(self, group_id: int):
|
||||
"""
|
||||
说明:
|
||||
游戏结束标记
|
||||
参数:
|
||||
:param group_id: 群号
|
||||
"""
|
||||
game_manager.end(group_id, self.module)
|
||||
|
||||
def set_status(self, group_id: int, status: int):
|
||||
"""
|
||||
说明:
|
||||
设置游戏状态，根据状态发送不同的提示消息 msg_data
|
||||
参数:
|
||||
:param group_id: 群号
|
||||
:param status: 状态码
|
||||
"""
|
||||
game_manager.set_status(group_id, self.module, status)
|
||||
Reference in New Issue
Block a user