Using commands.wrap to parse your command’s arguments

Introduction

To plugin developers for older (pre-0.80) versions of Supybot, one of the more annoying aspects of writing commands was handling the arguments that were passed in. In fact, many commands often had to duplicate parsing and verification code, resulting in lots of duplicated code for not a whole lot of action. So, instead of forcing plugin writers to come up with their own ways of cleaning it up, we wrote up the wrap function to handle all of it.

It allows a much simpler and more flexible way of checking things than before and it doesn’t require that you know the bot internals to do things like check and see if a user exists, or check if a command name exists and whatnot.

If you are a plugin author this document is absolutely required reading, as it will massively ease the task of writing commands.

Using Wrap

First off, to get the wrap function, it is recommended (strongly) that you use the following import line:

from supybot.commands import *

This will allow you to access the wrap command (and it allows you to do it without the commands prefix). Note that this line is added to the imports of plugin templates generated by the supybot-plugin-create script.

Let’s write a quickie command that uses wrap to get a feel for how it makes our lives better. Let’s write a command that repeats a string of text a given number of times. So you could say “repeat 3 foo” and it would say “foofoofoo”. Not a very useful command, but it will serve our purpose just fine. Here’s how it would be done without wrap:

def repeat(self, irc, msg, args):
    """<num> <text>

    Repeats <text> <num> times.
    """
    (num, text) = privmsg.getArgs(args, required=2)
    try:
        num = int(num)
    except ValueError:
        raise callbacks.ArgumentError
    irc.reply(num * text)

Note that all of the argument validation and parsing takes up 5 of the 6 lines (and you should have seen it before we had privmsg.getArgs!). Now, here’s what our command will look like with wrap applied:

def repeat(self, irc, msg, args, num, text):
    """<num> <text>

    Repeats <text> <num> times.
    """
    irc.reply(text * num)
repeat = wrap(repeat, ['int', 'text'])

Pretty short, eh? With wrap all of the argument parsing and validation is handled for us and we get the arguments we want, formatted how we want them, and converted into whatever types we want them to be - all in one simple function call that is used to wrap the function! So now the code inside each command really deals with how to execute the command and not how to deal with the input.

So, now that you see the benefits of wrap, let’s figure out what stuff we have to do to use it.

Syntax Changes

There are two syntax changes to the old style that are implemented. First, the definition of the command function must be changed. The basic syntax for the new definition is:

def commandname(self, irc, msg, args, <arg1>, <arg2>, ...):

Where arg1 and arg2 (up through as many as you want) are the variables that will store the parsed arguments. “Now where do these parsed arguments come from?” you ask. Well, that’s where the second syntax change comes in. The second syntax change is the actual use of the wrap function itself to decorate our command names. The basic decoration syntax is:

commandname = wrap(commandname, [converter1, converter2, ...])

Note

This should go on the line immediately following the body of the command’s definition, so it can easily be located (and it obviously must go after the command’s definition so that commandname is defined).

Each of the converters in the above listing should be one of the converters in commands.py (I will describe each of them in detail later.) The converters are applied in order to the arguments given to the command, generally taking arguments off of the front of the argument list as they go. Note that each of the arguments is actually a string containing the NAME of the converter to use and not a reference to the actual converter itself. This way we can have converters with names like int and not have to worry about polluting the builtin namespace by overriding the builtin int.

As you will find out when you look through the list of converters below, some of the converters actually take arguments. The syntax for supplying them (since we aren’t actually calling the converters, but simply specifying them), is to wrap the converter name and args list into a tuple. For example:

commandname = wrap(commandname, [(converterWithArgs, arg1, arg2),
                                 converterWithoutArgs1, converterWithoutArgs2])

For the most part you won’t need to use an argument with the converters you use either because the defaults are satisfactory or because it doesn’t even take any.

Customizing Wrap

Converters alone are a pretty powerful tool, but for even more advanced (yet simpler!) argument handling you may want to use contexts. Contexts describe how the converters are applied to the arguments, while the converters themselves do the actual parsing and validation.

For example, one of the contexts is “optional”. By using this context, you’re saying that a given argument is not required, and if the supplied converter doesn’t find anything it likes, we should use some default. Yet another example is the “reverse” context. This context tells the supplied converter to look at the last argument and work backwards instead of the normal first-to-last way of looking at arguments.

