Exemple #1
0
    def decorator(func):
        async def new_func(*def_args, **def_kwargs):
            message = def_args[0]

            if cmds:
                message.conf['cmds'] = cmds

            if allow_kwargs is False:
                def_kwargs = dict()

            # Sentry
            with configure_scope() as scope:
                scope.set_extra("update", str(message))

            if DEBUG_MODE:
                # log.debug('[*] Starting {}.'.format(func.__name__))
                # log.debug('Event: \n' + str(message))
                start = time.time()
                await func(*def_args, **def_kwargs)
                log.debug('[*] {} Time: {} sec.'.format(
                    func.__name__,
                    time.time() - start))
            else:
                await func(*def_args, **def_kwargs)
            raise SkipHandler()

        if f == 'cb':
            dp.register_callback_query_handler(new_func, *args,
                                               **register_kwargs)
        else:
            dp.register_message_handler(new_func, *args, **register_kwargs)
            if allow_edited is True:
                dp.register_edited_message_handler(new_func, *args,
                                                   **register_kwargs)
Exemple #2
0
    def decorator(func):
        async def new_func(message, *args, **def_kwargs):

            if RATE_LIMIT and await prevent_flooding(message,
                                                     message.text) is False:
                return

            if allow_kwargs is False:
                def_kwargs = dict()
            await func(message, *args, **def_kwargs)
            raise SkipHandler()

        dp.register_message_handler(new_func, *args, **register_kwargs)
        if allow_edited is True:
            dp.register_edited_message_handler(new_func, *args,
                                               **register_kwargs)
Exemple #3
0
    def decorator(func):
        async def new_func(message, *args, **def_kwargs):

            if RATE_LIMIT and await prevent_flooding(message, message.text) is False:
                return

            if allow_kwargs is False:
                def_kwargs = dict()
            if DEBUG_MODE:
                logger.debug('[*] Starting {}.'.format(func.__name__))
                start = time.time()
                await func(message, *args, **def_kwargs)
                logger.debug('[*] {} Time: {} sec.'.format(func.__name__, time.time() - start))
            else:
                await func(message, *args, **def_kwargs)
            raise SkipHandler()

        dp.register_message_handler(new_func, *args, **register_kwargs)
        if allow_edited is True:
            dp.register_edited_message_handler(new_func, *args, **register_kwargs)
Exemple #4
0
    def decorator(func):
        if ALLOW_COMMANDS_FROM_EXC is True:
            P = '[/!]'
        else:
            P = '/'

        if 'not_gbanned' not in kwargs and BLOCK_GBANNED_USERS is True:
            kwargs['not_gbanned'] = True

        if 'not_forwarded' not in kwargs and ALLOW_F_COMMANDS is False:
            kwargs['not_forwarded'] = True

        cmd = "^{0}(?i:{1}|{1}@{2})(?: |$)".format(P, command, BOT_USERNAME)

        async def new_func(*args, **def_kwargs):
            if 'allow_kwargs' not in kwargs:
                def_kwargs = dict()
            await func(*args, **def_kwargs)
            raise SkipHandler()

        dp.register_message_handler(new_func, regexp=cmd, **kwargs)
        if allow_edited is True:
            dp.register_edited_message_handler(new_func, regexp=cmd, **kwargs)