blob: 638ab9ba13a82127b62edfccb7da32aa883502eb (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
import typing
import pyrogram
from pyrogram.api import types as tgtypes
from lib.common import CommonContext
from util import property_decorator
commands = {}
def get_all_commands():
return commands.values()
class CommandContext(CommonContext):
def __init__(self, client: pyrogram.Client, channel, args: typing.List[str], message: tgtypes.Message):
super().__init__(client, channel, message)
import re
self.args = args
self.rest_content = re.sub('^.*? ', '', message.message)
def register_command(func):
if not is_command(func):
return
commands[get_command_name(func)] = func
def is_command(func):
return hasattr(func, 'name') and func.name is not None
def get_command_name(func):
return func.name
def get_command_description(func):
return func.description
def get_command_by_name(command_name: str):
return commands[command_name.lower()]
name = property_decorator('name')
description = property_decorator('description')
|
