mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
96 lines
3.4 KiB
Python
96 lines
3.4 KiB
Python
from nonebot.adapters.cqhttp import Bot, MessageEvent
|
||
from nonebot.typing import T_State
|
||
from nonebot.permission import SUPERUSER
|
||
from nonebot import on_command
|
||
from .data_source import check_update, get_latest_version_data
|
||
from services.log import logger
|
||
from utils.utils import scheduler, get_bot
|
||
from pathlib import Path
|
||
from configs.config import AUTO_UPDATE_ZHENXUN
|
||
from nonebot.rule import to_me
|
||
import platform
|
||
import os
|
||
|
||
|
||
update_zhenxun = on_command('检查更新真寻', permission=SUPERUSER, priority=1, block=True)
|
||
|
||
restart = on_command('重启', aliases={'restart'}, permission=SUPERUSER, rule=to_me(), priority=1, block=True)
|
||
|
||
|
||
@update_zhenxun.handle()
|
||
async def _(bot: Bot, event: MessageEvent, state: T_State):
|
||
try:
|
||
code = await check_update(bot)
|
||
except Exception as e:
|
||
logger.error(f'更新真寻未知错误 {type(e)}：{e}')
|
||
await bot.send_private_msg(
|
||
user_id=int(list(bot.config.superusers)[0]),
|
||
message=f'更新真寻未知错误 {type(e)}：{e}'
|
||
)
|
||
else:
|
||
if code == 200:
|
||
await bot.send_private_msg(
|
||
user_id=int(list(bot.config.superusers)[0]),
|
||
message=f'更新完毕，请重启真寻....'
|
||
)
|
||
|
||
|
||
@restart.handle()
|
||
async def _(bot: Bot, event: MessageEvent, state: T_State):
|
||
if str(platform.system()).lower() == 'windows':
|
||
await restart.finish('暂无windows重启脚本...')
|
||
|
||
|
||
@restart.got('flag', prompt='确定是否重启真寻？（重启失败咱们将失去联系，请谨慎！）')
|
||
async def _(bot: Bot, event: MessageEvent, state: T_State):
|
||
flag = state['flag']
|
||
if flag.lower() in ['true', '是', '好', '确定', '确定是']:
|
||
await restart.send('开始重启真寻..请稍等...')
|
||
open('is_restart', 'w')
|
||
os.system('./restart.sh')
|
||
else:
|
||
await restart.send('已取消操作...')
|
||
|
||
|
||
@scheduler.scheduled_job(
|
||
"cron",
|
||
hour=12,
|
||
minute=0,
|
||
)
|
||
async def _():
|
||
if AUTO_UPDATE_ZHENXUN:
|
||
_version = "v0.0.0"
|
||
_version_file = Path() / "__version__"
|
||
if _version_file.exists():
|
||
_version = (
|
||
open(_version_file, "r", encoding="utf8").readline().split(":")[-1].strip()
|
||
)
|
||
data = await get_latest_version_data()
|
||
if data:
|
||
latest_version = data["name"]
|
||
if _version != latest_version:
|
||
bot = get_bot()
|
||
await bot.send_private_msg(
|
||
user_id=int(list(bot.config.superusers)[0]),
|
||
message=f'检测到真寻版本更新\n'
|
||
f'当前版本：{_version}，最新版本：{latest_version}\n'
|
||
f'尝试自动更新...'
|
||
)
|
||
try:
|
||
code = await check_update(bot)
|
||
except Exception as e:
|
||
logger.error(f'更新真寻未知错误 {type(e)}：{e}')
|
||
await bot.send_private_msg(
|
||
user_id=int(list(bot.config.superusers)[0]),
|
||
message=f'更新真寻未知错误 {type(e)}：{e}\n'
|
||
)
|
||
else:
|
||
if code == 200:
|
||
await bot.send_private_msg(
|
||
user_id=int(list(bot.config.superusers)[0]),
|
||
message=f'更新完毕，请重启真寻....'
|
||
)
|
||
|
||
|
||
|