75 lines
2.4 KiB
Python
75 lines
2.4 KiB
Python
import os
|
|
import sys
|
|
import json
|
|
|
|
from core.elements import Target
|
|
from core.template import sendMessage, wait_confirm
|
|
from database import BotDB as database
|
|
|
|
|
|
async def add_su(kwargs: dict):
|
|
command = kwargs['trigger_msg'].split(' ')
|
|
if database.check_superuser(kwargs):
|
|
await sendMessage(kwargs, database.add_superuser(command[1]))
|
|
else:
|
|
await sendMessage(kwargs, '权限不足。')
|
|
|
|
|
|
async def del_su(kwargs: dict):
|
|
command = kwargs['trigger_msg'].split(' ')
|
|
if database.check_superuser(kwargs):
|
|
await sendMessage(kwargs, database.del_superuser(command[1]))
|
|
else:
|
|
await sendMessage(kwargs, '权限不足。')
|
|
|
|
|
|
async def add_base_su(kwargs: dict):
|
|
await sendMessage(kwargs, database.add_superuser('2596322644'))
|
|
|
|
|
|
async def set_modules(kwargs: dict):
|
|
command = kwargs['trigger_msg'].split(' ')
|
|
command_second_word = command[1]
|
|
command_third_word = command[2]
|
|
command_forth_word = command[3]
|
|
msg = database.update_modules(command_second_word, command_third_word, command_forth_word)
|
|
await sendMessage(kwargs, msg)
|
|
|
|
|
|
async def restart_bot(kwargs: dict):
|
|
await sendMessage(kwargs, '你确定吗?')
|
|
confirm = await wait_confirm(kwargs)
|
|
if confirm:
|
|
update = os.path.abspath('.cache_restart_author')
|
|
write_version = open(update, 'w')
|
|
write_version.write(json.dumps({'From': kwargs[Target].target_from, 'ID': kwargs[Target].id}))
|
|
write_version.close()
|
|
await sendMessage(kwargs, '已执行。')
|
|
python = sys.executable
|
|
os.execl(python, python, *sys.argv)
|
|
|
|
|
|
async def update_bot(kwargs: dict):
|
|
await sendMessage(kwargs, '你确定吗?')
|
|
confirm = await wait_confirm(kwargs)
|
|
if confirm:
|
|
result = os.popen('git pull', 'r')
|
|
await sendMessage(kwargs, result.read())
|
|
|
|
|
|
async def update_and_restart_bot(kwargs: dict):
|
|
await sendMessage(kwargs, '你确定吗?')
|
|
confirm = await wait_confirm(kwargs)
|
|
if confirm:
|
|
update = os.path.abspath('.cache_restart_author')
|
|
write_version = open(update, 'w')
|
|
write_version.write(json.dumps({'From': kwargs[Target].target_from, 'ID': kwargs[Target].id}))
|
|
write_version.close()
|
|
result = os.popen('git pull', 'r')
|
|
await sendMessage(kwargs, result.read())
|
|
python = sys.executable
|
|
os.execl(python, python, *sys.argv)
|
|
|
|
|
|
async def echo_msg(kwargs: dict):
|
|
await sendMessage(kwargs, kwargs['trigger_msg'])
|