summaryrefslogtreecommitdiff
path: root/wolfram/lat.py
blob: 92f31d13d27985c83029129ad619a51768344c4e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
#!/usr/bin/env python
# -*- coding: utf-8 -*-

import requests
from urllib import quote_plus
import xml.etree.ElementTree as ET

appid = 'U6E8UV-AP3QJAT6RK'

def get_plaintext_query(latex):    
    r = requests.get('http://api.wolframalpha.com/v2/query?input=%s&appid=%s' % (quote_plus(latex), appid))
    root = ET.fromstring(r.text.encode('utf8'))

    for pod in root:
        if pod.attrib.get('title', '') in ['Decimal approximation', 'Definite integral']:
            subpod = pod.find('subpod')
            result = subpod.find('plaintext').text

            if pod.attrib.get('title', '') == 'Definite integral':
                return result.split('~~')[1]
            else:
                return result

if __name__ == '__main__':
    from sys import stdin
    print get_plaintext_query(stdin.read())