sopel-irc / sopel

    return actual_decorator

def require_admin(message=None, reply=False):

    """Decorate a function to require the triggering user to be a bot admin.

    :param str message: optional message said to non-admin user

    :param bool reply: use :meth:`~.bot.Sopel.reply` instead of

                       :meth:`~.bot.Sopel.say` when ``True``; defaults to

                       ``False``

    When the triggering user is not an admin, the command is not run, and the

    bot will say the ``message`` if given. By default, it uses

    :meth:`bot.say() <.bot.Sopel.say>`, but when ``reply`` is true, then it

    uses :meth:`bot.reply() <.bot.Sopel.reply>` instead.

    .. versionchanged:: 7.0.0

        Added the ``reply`` parameter.

    """

    def actual_decorator(function):

        @functools.wraps(function)

        def guarded(bot, trigger, *args, **kwargs):

            if not trigger.admin:

                if message and not callable(message):

                    if reply:

                        bot.reply(message)

                    else:

                        bot.say(message)

            else:

                return function(bot, trigger, *args, **kwargs)

        return guarded

    # Hack to allow decorator without parens

    if callable(message):

    return actual_decorator

def require_owner(message=None, reply=False):

    """Decorate a function to require the triggering user to be the bot owner.

    :param str message: optional message said to non-owner user

    :param bool reply: use :meth:`~.bot.Sopel.reply` instead of

                       :meth:`~.bot.Sopel.say` when ``True``; defaults to

                       ``False``

    When the triggering user is not the bot's owner, the command is not run,

    and the bot will say ``message`` if given. By default, it uses

    :meth:`bot.say() <.bot.Sopel.say>`, but when ``reply`` is true, then it

    uses :meth:`bot.reply() <.bot.Sopel.reply>` instead.

    .. versionchanged:: 7.0.0

        Added the ``reply`` parameter.

    """

    def actual_decorator(function):

        @functools.wraps(function)

        def guarded(bot, trigger, *args, **kwargs):

            if not trigger.owner:

                if message and not callable(message):

                    if reply:

                        bot.reply(message)

                    else:

                        bot.say(message)

            else:

                return function(bot, trigger, *args, **kwargs)

        return guarded

    # Hack to allow decorator without parens