This commit is contained in:
deathbybandaid 2019-06-24 15:10:56 -04:00
parent 119a6d80d8
commit 6a3e6cf210

View File

@ -69,7 +69,7 @@ def prerun_old(trigger_command_type='module'):
return actual_decorator
def prerun(message="this is not privmsg", reply=False):
def prerun(trigger_command_type="this is not privmsg"):
def actual_decorator(function):
@functools.wraps(function)
@ -77,18 +77,15 @@ def prerun(message="this is not privmsg", reply=False):
# Assign trigger and bot for easy access later
bot, trigger = args[0:2]
if not trigger.is_privmsg:
if message and not callable(message):
if reply:
bot.reply(message)
else:
bot.say(message)
if trigger_command_type and not callable(trigger_command_type):
bot.say(trigger_command_type)
else:
return function(*args, **kwargs)
return _nop
# Hack to allow decorator without parens
if callable(message):
return actual_decorator(message)
if callable(trigger_command_type):
return actual_decorator(trigger_command_type)
return actual_decorator