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')
|