async def help_(message: discord.Message, command: str.lower=None, *args):
""" Display commands or their usage and description. """
# Display the specific command
if command:
if command.startswith(config.command_prefix):
command = command[len(config.command_prefix):]
cmd = plugins.get_command(command)
if not cmd:
return
# Get the specific command with arguments and send the help
cmd = plugins.get_sub_command(cmd, *args)
await client.say(message, utils.format_help(cmd))
# Display every command
else:
commands = []
for plugin in plugins.all_values():
if getattr(plugin, "__commands", False): # Massive pile of shit that works (so sorry)
commands.extend(
cmd.name_prefix.split()[0] for cmd in plugin.__commands
if not cmd.hidden and
(not getattr(getattr(cmd, "function"), "owner", False) or
utils.is_owner(message.author))
)
commands = ", ".join(sorted(commands))
m = "**Commands**: ```{0}```Use `{1}help <command>`, `{1}<command> {2}` or " \
"`{1}<command> {3}` for command specific help.".format(
commands, config.command_prefix, *config.help_arg)
await client.say(message, m)
async def add_tasks():
""" Create any tasks for plugins' on_ready() coroutine and create task
for autosaving. """
await client.wait_until_ready()
logging.info("Setting up background tasks.")
# Call any on_ready function in plugins
for plugin in plugins.all_values():
if hasattr(plugin, "on_ready"):
client.loop.create_task(plugin.on_ready())
client.loop.create_task(autosave())
async def add_tasks():
""" Create any tasks for plugins' on_ready() coroutine and create task
for autosaving. """
await client.wait_until_ready()
logging.info("Setting up background tasks.")
# Call any on_ready function in plugins
for plugin in plugins.all_values():
if hasattr(plugin, "on_ready"):
client.loop.create_task(plugin.on_ready())
client.loop.create_task(autosave())
def help_(client: discord.Client, message: discord.Message, command: str.lower=None, *args):
""" Display commands or their usage and description. """
# Display the specific command
if command:
if command.startswith(config.command_prefix):
command = command[1:]
for plugin in plugins.all_values():
cmd = plugins.get_command(plugin, command)
if not cmd:
continue
# Get the specific command with arguments and send the help
cmd = plugins.get_sub_command(cmd, args)
yield from client.say(message, utils.format_help(cmd))
break
# Display every command
else:
commands = []
for plugin in plugins.all_values():
if getattr(plugin, "__commands", False): # Massive pile of shit that works (so sorry)
commands.extend(
cmd.name_prefix.split()[0] for cmd in plugin.__commands
if not cmd.hidden and
(not getattr(getattr(cmd, "function"), "__owner__", False) or
utils.is_owner(message.author))
)
commands = ", ".join(sorted(commands))
m = "**Commands**:```{0}```Use `{1}help <command>`, `{1}<command> {2}` or " \
"`{1}<command> {3}` for command specific help.".format(
commands, config.command_prefix, *config.help_arg)
yield from client.say(message, m)
async def on_message(message: discord.Message):
""" What to do on any message received.
The bot will handle all commands in plugins and send on_message to plugins using it. """
# Make sure the client is ready before processing commands
await client.wait_until_ready()
start_time = datetime.now()
# We don't care about channels we can't write in as the bot usually sends feedback
if not message.channel.is_private and not message.server.me.permissions_in(message.channel).send_messages:
return
# Don't accept commands from bot accounts
if message.author.bot:
return
# Split content into arguments by space (surround with quotes for spaces)
cmd_args = utils.split(message.content)
# Get command name
cmd = ""
if cmd_args[0].startswith(config.command_prefix) and len(cmd_args[0]) > len(config.command_prefix):
cmd = cmd_args[0][len(config.command_prefix):]
# Handle commands
for plugin in plugins.all_values():
# If there was a command and the bot can send messages in the channel, parse the command
if not cmd:
continue
command = plugins.get_command(plugin, cmd)
if command:
parsed_command, args, kwargs = await parse_command(command, cmd_args, message)
if parsed_command:
log_message(message) # Log the command
client.loop.create_task(execute_command(parsed_command, message, *args, **kwargs)) # Run command
# Log time spent parsing the command
stop_time = datetime.now()
time_elapsed = (stop_time - start_time).total_seconds() / 1000
logging.debug("Time spent parsing command: {elapsed:.6f}ms".format(elapsed=time_elapsed))
async def help_(message: discord.Message, command: str.lower = None, *args):
""" Display commands or their usage and description. """
command_prefix = config.server_command_prefix(message.server)
# Display the specific command
if command:
if command.startswith(command_prefix):
command = command[len(command_prefix):]
cmd = plugins.get_command(command)
if not cmd:
return
# Get the specific command with arguments and send the help
cmd = plugins.get_sub_command(cmd, *args)
await client.say(message, plugins.format_help(cmd, message.server))
# Display every command
else:
commands = []
for plugin in plugins.all_values():
# Only go through plugins with actual commands
if not getattr(plugin, "__commands", False):
continue
# Add all commands that the user can use
for cmd in plugin.__commands:
if not cmd.hidden and plugins.can_use_command(
cmd, message.author, message.channel):
commands.append(cmd.name_prefix(message.server).split()[0])
commands = ", ".join(sorted(commands))
m = "**Commands**: ```{0}```Use `{1}help <command>`, `{1}<command> {2}` or " \
"`{1}<command> {3}` for command specific help.".format(
commands, command_prefix, *config.help_arg)
await client.say(message, m)