First :D
This commit is contained in:
commit
37588421f3
100 changed files with 22673 additions and 0 deletions
56
plugins/wolframalpha.py
Normal file
56
plugins/wolframalpha.py
Normal file
|
@ -0,0 +1,56 @@
|
|||
import re
|
||||
|
||||
from util import hook, http
|
||||
|
||||
|
||||
@hook.command('wa')
|
||||
@hook.command
|
||||
def wolframalpha(inp):
|
||||
".wa/.wolframalpha <query> -- scrapes Wolfram Alpha's" \
|
||||
" results for <query>"
|
||||
|
||||
url = "http://www.wolframalpha.com/input/?asynchronous=false"
|
||||
|
||||
h = http.get_html(url, i=inp)
|
||||
|
||||
pods = h.xpath("//div[@class='pod ']")
|
||||
|
||||
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:
|
||||
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)
|
||||
if results:
|
||||
pod_texts.append(heading + ' ' + '|'.join(results))
|
||||
|
||||
ret = '. '.join(pod_texts)
|
||||
|
||||
if not pod_texts:
|
||||
return 'no results'
|
||||
|
||||
ret = re.sub(r'\\(.)', r'\1', ret)
|
||||
|
||||
def unicode_sub(match):
|
||||
return unichr(int(match.group(1), 16))
|
||||
|
||||
ret = re.sub(r'\\:([0-9a-z]{4})', unicode_sub, ret)
|
||||
|
||||
if len(ret) > 430:
|
||||
ret = ret[:ret.rfind(' ', 0, 430)]
|
||||
ret = re.sub(r'\W+$', '', ret) + '...'
|
||||
|
||||
if not ret:
|
||||
return 'no result'
|
||||
|
||||
return ret
|
Reference in a new issue