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

Add source cmd to regex

Browse Source
This commit is contained in:
iwumingz 2022-04-08 21:58:47 +08:00
parent 332cebf22f
commit a5c9d4e077
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
core/custom.py
View File

@ -15,18 +15,19 @@ from typing import Any, Dict
import yaml import yaml
from pyrogram import filters from pyrogram import filters
from pyrogram.types import Message from pyrogram.types import Message
from tools.constants import STORE_TRACE_DATA from tools.constants import STORE_TRACE_DATA, COMMAND_YML
from tools.storage import SimpleStore from tools.storage import SimpleStore
command_data: Dict[str, Any] = yaml.full_load(open("./data/command.yml", 'rb')) command_data: Dict[str, Any] = yaml.full_load(open(COMMAND_YML, 'rb'))
def command(key: str): def command(key: str):
"""匹配UserBot指令""" """匹配UserBot指令"""
# 指令前缀
# 例一：prefixes = "-"
# 例二：prefixes = ["-", "/", "+"]
prefixes = command_data.get('help').get('all_prefixes') 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)
cmd = command_data.get(key).get('cmd') cmd = command_data.get(key).get('cmd')
return filters.me & filters.text & filters.command(cmd, prefixes) return filters.me & filters.text & filters.command(cmd, prefixes)