So, that should give you a feel for the role that contexts play. They are not by any means necessary to use wrap. All of the stuff we’ve done to this point will work as-is. However, contexts let you do some very powerful things in very easy ways, and are a good thing to know how to use.

Now, how do you use them? Well, they are in the global namespace of src/commands.py, so your previous import line will import them all; you can call them just as you call wrap. In fact, the way you use them is you simply call the context function you want to use, with the converter (and its arguments) as arguments. It’s quite simple. Here’s an example:

commandname = wrap(commandname, [optional('int'), many('something')])

In this example, our command is looking for an optional integer argument first. Then, after that, any number of arguments which can be anything (as long as they are something, of course).

Do note, however, that the type of the arguments that are returned can be changed if you apply a context to it. So, optional(“int”) may very well return None as well as something that passes the “int” converter, because after all it’s an optional argument and if it is None, that signifies that nothing was there. Also, for another example, many(“something”) doesn’t return the same thing that just “something” would return, but rather a list of “something”s.

Converter List

Below is a list of all the available converters to use with wrap. If the converter accepts any arguments, they are listed after it and if they are optional, the default value is shown.

Numbers and time

expiry
Takes a number of seconds and adds it to the current time to create an expiration timestamp.
id, kind=”integer”
Returns something that looks like an integer ID number. Takes an optional “kind” argument for you to state what kind of ID you are looking for, though this doesn’t affect the integrity-checking. Basically requires that the argument be an integer, does no other integrity-checking, and provides a nice error message with the kind in it.
index
Basically (“int”, “index”), but with a twist. This will take a 1-based index and turn it into a 0-based index (which is more useful in code). It doesn’t transform 0, and it maintains negative indices as is (note that it does allow them!).
int, type=”integer”, p=None
Gets an integer. The “type” text can be used to customize the error message received when the argument is not an integer. “p” is an optional predicate to test the integer with. If p(i) fails (where i is the integer arg parsed out of the argument string), the arg will not be accepted.
now
Simply returns the current timestamp as an arg, does not reference or modify the argument list.
long, type=”long”
Basically the same as int minus the predicate, except that it converts the argument to a long integer regardless of the size of the int.
float, type=”floating point number”
Basically the same as int minus the predicate, except that it converts the argument to a float.
nonInt, type=”non-integer value”
Accepts everything but integers, and returns them unchanged. The “type” value, as always, can be used to customize the error message that is displayed upon failure.
positiveInt
Accepts only positive integers.
nonNegativeInt
Accepts only non-negative integers.

Channel

channelDb
Sets the channel appropriately in order to get to the databases for that channel (handles whether or not a given channel uses channel-specific databases and whatnot).
channel
Gets a channel to use the command in. If the channel isn’t supplied, uses the channel the message was sent in. If using a different channel, does sanity-checking to make sure the channel exists on the current IRC network.
inChannel
Requires that the command be called from within any channel that the bot is currently in or with one of those channels used as an argument to the command.
onlyInChannel
Requires that the command be called from within any channel that the bot is currently in.
callerInGivenChannel
Takes the given argument as a channel and makes sure that the caller is in that channel.
public
Requires that the command be sent in a channel instead of a private message.
private
Requires that the command be sent in a private message instead of a channel.
validChannel
Gets a channel argument once it makes sure it’s a valid channel.

Words

color
Accepts arguments that describe a text color code (e.g., “black”, “light blue”) and returns the mIRC color code for that color. (Note that many other IRC clients support the mIRC color code scheme, not just mIRC)
letter
Looks for a single letter. (Technically, it looks for any one-element sequence).
literal, literals, errmsg=None
Takes a required sequence or string (literals) and any argument that uniquely matches the starting substring of one of the literals is transformed into the full literal. For example, with ("literal", ("bar", "baz", "qux")), you’d get “bar” for “bar”, “baz” for “baz”, and “qux” for any of “q”, “qu”, or “qux”. “b” and “ba” would raise errors because they don’t uniquely identify one of the literals in the list. You can override errmsg to provide a specific (full) error message, otherwise the default argument error message is displayed.
lowered
Returns the argument lowered (NOTE: it is lowered according to IRC conventions, which does strange mapping with some punctuation characters).
to
Returns the string “to” if the arg is any form of “to” (case-insensitive).

