184 lines
7.6 KiB
Python
184 lines
7.6 KiB
Python
import inspect
|
|
import re
|
|
from typing import Union, overload
|
|
|
|
from apscheduler.triggers.combining import AndTrigger, OrTrigger
|
|
from apscheduler.triggers.cron import CronTrigger
|
|
from apscheduler.triggers.date import DateTrigger
|
|
from apscheduler.triggers.interval import IntervalTrigger
|
|
|
|
from core.loader import ModulesManager
|
|
from core.parser.args import parse_template
|
|
from core.types import Module
|
|
from core.types.module.component_meta import *
|
|
|
|
|
|
class Bind:
|
|
class Module:
|
|
def __init__(self, bind_prefix):
|
|
self.bind_prefix = bind_prefix
|
|
|
|
def command(self,
|
|
help_doc: Union[str, list, tuple] = None,
|
|
*help_docs,
|
|
options_desc: dict = None,
|
|
required_admin: bool = False,
|
|
required_superuser: bool = False,
|
|
required_base_superuser: bool = False,
|
|
available_for: Union[str, list, tuple] = '*',
|
|
exclude_from: Union[str, list, tuple] = '',
|
|
priority: int = 1):
|
|
def decorator(function):
|
|
nonlocal help_doc
|
|
if isinstance(help_doc, str):
|
|
help_doc = [help_doc]
|
|
if help_docs:
|
|
help_doc += help_docs
|
|
if help_doc is None:
|
|
help_doc = []
|
|
|
|
ModulesManager.bind_to_module(
|
|
self.bind_prefix,
|
|
CommandMeta(
|
|
function=function,
|
|
help_doc=parse_template(help_doc),
|
|
options_desc=options_desc,
|
|
required_admin=required_admin,
|
|
required_superuser=required_superuser,
|
|
required_base_superuser=required_base_superuser,
|
|
available_for=available_for,
|
|
exclude_from=exclude_from,
|
|
priority=priority))
|
|
return function
|
|
|
|
return decorator
|
|
|
|
def regex(self, pattern: Union[str, re.Pattern], mode: str = "M", flags: re.RegexFlag = 0,
|
|
desc: str = None,
|
|
required_admin: bool = False,
|
|
required_superuser: bool = False,
|
|
required_base_superuser: bool = False,
|
|
available_for: Union[str, list, tuple] = '*',
|
|
exclude_from: Union[str, list, tuple] = '',
|
|
show_typing: bool = True, logging: bool = True):
|
|
def decorator(function):
|
|
ModulesManager.bind_to_module(
|
|
self.bind_prefix,
|
|
RegexMeta(
|
|
function=function,
|
|
pattern=pattern,
|
|
mode=mode,
|
|
flags=flags,
|
|
desc=desc,
|
|
required_admin=required_admin,
|
|
required_superuser=required_superuser,
|
|
required_base_superuser=required_base_superuser,
|
|
available_for=available_for,
|
|
exclude_from=exclude_from,
|
|
show_typing=show_typing,
|
|
logging=logging))
|
|
return function
|
|
|
|
return decorator
|
|
|
|
def schedule(self, trigger: Union[AndTrigger, OrTrigger, DateTrigger, CronTrigger, IntervalTrigger]):
|
|
def decorator(function):
|
|
ModulesManager.bind_to_module(self.bind_prefix, ScheduleMeta(function=function, trigger=trigger))
|
|
return function
|
|
|
|
return decorator
|
|
|
|
def hook(self, name: str = None):
|
|
def decorator(function):
|
|
ModulesManager.bind_to_module(self.bind_prefix, HookMeta(function=function, name=name))
|
|
return function
|
|
|
|
return decorator
|
|
|
|
on_command = command
|
|
on_regex = regex
|
|
on_schedule = schedule
|
|
on_hook = hook
|
|
|
|
@overload
|
|
def handle(self,
|
|
help_doc: Union[str, list, tuple] = None,
|
|
*help_docs,
|
|
options_desc: dict = None,
|
|
required_admin: bool = False,
|
|
required_superuser: bool = False,
|
|
required_base_superuser: bool = False,
|
|
available_for: Union[str, list, tuple] = '*',
|
|
exclude_from: Union[str, list, tuple] = '',
|
|
priority: int = 1):
|
|
...
|
|
|
|
@overload
|
|
def handle(self, pattern: Union[str, re.Pattern], mode: str = "M", flags: re.RegexFlag = 0,
|
|
desc: str = None,
|
|
required_admin: bool = False,
|
|
required_superuser: bool = False,
|
|
required_base_superuser: bool = False,
|
|
available_for: Union[str, list, tuple] = '*',
|
|
exclude_from: Union[str, list, tuple] = '',
|
|
show_typing: bool = True, logging: bool = True):
|
|
...
|
|
|
|
@overload
|
|
def handle(self, trigger: Union[AndTrigger, OrTrigger, DateTrigger, CronTrigger, IntervalTrigger]):
|
|
...
|
|
|
|
def handle(self, *args, **kwargs):
|
|
first_key = args[0] if args else (kwargs[list(kwargs.keys())[0]] if kwargs else None)
|
|
if isinstance(first_key, re.Pattern):
|
|
return self.regex(*args, **kwargs)
|
|
elif isinstance(first_key, (AndTrigger, OrTrigger, DateTrigger, CronTrigger, IntervalTrigger)):
|
|
return self.schedule(*args, **kwargs)
|
|
return self.command(*args, **kwargs)
|
|
|
|
|
|
def module(
|
|
bind_prefix: str,
|
|
alias: Union[str, list, tuple, dict] = None,
|
|
desc: str = None,
|
|
recommend_modules: Union[str, list, tuple] = None,
|
|
developers: Union[str, list, tuple] = None,
|
|
required_admin: bool = False,
|
|
base: bool = False,
|
|
required_superuser: bool = False,
|
|
required_base_superuser: bool = False,
|
|
available_for: Union[str, list, tuple] = '*',
|
|
exclude_from: Union[str, list, tuple] = '',
|
|
support_languages: Union[str, list, tuple] = None
|
|
):
|
|
"""
|
|
|
|
:param bind_prefix: 绑定的命令前缀。
|
|
:param alias: 此命令的别名。
|
|
同时被用作命令解析，当此项不为空时将会尝试解析其中的语法并储存结果在 MessageSession.parsed_msg 中。
|
|
:param desc: 此命令的简介。
|
|
:param recommend_modules: 推荐打开的其他模块。
|
|
:param developers: 模块作者。
|
|
:param required_admin: 此命令是否需要群组管理员权限。
|
|
:param base: 将此命令设为基础命令。设为基础命令后此命令将被强制开启。
|
|
:param required_superuser: 将此命令设为机器人的超级管理员才可执行。
|
|
:param required_base_superuser: 将此命令设为机器人的基础超级管理员才可执行。
|
|
:param available_for: 此命令支持的平台列表。
|
|
:param exclude_from: 此命令排除的平台列表。
|
|
:return: 此类型的模块。
|
|
"""
|
|
module = Module(alias=alias,
|
|
bind_prefix=bind_prefix,
|
|
desc=desc,
|
|
recommend_modules=recommend_modules,
|
|
developers=developers,
|
|
base=base,
|
|
required_admin=required_admin,
|
|
required_superuser=required_superuser,
|
|
required_base_superuser=required_base_superuser,
|
|
available_for=available_for,
|
|
exclude_from=exclude_from,
|
|
support_languages=support_languages)
|
|
frame = inspect.currentframe()
|
|
ModulesManager.add_module(module, frame.f_back.f_globals["__name__"])
|
|
return Bind.Module(bind_prefix)
|