34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
|
import asyncio
|
|||
|
|
import sys
|
|||
|
|
|
|||
|
|
from core import command
|
|||
|
|
from pyrogram import Client
|
|||
|
|
from pyrogram.types import Message
|
|||
|
|
from tools.constants import SYCGRAM, SYCGRAM_ERROR, SYCGRAM_INFO, SYCGRAM_WARNING, UPDATE_CMD
|
|||
|
|
from tools.helpers import basher
|
|||
|
|
|
|||
|
|
|
|||
|
|
@Client.on_message(command("restart"))
|
|||
|
|
async def restart(_: Client, msg: Message):
|
|||
|
|
"""重启容器"""
|
|||
|
|
text = f"**{SYCGRAM_INFO}**\n> # `The {SYCGRAM} is restarting......`"
|
|||
|
|
await msg.edit_text(text=text, parse_mode='md')
|
|||
|
|
sys.exit()
|
|||
|
|
|
|||
|
|
|
|||
|
|
@Client.on_message(command("update"))
|
|||
|
|
async def update(_: Client, msg: Message):
|
|||
|
|
"""更新sycgram到主分支的最新版本"""
|
|||
|
|
text = f"**{SYCGRAM_INFO}**\n> # `It's updating container to the latest version......`"
|
|||
|
|
await msg.edit_text(text, parse_mode='md')
|
|||
|
|
try:
|
|||
|
|
_ = await basher(UPDATE_CMD, timeout=60)
|
|||
|
|
except asyncio.exceptions.TimeoutError:
|
|||
|
|
text = f"**{SYCGRAM_WARNING}**\n> # `Update Timeout！`"
|
|||
|
|
except Exception as e:
|
|||
|
|
text = f"**{SYCGRAM_ERROR}**\n> # `{e}`"
|
|||
|
|
else:
|
|||
|
|
text = f"**{SYCGRAM_INFO}**\n> # `Your {SYCGRAM} version is the latest.`"
|
|||
|
|
finally:
|
|||
|
|
await msg.edit_text(text, parse_mode='md')
|