Mikaela/Limnoria
1
0
Fork 0
mirror of https://github.com/Mikaela/Limnoria.git synced 2024-11-23 19:19:32 +01:00
Code Issues Projects Releases Wiki Activity

Fixed calc command.

Browse Source
This commit is contained in:
Jeremy Fincher 2003-09-07 20:29:42 +00:00
parent 4bb8d6a479
commit 66b62a42cd
2 changed files with 36 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
plugins/FunCommands.py
View File

@ -499,9 +499,13 @@ class FunCommands(callbacks.Privmsg):
_mathEnv = {'__builtins__': new.module('__builtins__'), 'i': 1j}
_mathEnv.update(math.__dict__)
_mathEnv.update(cmath.__dict__)
_mathInt = re.compile(r'(?<!\d|\.)(\d+)(?!\d+|\.|\.\d+)')
_mathHex = re.compile(r'(0x[A-Fa-f\d]+)')
_mathOctal = re.compile(r'(^|[^\dA-Fa-f.])(0[0-7]+)')
_mathRe = re.compile(r'((?:(?<![A-Fa-f\d])-)?'
r'(?:0x[A-Fa-f\d]+|'
r'0[0-7]+|'
r'\d+\.\d+|'
r'\.\d+|'
r'\d+\.|'
r'\d+))')
def _complexToString(self, x):
real = x.real
imag = x.imag
@ -517,10 +521,18 @@ class FunCommands(callbacks.Privmsg):
imag = 0
if imag == 0:
return str(real)
elif real == 0:
return '%s*i' % imag
elif imag == 1:
imag = 'i'
elif imag == -1:
imag = '-i'
elif imag < 0:
imag = '%si' % imag
elif imag > 0:
imag = '+%si' % imag
if real == 0:
return imag.lstrip('+')
else:
return '%s+%si' % (real, imag)
return '%s%s' % (real, imag)
def calc(self, irc, msg, args):
"""<math expression>
@ -531,20 +543,19 @@ class FunCommands(callbacks.Privmsg):
text = privmsgs.getArgs(args)
text = text.translate(string.ascii, '_[] \t')
text = text.replace('lambda', '')
def hex2float(m):
literal = m.group(1)
i = long(literal, 16)
return '%s.0' % i
def oct2float(m):
(previous, literal) = m.groups()
i = long(literal, 8)
return '%s%s.0' % (previous, i)
text = self._mathHex.sub(hex2float, text)
#debug.printf('After unhexing: %r' % text)
text = self._mathOctal.sub(oct2float, text)
#debug.printf('After unocting: %r' % text)
text = self._mathInt.sub(r'\1.0', text)
#debug.printf('After uninting: %r' % text)
def handleMatch(m):
s = m.group(1)
if s.startswith('0x'):
i = int(s, 16)
elif s.startswith('0') and '.' not in s:
try:
i = int(s, 8)
except ValueError:
i = int(s)
else:
i = float(s)
return str(complex(i))
text = self._mathRe.sub(handleMatch, text)
try:
x = complex(eval(text, self._mathEnv, self._mathEnv))
irc.reply(msg, self._complexToString(x))
@ -564,7 +575,7 @@ class FunCommands(callbacks.Privmsg):
stack = []
for arg in args:
try:
stack.append(float(arg))
stack.append(complex(arg))
except ValueError: # Not a float.
if arg in self._mathEnv:
f = self._mathEnv[arg]

4
test/test_FunCommands.py
View File

@ -57,6 +57,10 @@ class FunCommandsTest(PluginTestCase, PluginDocumentation):
def testCalc(self):
self.assertResponse('calc 5*0.06', str(5*0.06))
self.assertResponse('calc 2.0-7.0', str(2-7))
self.assertResponse('calc (-1)**.5', 'i')
self.assertResponse('calc e**(i*pi)+1', '0')
self.assertResponse('calc (-5)**.5', '2.2360679775i')
self.assertResponse('calc -((-5)**.5)', '-2.2360679775i')
def testChr(self):
for i in range(256):