ChCh/CloudBot
Archived
11
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

make wolframalpha and urban use apis instead of scraping - commit from Skybot

Browse source
This commit is contained in:
Luke Rogers 2012-02-16 07:01:41 +13:00
parent b37165b8d2
commit f75aeea2a9
2 changed files with 23 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

37
plugins/wolframalpha.py
View file

@ -5,34 +5,31 @@ from util import hook, http
@hook.command('wa')
@hook.command
def wolframalpha(inp):
".wa/.wolframalpha <query> -- scrapes Wolfram Alpha's" \
" results for <query>"
def wolframalpha(inp, bot=None):
".wa/.wolframalpha <query> -- computes <query> using Wolfram Alpha"
url = "http://www.wolframalpha.com/input/?asynchronous=false"
api_key = bot.config.get("api_keys", {}).get("wolframalpha", None)
if api_key is None:
return "error: no api key set"
h = http.get_html(url, i=inp)
url = 'http://api.wolframalpha.com/v2/query?format=plaintext'
pods = h.xpath("//div[@class='pod ']")
result = http.get_xml(url, input=inp, appid=api_key)
pod_texts = []
for pod in pods:
heading = pod.find('h2')
if heading is not None:
heading = heading.text_content().strip()
if heading.startswith('Input'):
continue
else:
for pod in result.xpath("//pod"):
title = pod.attrib['title']
if pod.attrib['id'] == 'Input':
continue
results = []
for alt in pod.xpath('div/div[@class="output pnt"]/img/@alt'):
alt = alt.strip().replace('\\n', '; ')
alt = re.sub(r'\s+', ' ', alt)
if alt:
results.append(alt)
for subpod in pod.xpath('subpod/plaintext/text()'):
subpod = subpod.strip().replace('\\n', '; ')
subpod = re.sub(r'\s+', ' ', subpod)
if subpod:
results.append(subpod)
if results:
pod_texts.append(heading + ' ' + '|'.join(results))
pod_texts.append(title + ': ' + '|'.join(results))
ret = '. '.join(pod_texts)
@ -51,6 +48,6 @@ def wolframalpha(inp):
ret = re.sub(r'\W+$', '', ret) + '...'
if not ret:
return 'no result'
return 'no results'
return ret