Network

ip
Checks and makes sure the argument looks like a valid IP and then returns it.
url
Checks for a valid URL.
httpUrl
Checks for a valid HTTP URL.

Users, nicks, and permissions

haveOp, action=”do that”
Simply requires that the bot have ops in the channel that the command is called in. The action parameter completes the error message: “I need to be opped to …”.
nick
Checks that the arg is a valid nick on the current IRC server.
seenNick
Checks that the arg is a nick that the bot has seen (NOTE: this is limited by the size of the history buffer that the bot has).
nickInChannel
Requires that the argument be a nick that is in the current channel, and returns that nick.
capability
Used to retrieve an argument that describes a capability.
hostmask
Returns the hostmask of any provided nick or hostmask argument.
banmask
Returns a generic banmask of the provided nick or hostmask argument.
user
Requires that the caller be a registered user.
otherUser
Returns the user specified by the username or hostmask in the argument.
owner
Requires that the command caller has the “owner” capability.
admin
Requires that the command caller has the “admin” capability.
checkCapability, capability
Checks to make sure that the caller has the specified capability.
checkChannelCapability, capability
Checks to make sure that the caller has the specified capability on the channel the command is called in.

Matching

anything
Returns anything as is.
something, errorMsg=None, p=None
Takes anything but the empty string. errorMsg can be used to customize the error message. p is any predicate function that can be used to test the validity of the input.
somethingWithoutSpaces
Same as something, only with the exception of disallowing spaces of course.
matches, regexp, errmsg
Searches the args with the given regexp and returns the matches. If no match is found, errmsg is given.
regexpMatcher
Gets a matching regexp argument (m// or //).
glob
Gets a glob string. Basically, if there are no wildcards (*, ?) in the argument, returns *string*, making a glob string that matches anything containing the given argument.
regexpReplacer
Gets a replacing regexp argument (s//).

Other

networkIrc, errorIfNoMatch=False
Returns the IRC object of the specified IRC network. If one isn’t specified, the IRC object of the IRC network the command was called on is returned.
plugin, require=True
Returns the plugin specified by the arg or None. If require is True, an error is raised if the plugin cannot be retrieved.
boolean
Converts the text string to a boolean value. Acceptable true values are: “1”, “true”, “on”, “enable”, or “enabled” (case-insensitive). Acceptable false values are: “0”, false”, “off”, “disable”, or “disabled” (case-insensitive).
filename
Used to get a filename argument.
commandName
Returns the canonical command name version of the given string (ie, the string is lowercased and dashes and underscores are removed).
text
Takes the rest of the arguments as one big string. Note that this differs from the “anything” context in that it clobbers the arg string when it’s done. Using any converters after this is most likely incorrect.

Contexts List

What contexts are available for me to use?

The list of available contexts is below. Unless specified otherwise, it can be assumed that the type returned by the context itself matches the type of the converter it is applied to.

Options

optional
Look for an argument that satisfies the supplied converter, but if it’s not the type I’m expecting or there are no arguments for us to check, then use the default value. Will return the converted argument as is or None.
additional
Look for an argument that satisfies the supplied converter, making sure that it’s the right type. If there aren’t any arguments to check, then use the default value. Will return the converted argument as is or None.
first
Tries each of the supplied converters in order and returns the result of the first successfully applied converter.

Multiplicity

any
Looks for any number of arguments matching the supplied converter. Will return a sequence of converted arguments or None.
many
Looks for multiple arguments matching the supplied converter. Expects at least one to work, otherwise it will fail. Will return the sequence of converted arguments.
commalist
Looks for a comma separated list of arguments that match the supplied converter. Returns a list of the successfully converted arguments. If any of the arguments fail, this whole context fails.

Other

rest
Treat the rest of the arguments as one big string, and then convert. If the conversion is unsuccessful, restores the arguments.
getopts
Handles –option style arguments. Each option should be a key in a dictionary that maps to the name of the converter that is to be used on that argument. To make the option take no argument, use “” as the converter name in the dictionary. For no conversion, use None as the converter name in the dictionary.
reverse
Reverse the argument list, apply the converters, and then reverse the argument list back.

Final Word

Now that you know how to use wrap, and you have a list of converters and contexts you can use, your task of writing clean, simple, and safe plugin code should become much easier. Enjoy!