import re from util import hook, http @hook.command('wa') @hook.command def wolframalpha(inp): ".wa/.wolframalpha -- scrapes Wolfram Alpha's" \ "results for " 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 image in pod.xpath('div/div[@class="output"]/img'): alt = image.attrib['alt'].strip() alt = alt.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