Переглянути джерело

adding a calculator module, based on frink

tags/v0.1
Ben Kurtovic 13 роки тому
джерело
коміт
795c03a5ee
1 змінених файлів з 71 додано та 0 видалено
  1. +71
    -0
      irc/commands/calc.py

+ 71
- 0
irc/commands/calc.py Переглянути файл

@@ -0,0 +1,71 @@
# -*- coding: utf-8 -*-

# A somewhat advanced calculator: http://futureboy.us/fsp/frink.fsp.

import re
import urllib

from irc.base_command import BaseCommand

class Calc(BaseCommand):
def get_hooks(self):
return ["msg"]

def get_help(self, command):
return "A somewhat advanced calculator: see http://futureboy.us/fsp/frink.fsp for details."

def check(self, data):
if data.is_command and data.command == "calc":
return True
return False

def process(self, data):
if not data.args:
self.connection.reply(data, "What do you want me to calculate?")
return

query = ' '.join(data.args)
query = self.cleanup(query)

url = "http://futureboy.us/fsp/frink.fsp?fromVal=%s" % urllib.quote(query)
result = urllib.urlopen(url).read()

r_result = re.compile(r'(?i)<A NAME=results>(.*?)</A>')
r_tag = re.compile(r'<\S+.*?>')

match = r_result.search(result)
if not match:
self.connection.reply(data, "Calculation error.")
return

result = match.group(1)
result = r_tag.sub("", result) # strip span.warning tags
result = result.replace("&gt;", ">")
result = result.replace("(undefined symbol)", "(?) ")
result = result.strip()

if not result:
result = '?'
elif " in " in query:
result += " " + query.split(" in ", 1)[1]

res = "%s = %s" % (query, result)
self.connection.reply(data, res)

def cleanup(self, query):
fixes = [
(' in ', ' -> '),
(' over ', ' / '),
(u'£', 'GBP '),
(u'€', 'EUR '),
('\$', 'USD '),
(r'\bKB\b', 'kilobytes'),
(r'\bMB\b', 'megabytes'),
(r'\bGB\b', 'kilobytes'),
('kbps', '(kilobits / second)'),
('mbps', '(megabits / second)')
]

for original, fix in fixes:
query = re.sub(original, fix, query)
return query.strip()

Завантаження…
Відмінити
Зберегти