xtaodada/sycgram
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Modify restore method

Browse Source
This commit is contained in:
iwumingz 2022-04-09 20:44:01 +08:00
parent ca9946d0a6
commit 5e3a598abb
1 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
plugins/update.py
View File

@ -1,13 +1,12 @@
import asyncio import asyncio
import shutil
import sys import sys
from core import command from core import command
from loguru import logger from loguru import logger
from pyrogram import Client from pyrogram import Client
from pyrogram.types import Message from pyrogram.types import Message
from tools.constants import (COMMAND_YML, EXAMPLE_YML, SYCGRAM, SYCGRAM_ERROR, from tools.constants import (SYCGRAM, SYCGRAM_ERROR, SYCGRAM_INFO,
SYCGRAM_INFO, SYCGRAM_WARNING, UPDATE_CMD) SYCGRAM_WARNING, UPDATE_CMD)
from tools.helpers import Parameters, basher, show_cmd_tip, show_exception from tools.helpers import Parameters, basher, show_cmd_tip, show_exception
from tools.updates import (get_alias_of_cmds, pull_and_update_command_yml, from tools.updates import (get_alias_of_cmds, pull_and_update_command_yml,
reset_cmd_alias, update_cmd_alias, reset_cmd_alias, update_cmd_alias,
@ -47,13 +46,11 @@ async def prefix(_: Client, msg: Message):
punctuation = list("""!#$%&*+,-./:;=?@^~！？。，；·\\""") punctuation = list("""!#$%&*+,-./:;=?@^~！？。，；·\\""")
if pfx == "reset": if pfx == "reset":
try: try:
_ = shutil.copy(EXAMPLE_YML, COMMAND_YML) await pull_and_update_command_yml(is_update=False)
except shutil.SameFileError:
return await msg.edit_text("😂 No need to reset")
except Exception as e: except Exception as e:
return await show_exception(msg, e) return await show_exception(msg, e)
else: else:
await msg.edit_text("✅ Reset command.yml") await msg.edit_text("✅ Restore command.yml to default.")
sys.exit() sys.exit()
elif len(pfx) == 0 or len(pfx) > 1 or pfx not in punctuation: elif len(pfx) == 0 or len(pfx) > 1 or pfx not in punctuation: