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

Command dict convert to constant

Browse Source
This commit is contained in:
iwumingz 2022-04-09 11:54:00 +08:00
parent 2977d0a71b
commit 48e22f89d1
2 changed files with 12 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
core/__init__.py
View File

@ -1,4 +1,4 @@
from .custom import command
from .custom import command, CMDS_DATA, CMDS_PREFIX
from pyrogram import Client
app = Client(
@ -9,5 +9,8 @@ app = Client(
__all__ = (
'app', 'command',
'app',
'command',
'CMDS_DATA',
'CMDS_PREFIX',
)

14
core/custom.py
View File

@ -18,18 +18,18 @@ from pyrogram.types import Message
from tools.constants import STORE_TRACE_DATA, COMMAND_YML
from tools.storage import SimpleStore
command_data: Dict[str, Any] = yaml.full_load(open(COMMAND_YML, 'rb'))
CMDS_DATA: Dict[str, Any] = yaml.full_load(open(COMMAND_YML, 'rb'))
CMDS_PREFIX = CMDS_DATA.get('help').get('all_prefixes')
def command(key: str):
"""匹配UserBot指令"""
prefixes = command_data.get('help').get('all_prefixes')
if type(command_data.get(key).get('cmd')) == str:
cmd = [command_data.get(key).get('cmd'), key]
return filters.me & filters.text & filters.command(cmd, prefixes)
if type(CMDS_DATA.get(key).get('cmd')) == str:
cmd = [CMDS_DATA.get(key).get('cmd'), key]
return filters.me & filters.text & filters.command(cmd, CMDS_PREFIX)
cmd = command_data.get(key).get('cmd')
return filters.me & filters.text & filters.command(cmd, prefixes)
cmd = CMDS_DATA.get(key).get('cmd')
return filters.me & filters.text & filters.command(cmd, CMDS_PREFIX)
def is_traced():