Limnoria/plugins/Conditional/plugin.py

344 lines
11 KiB
Python
Raw Normal View History

2011-02-20 11:13:57 +01:00
###
# Copyright (c) 2010, Daniel Folkinshteyn
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
# * Redistributions of source code must retain the above copyright notice,
# this list of conditions, and the following disclaimer.
# * Redistributions in binary form must reproduce the above copyright notice,
# this list of conditions, and the following disclaimer in the
# documentation and/or other materials provided with the distribution.
# * Neither the name of the author of this software nor the name of
# contributors to this software may be used to endorse or promote products
# derived from this software without specific prior written consent.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
###
import supybot.utils as utils
import supybot.plugins as plugins
import supybot.commands as commands
2011-02-20 11:13:57 +01:00
import supybot.ircutils as ircutils
import supybot.callbacks as callbacks
import re
import collections
2011-02-20 11:13:57 +01:00
try:
from supybot.i18n import PluginInternationalization
from supybot.i18n import internationalizeDocstring
_ = PluginInternationalization('Conditional')
except:
# This are useless functions that's allow to run the plugin on a bot
# without the i18n plugin
_ = lambda x:x
internationalizeDocstring = lambda x:x
first = commands.first
many = commands.many
wrap = commands.wrap
getopts = commands.getopts
2011-02-20 11:13:57 +01:00
boolean_or_int = first('int', 'boolean')
2011-02-20 11:13:57 +01:00
class Conditional(callbacks.Plugin):
"""This plugin provides logic operators and other commands that
enable you to run commands only if a condition is true. Useful for nested
commands and scripting."""
2011-02-20 11:13:57 +01:00
threaded = True
def __init__(self, irc):
callbacks.Plugin.__init__(self, irc)
2011-02-20 11:13:57 +01:00
def _runCommandFunction(self, irc, msg, command):
"""Run a command from message, as if command was sent over IRC."""
tokens = callbacks.tokenize(command)
2011-02-20 11:13:57 +01:00
try:
self.Proxy(irc.irc, msg, tokens)
except Exception as e:
self.log.exception('Uncaught exception in requested function:')
@internationalizeDocstring
2011-02-20 11:13:57 +01:00
def cif(self, irc, msg, args, condition, ifcommand, elsecommand):
"""<condition> <ifcommand> <elsecommand>
2011-02-20 11:13:57 +01:00
Runs <ifcommand> if <condition> evaluates to true, runs <elsecommand>
if it evaluates to false.
2011-02-20 11:13:57 +01:00
Use other logical operators defined in this plugin and command nesting
to your advantage here.
"""
if condition:
self._runCommandFunction(irc, msg, ifcommand)
else:
self._runCommandFunction(irc, msg, elsecommand)
irc.noReply()
cif = wrap(cif, [boolean_or_int, 'something', 'something'])
@internationalizeDocstring
2011-02-20 11:13:57 +01:00
def cand(self, irc, msg, args, conds):
"""<cond1> [<cond2> ... <condN>]
2011-02-20 11:13:57 +01:00
Returns true if all conditions supplied evaluate to true.
"""
if all(conds):
2011-02-20 11:13:57 +01:00
irc.reply("true")
else:
irc.reply("false")
cand = wrap(cand, [many(boolean_or_int),])
@internationalizeDocstring
2011-02-20 11:13:57 +01:00
def cor(self, irc, msg, args, conds):
"""<cond1> [<cond2> ... <condN>]
2011-02-20 11:13:57 +01:00
Returns true if any one of conditions supplied evaluates to true.
"""
if any(conds):
2011-02-20 11:13:57 +01:00
irc.reply("true")
else:
irc.reply("false")
cor = wrap(cor, [many(boolean_or_int),])
@internationalizeDocstring
2011-02-20 11:13:57 +01:00
def cxor(self, irc, msg, args, conds):
"""<cond1> [<cond2> ... <condN>]
2011-02-20 11:13:57 +01:00
Returns true if only one of conditions supplied evaluates to true.
"""
if sum(conds) == 1:
irc.reply("true")
else:
irc.reply("false")
cxor = wrap(cxor, [many(boolean_or_int),])
@internationalizeDocstring
2011-02-20 11:13:57 +01:00
def ceq(self, irc, msg, args, item1, item2):
"""<item1> <item2>
Does a string comparison on <item1> and <item2>.
2011-02-20 11:13:57 +01:00
Returns true if they are equal.
"""
if item1 == item2:
irc.reply('true')
else:
irc.reply('false')
ceq = wrap(ceq, ['anything', 'anything'])
@internationalizeDocstring
2011-02-20 11:13:57 +01:00
def ne(self, irc, msg, args, item1, item2):
"""<item1> <item2>
Does a string comparison on <item1> and <item2>.
2011-02-20 11:13:57 +01:00
Returns true if they are not equal.
"""
if item1 != item2:
irc.reply('true')
else:
irc.reply('false')
ne = wrap(ne, ['anything', 'anything'])
@internationalizeDocstring
2011-02-20 11:13:57 +01:00
def gt(self, irc, msg, args, item1, item2):
"""<item1> <item2>
Does a string comparison on <item1> and <item2>.
2011-02-20 11:13:57 +01:00
Returns true if <item1> is greater than <item2>.
"""
if item1 > item2:
irc.reply('true')
else:
irc.reply('false')
gt = wrap(gt, ['anything', 'anything'])
@internationalizeDocstring
2011-02-20 11:13:57 +01:00
def ge(self, irc, msg, args, item1, item2):
"""<item1> <item2>
Does a string comparison on <item1> and <item2>.
2011-02-20 11:13:57 +01:00
Returns true if <item1> is greater than or equal to <item2>.
"""
if item1 >= item2:
irc.reply('true')
else:
irc.reply('false')
ge = wrap(ge, ['anything', 'anything'])
@internationalizeDocstring
2011-02-20 11:13:57 +01:00
def lt(self, irc, msg, args, item1, item2):
"""<item1> <item2>
Does a string comparison on <item1> and <item2>.
2011-02-20 11:13:57 +01:00
Returns true if <item1> is less than <item2>.
"""
if item1 < item2:
irc.reply('true')
else:
irc.reply('false')
lt = wrap(lt, ['anything', 'anything'])
@internationalizeDocstring
2011-02-20 11:13:57 +01:00
def le(self, irc, msg, args, item1, item2):
"""<item1> <item2>
Does a string comparison on <item1> and <item2>.
2011-02-20 11:13:57 +01:00
Returns true if <item1> is less than or equal to <item2>.
"""
if item1 <= item2:
irc.reply('true')
else:
irc.reply('false')
le = wrap(le, ['anything', 'anything'])
@internationalizeDocstring
def match(self, irc, msg, args, optlist, item1, item2):
"""[--case-insensitive] <item1> <item2>
Determines if <item1> is a substring of <item2>.
2011-02-20 11:13:57 +01:00
Returns true if <item1> is contained in <item2>.
Will only match case if --case-insensitive is not given.
2011-02-20 11:13:57 +01:00
"""
optlist = dict(optlist)
if 'case-insensitive' in optlist:
item1 = item1.lower()
item2 = item2.lower()
2011-02-20 11:13:57 +01:00
if item2.find(item1) != -1:
irc.reply('true')
else:
irc.reply('false')
match = wrap(match, [getopts({'case-insensitive': ''}),
'something', 'something'])
2011-02-20 11:13:57 +01:00
@internationalizeDocstring
2011-02-20 11:13:57 +01:00
def nceq(self, irc, msg, args, item1, item2):
"""<item1> <item2>
Does a numeric comparison on <item1> and <item2>.
2011-02-20 11:13:57 +01:00
Returns true if they are equal.
"""
if item1 == item2:
irc.reply('true')
else:
irc.reply('false')
nceq = wrap(nceq, ['float', 'float'])
@internationalizeDocstring
2011-02-20 11:13:57 +01:00
def nne(self, irc, msg, args, item1, item2):
"""<item1> <item2>
Does a numeric comparison on <item1> and <item2>.
2011-02-20 11:13:57 +01:00
Returns true if they are not equal.
"""
if item1 != item2:
irc.reply('true')
else:
irc.reply('false')
nne = wrap(nne, ['float', 'float'])
@internationalizeDocstring
2011-02-20 11:13:57 +01:00
def ngt(self, irc, msg, args, item1, item2):
"""<item1> <item2>
Does a numeric comparison on <item1> and <item2>.
2011-07-08 10:41:36 +02:00
Returns true if <item1> is greater than <item2>.
2011-02-20 11:13:57 +01:00
"""
if item1 > item2:
irc.reply('true')
else:
irc.reply('false')
ngt = wrap(ngt, ['float', 'float'])
@internationalizeDocstring
2011-02-20 11:13:57 +01:00
def nge(self, irc, msg, args, item1, item2):
"""<item1> <item2>
Does a numeric comparison on <item1> and <item2>.
2011-02-20 11:13:57 +01:00
Returns true if <item1> is greater than or equal to <item2>.
"""
if item1 >= item2:
irc.reply('true')
else:
irc.reply('false')
nge = wrap(nge, ['float', 'float'])
@internationalizeDocstring
2011-02-20 11:13:57 +01:00
def nlt(self, irc, msg, args, item1, item2):
"""<item1> <item2>
Does a numeric comparison on <item1> and <item2>.
2011-02-20 11:13:57 +01:00
Returns true if <item1> is less than <item2>.
"""
if item1 < item2:
irc.reply('true')
else:
irc.reply('false')
nlt = wrap(nlt, ['float', 'float'])
@internationalizeDocstring
2011-02-20 11:13:57 +01:00
def nle(self, irc, msg, args, item1, item2):
"""<item1> <item2>
Does a numeric comparison on <item1> and <item2>.
2011-02-20 11:13:57 +01:00
Returns true if <item1> is less than or equal to <item2>.
"""
if item1 <= item2:
irc.reply('true')
else:
irc.reply('false')
nle = wrap(nle, ['float', 'float'])
def cerror(self, irc, msg, args, testcommand):
"""<testcommand>
Runs <testcommand> and returns true if it raises an error;
false otherwise.
"""
tokens = callbacks.tokenize(testcommand)
InvalidCommand = collections.namedtuple('InvalidCommand',
'command')
replies = []
errors = []
class ErrorReportingProxy(self.Proxy):
def reply(self2, s, *args, **kwargs):
replies.append(s)
def error(self2, s, Raise=False, *args, **kwargs):
errors.append(s)
if Raise:
raise ArgumentError
def _callInvalidCommands(self2):
errors.append(InvalidCommand(self2.args))
def evalArgs(self2):
# We don't want the replies in the nested command to
# be stored here.
super(ErrorReportingProxy, self2).evalArgs(withClass=self.Proxy)
try:
ErrorReportingProxy(irc.irc, msg, tokens)
except callbacks.ArgumentError as e:
pass
# TODO: do something with the results
if errors:
irc.reply('true')
else:
irc.reply('false')
cerror = wrap(cerror, ['something'])
Condition = internationalizeDocstring(Conditional)
2011-02-20 11:13:57 +01:00
Class = Conditional
# vim:set shiftwidth=4 softtabstop=4 expandtab textwidth=79: