113 lines
3.6 KiB
Python
113 lines
3.6 KiB
Python
from json import dump, load
|
||
from os import makedirs, path
|
||
from typing import List, Optional
|
||
|
||
from nonebot import get_bot, get_bots, get_driver, get_plugin_config
|
||
from nonebot.adapters import Bot
|
||
from pydantic import BaseModel
|
||
|
||
|
||
class Config(BaseModel):
|
||
# 管理员的QQ号（别问我为什么）
|
||
mc_status_admin_qqnum: List[int] = [] # 必填
|
||
# 机器人的QQ号（如果写了就按优先级响应，否则就第一个连上的响应） ['1234','5678','6666']
|
||
mc_status_bot_qqnum_list: List[str] = [] # 可选
|
||
# 数据文件名
|
||
mc_status_data_filename: str = "mc_status_data.json"
|
||
|
||
|
||
class Var:
|
||
# 处理消息的bot
|
||
handle_bot: Optional[Bot] = None
|
||
# {"123456": {"提肛": ["mc.hypixel.net:25565","java"]}}
|
||
group_list = {}
|
||
|
||
|
||
driver = get_driver()
|
||
pc = get_plugin_config(Config)
|
||
var = Var()
|
||
|
||
|
||
@driver.on_startup
|
||
async def on_startup():
|
||
if not path.exists(f"data"):
|
||
makedirs(f"data")
|
||
|
||
if not path.exists(f"data/{pc.mc_status_data_filename}"):
|
||
save_file()
|
||
else:
|
||
load_file()
|
||
|
||
|
||
def load_file():
|
||
with open(f"data/{pc.mc_status_data_filename}", "r", encoding="utf-8") as r:
|
||
tmp_data = load(r)
|
||
for i in tmp_data:
|
||
var.group_list[int(i)] = tmp_data[i]
|
||
|
||
|
||
def save_file():
|
||
with open(f"data/{pc.mc_status_data_filename}", "w", encoding="utf-8") as w:
|
||
dump(var.group_list, w, indent=4, ensure_ascii=False)
|
||
|
||
|
||
# qq机器人连接时执行
|
||
@driver.on_bot_connect
|
||
async def on_bot_connect(bot: Bot):
|
||
# 是否有写bot qq，如果写了只处理bot qq在列表里的
|
||
if pc.mc_status_bot_qqnum_list and bot.self_id in pc.mc_status_bot_qqnum_list:
|
||
# 如果已经有bot连了
|
||
if var.handle_bot:
|
||
# 当前bot qq 下标
|
||
handle_bot_id_index = pc.mc_status_bot_qqnum_list.index(
|
||
var.handle_bot.self_id
|
||
)
|
||
# 新连接的bot qq 下标
|
||
new_bot_id_index = pc.mc_status_bot_qqnum_list.index(bot.self_id)
|
||
# 判断优先级，下标越低优先级越高
|
||
if new_bot_id_index < handle_bot_id_index:
|
||
var.handle_bot = bot
|
||
|
||
# 没bot连就直接给
|
||
else:
|
||
var.handle_bot = bot
|
||
|
||
# 不写就给第一个连的
|
||
elif not pc.mc_status_bot_qqnum_list and not var.handle_bot:
|
||
var.handle_bot = bot
|
||
|
||
|
||
# qq机器人断开时执行
|
||
@driver.on_bot_disconnect
|
||
async def on_bot_disconnect(bot: Bot):
|
||
# 判断掉线的是否为handle bot
|
||
if bot == var.handle_bot:
|
||
# 如果有写bot qq列表
|
||
if pc.mc_status_bot_qqnum_list:
|
||
# 获取当前连着的bot列表(需要bot是在bot qq列表里)
|
||
available_bot_id_list = [
|
||
bot_id for bot_id in get_bots() if bot_id in pc.mc_status_bot_qqnum_list
|
||
]
|
||
if available_bot_id_list:
|
||
# 打擂台排序？
|
||
new_bot_index = pc.mc_status_bot_qqnum_list.index(
|
||
available_bot_id_list[0]
|
||
)
|
||
for bot_id in available_bot_id_list:
|
||
now_bot_index = pc.mc_status_bot_qqnum_list.index(bot_id)
|
||
if now_bot_index < new_bot_index:
|
||
new_bot_index = now_bot_index
|
||
# 取下标在qq列表里最小的bot qq为新的handle bot
|
||
var.handle_bot = get_bot(pc.mc_status_bot_qqnum_list[new_bot_index])
|
||
|
||
else:
|
||
var.handle_bot = None
|
||
|
||
# 不写就随便给一个连着的(如果有)
|
||
elif var.handle_bot:
|
||
try:
|
||
new_bot = get_bot()
|
||
var.handle_bot = new_bot
|
||
except ValueError:
|
||
var.handle_bot = None
|