mirror of
https://github.com/Mikaela/Limnoria.git
synced 2024-11-09 12:29:22 +01:00
37597bfe94
This is the only way to limit the execution time of a possibly long-running python statement. Use this on String.re, due to the possibility of pathologically long re matching in python. This allows us to remove the 'trusted-only' restriction on string.re. In the future, this should probably be used in other places that take user-supplied regexps, such as 'misc last --regexp', for example, as well as other potentially long-running tasks that can block the bot. Conflicts: plugins/String/plugin.py src/commands.py
199 lines
7.3 KiB
Python
199 lines
7.3 KiB
Python
###
|
|
# Copyright (c) 2003-2005, Jeremiah Fincher
|
|
# Copyright (c) 2008-2009, James Vega
|
|
# 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 types
|
|
import binascii
|
|
|
|
import supybot.utils as utils
|
|
from supybot.commands import *
|
|
import supybot.commands as commands
|
|
import supybot.plugins as plugins
|
|
import supybot.ircutils as ircutils
|
|
import supybot.callbacks as callbacks
|
|
from supybot.i18n import PluginInternationalization, internationalizeDocstring
|
|
_ = PluginInternationalization('String')
|
|
|
|
import multiprocessing
|
|
|
|
class String(callbacks.Plugin):
|
|
@internationalizeDocstring
|
|
def ord(self, irc, msg, args, letter):
|
|
"""<letter>
|
|
|
|
Returns the 8-bit value of <letter>.
|
|
"""
|
|
irc.reply(str(ord(letter)))
|
|
ord = wrap(ord, ['letter'])
|
|
|
|
@internationalizeDocstring
|
|
def chr(self, irc, msg, args, i):
|
|
"""<number>
|
|
|
|
Returns the character associated with the 8-bit value <number>
|
|
"""
|
|
try:
|
|
irc.reply(chr(i))
|
|
except ValueError:
|
|
irc.error(_('That number doesn\'t map to an 8-bit character.'))
|
|
chr = wrap(chr, ['int'])
|
|
|
|
@internationalizeDocstring
|
|
def encode(self, irc, msg, args, encoding, text):
|
|
"""<encoding> <text>
|
|
|
|
Returns an encoded form of the given text; the valid encodings are
|
|
available in the documentation of the Python codecs module:
|
|
<http://docs.python.org/library/codecs.html#standard-encodings>.
|
|
"""
|
|
try:
|
|
irc.reply(text.encode(encoding).rstrip('\n'))
|
|
except LookupError:
|
|
irc.errorInvalid(_('encoding'), encoding)
|
|
encode = wrap(encode, ['something', 'text'])
|
|
|
|
@internationalizeDocstring
|
|
def decode(self, irc, msg, args, encoding, text):
|
|
"""<encoding> <text>
|
|
|
|
Returns an un-encoded form of the given text; the valid encodings are
|
|
available in the documentation of the Python codecs module:
|
|
<http://docs.python.org/library/codecs.html#standard-encodings>.
|
|
"""
|
|
try:
|
|
irc.reply(text.decode(encoding))
|
|
except LookupError:
|
|
irc.errorInvalid(_('encoding'), encoding)
|
|
except binascii.Error:
|
|
irc.errorInvalid(_('base64 string'),
|
|
s=_('Base64 strings must be a multiple of 4 in '
|
|
'length, padded with \'=\' if necessary.'))
|
|
decode = wrap(decode, ['something', 'text'])
|
|
|
|
@internationalizeDocstring
|
|
def levenshtein(self, irc, msg, args, s1, s2):
|
|
"""<string1> <string2>
|
|
|
|
Returns the levenshtein distance (also known as the "edit distance"
|
|
between <string1> and <string2>)
|
|
"""
|
|
max = self.registryValue('levenshtein.max')
|
|
if len(s1) > max or len(s2) > max:
|
|
irc.error(_('Levenshtein distance is a complicated algorithm, try '
|
|
'it with some smaller inputs.'))
|
|
else:
|
|
irc.reply(str(utils.str.distance(s1, s2)))
|
|
levenshtein = wrap(levenshtein, ['something', 'text'])
|
|
|
|
@internationalizeDocstring
|
|
def soundex(self, irc, msg, args, text, length):
|
|
"""<string> [<length>]
|
|
|
|
Returns the Soundex hash to a given length. The length defaults to
|
|
4, since that's the standard length for a soundex hash. For unlimited
|
|
length, use 0.
|
|
"""
|
|
irc.reply(utils.str.soundex(text, length))
|
|
soundex = wrap(soundex, ['somethingWithoutSpaces', additional('int', 4)])
|
|
|
|
@internationalizeDocstring
|
|
def len(self, irc, msg, args, text):
|
|
"""<text>
|
|
|
|
Returns the length of <text>.
|
|
"""
|
|
irc.reply(str(len(text)))
|
|
len = wrap(len, ['text'])
|
|
|
|
@internationalizeDocstring
|
|
def re(self, irc, msg, args, ff, text):
|
|
"""<regexp> <text>
|
|
|
|
If <regexp> is of the form m/regexp/flags, returns the portion of
|
|
<text> that matches the regexp. If <regexp> is of the form
|
|
s/regexp/replacement/flags, returns the result of applying such a
|
|
regexp to <text>.
|
|
"""
|
|
if isinstance(ff, (types.FunctionType, types.MethodType)):
|
|
f = ff
|
|
else:
|
|
f = lambda s: ff.search(s) and ff.search(s).group(0) or ''
|
|
if f('') and len(f(' ')) > len(f(''))+1: # Matches the empty string.
|
|
s = _('You probably don\'t want to match the empty string.')
|
|
irc.error(s)
|
|
else:
|
|
t = self.registryValue('re.timeout')
|
|
try:
|
|
v = commands.process(f, text, timeout=t, pn=self.name(), cn='re')
|
|
irc.reply(v)
|
|
except commands.ProcessTimeoutError, e:
|
|
irc.error("ProcessTimeoutError: %s" % (e,))
|
|
re = thread(wrap(re, [first('regexpMatcher', 'regexpReplacer'),
|
|
'text']))
|
|
|
|
@internationalizeDocstring
|
|
def xor(self, irc, msg, args, password, text):
|
|
"""<password> <text>
|
|
|
|
Returns <text> XOR-encrypted with <password>. See
|
|
http://www.yoe.org/developer/xor.html for information about XOR
|
|
encryption.
|
|
"""
|
|
chars = utils.iter.cycle(password)
|
|
ret = [chr(ord(c) ^ ord(chars.next())) for c in text]
|
|
irc.reply(''.join(ret))
|
|
xor = wrap(xor, ['something', 'text'])
|
|
|
|
@internationalizeDocstring
|
|
def md5(self, irc, msg, args, text):
|
|
"""<text>
|
|
|
|
Returns the md5 hash of a given string. Read
|
|
http://www.rsasecurity.com/rsalabs/faq/3-6-6.html for more information
|
|
about md5.
|
|
"""
|
|
irc.reply(utils.crypt.md5(text).hexdigest())
|
|
md5 = wrap(md5, ['text'])
|
|
|
|
@internationalizeDocstring
|
|
def sha(self, irc, msg, args, text):
|
|
"""<text>
|
|
|
|
Returns the SHA hash of a given string. Read
|
|
http://www.secure-hash-algorithm-md5-sha-1.co.uk/ for more information
|
|
about SHA.
|
|
"""
|
|
irc.reply(utils.crypt.sha(text).hexdigest())
|
|
sha = wrap(sha, ['text'])
|
|
|
|
Class = String
|
|
|
|
|
|
# vim:set shiftwidth=4 softtabstop=4 expandtab textwidth=79:
|