microbot/ubot/command.py

# SPDX-License-Identifier: GPL-2.0-or-later

from types import FunctionType


class Command:
    def __init__(self, func: FunctionType, args: dict):
        self.module = func.__module__.split(".")[-1]
        self.function = func

        self.pattern = args.get("pattern")
        self.simple_pattern = args.get("simple_pattern", False)
        self.raw_pattern = args.get("raw_pattern", False)
        self.pattern_extra = args.get("pattern_extra", "")
        self.extra = args.get("extra", None)
        self.help = args.get("help", None)
        self.hide_help = args.get("hide_help", False)
        self.owner = args.get("owner", False)
        self.sudo = args.get("sudo", False)
        self.admin = args.get("admin", False)
        self.nsfw = args.get("nsfw", False)
        self.pass_nsfw = args.get("pass_nsfw", False)
        self.locking = args.get("locking", False)
        self.lock_reason = None
        self.user_locking = args.get("userlocking", False)
        self.locked_users = []
        self.chance = args.get("chance", None)
        self.fun = args.get("fun", False)
        self.not_disableable = args.get("no_disable", False) or self.owner or self.sudo or self.admin


class InlinePhotoCommand:
    def __init__(self, func: FunctionType, args: dict):
        self.module = func.__module__.split(".")[-1]
        self.function = func

        self.pattern = args.get("pattern")
        self.default = args.get("default", None)


class InlineArticleCommand:
    def __init__(self, func: FunctionType, args: dict):
        self.module = func.__module__.split(".")[-1]
        self.function = func

        self.pattern = args.get("pattern")
        self.default = args.get("default", None)


class CallbackQueryCommand:
    def __init__(self, func: FunctionType, args: dict):
        self.module = func.__module__.split(".")[-1]
        self.function = func

        self.data_id = args.get("data_id")
        self.extra = args.get("extra", None)