mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-14 21:52:56 +08:00
97 lines
3.1 KiB
Python
97 lines
3.1 KiB
Python
import os
|
||
from pathlib import Path
|
||
import platform
|
||
|
||
import aiofiles
|
||
import nonebot
|
||
from nonebot import on_command
|
||
from nonebot.adapters import Bot
|
||
from nonebot.params import ArgStr
|
||
from nonebot.permission import SUPERUSER
|
||
from nonebot.plugin import PluginMetadata
|
||
from nonebot.rule import to_me
|
||
from nonebot_plugin_uninfo import Uninfo
|
||
|
||
from zhenxun.configs.config import BotConfig
|
||
from zhenxun.configs.utils import PluginExtraData
|
||
from zhenxun.services.log import logger
|
||
from zhenxun.utils.enum import PluginType
|
||
from zhenxun.utils.message import MessageUtils
|
||
from zhenxun.utils.platform import PlatformUtils
|
||
|
||
__plugin_meta__ = PluginMetadata(
|
||
name="重启",
|
||
description="执行脚本重启真寻",
|
||
usage="""
|
||
重启
|
||
""".strip(),
|
||
extra=PluginExtraData(
|
||
author="HibiKier", version="0.1", plugin_type=PluginType.SUPERUSER
|
||
).to_dict(),
|
||
)
|
||
|
||
|
||
_matcher = on_command(
|
||
"重启",
|
||
permission=SUPERUSER,
|
||
rule=to_me(),
|
||
priority=1,
|
||
block=True,
|
||
)
|
||
|
||
driver = nonebot.get_driver()
|
||
|
||
|
||
RESTART_MARK = Path() / "is_restart"
|
||
|
||
RESTART_FILE = Path() / "restart.sh"
|
||
|
||
|
||
@_matcher.got(
|
||
"flag",
|
||
prompt=f"确定是否重启{BotConfig.self_nickname}？\n确定请回复[是|好|确定]\n（重启失败咱们将失去联系，请谨慎！）",
|
||
)
|
||
async def _(bot: Bot, session: Uninfo, flag: str = ArgStr("flag")):
|
||
if flag.lower() in {"true", "是", "好", "确定", "确定是"}:
|
||
await MessageUtils.build_message(
|
||
f"开始重启{BotConfig.self_nickname}..请稍等..."
|
||
).send()
|
||
async with aiofiles.open(RESTART_MARK, "w", encoding="utf8") as f:
|
||
await f.write(f"{bot.self_id} {session.user.id}")
|
||
logger.info("开始重启真寻...", "重启", session=session)
|
||
if str(platform.system()).lower() == "windows":
|
||
import sys
|
||
|
||
python = sys.executable
|
||
os.execl(python, python, *sys.argv)
|
||
else:
|
||
os.system("./restart.sh") # noqa: ASYNC221
|
||
else:
|
||
await MessageUtils.build_message("已取消操作...").send()
|
||
|
||
|
||
@driver.on_bot_connect
|
||
async def _(bot: Bot):
|
||
if str(platform.system()).lower() != "windows" and not RESTART_FILE.exists():
|
||
async with aiofiles.open(RESTART_FILE, "w", encoding="utf8") as f:
|
||
await f.write(
|
||
"pid=$(netstat -tunlp | grep "
|
||
+ str(bot.config.port)
|
||
+ " | awk '{print $7}')\n"
|
||
"pid=${pid%/*}\n"
|
||
"kill -9 $pid\n"
|
||
"sleep 3\n"
|
||
"python3 bot.py"
|
||
)
|
||
os.system("chmod +x ./restart.sh") # noqa: ASYNC221
|
||
logger.info("已自动生成 restart.sh(重启) 文件，请检查脚本是否与本地指令符合...")
|
||
if RESTART_MARK.exists():
|
||
async with aiofiles.open(RESTART_MARK, encoding="utf8") as f:
|
||
bot_id, user_id = (await f.read()).split()
|
||
if bot := nonebot.get_bot(bot_id):
|
||
if target := PlatformUtils.get_target(user_id=user_id):
|
||
await MessageUtils.build_message(
|
||
f"{BotConfig.self_nickname}已成功重启！"
|
||
).send(target, bot=bot)
|
||
RESTART_MARK.unlink()
|