command-line-thesaurus/google_define.py at master · empireshades/command-line-thesaurus · GitHub

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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
import argparse
import requests
from bs4 import BeautifulSoup

def get_soup(searchword):
    base = "https://www.google.com/search"
    params = { 'q' : 'define%s' % searchword }
    headers = { 'User-Agent' : 'Firefox 42.3' }
    req = requests.get(base, params=params, headers=headers)
    soup = BeautifulSoup(req.text, "html5lib")
    return soup

class definition():
    def __init__(self, soup):
        # Look for definition block
        try: rawchunk = soup.find('div', class_='yc7KLc')
        except: pass

        if rawchunk:
            # See if there was a suggested word after mis-spelled input
            try:
                if soup.find('a', class_='gL9Hy').text.split(' ')[-1]:
                    self.suggested = soup.find('a', class_='gL9Hy').text.split(' ')[-1]
            except: pass

            try: self.syllabic = rawchunk.find('span', attrs={'data-dobid':'hdw'}).text
            except: self.syllabic = ""

            try: self.phonetic = rawchunk.find('span', class_='lr_dct_ph').text
            except: self.phonetic = ""

            try: self.forms    = rawchunk.find('div', class_='xpdxpnd vk_gy').text
            except: pass

            try: self.fulldefs = [ i.text for i in rawchunk.find_all('div', attrs={'data-dobid':'dfn'}) ]
            except: pass

            try: self.sentence = rawchunk.find('div', class_='QIclbb XpoqFe').find('div', class_='vk_gy').text
            except: pass

            try: thesrus       = rawchunk.find_all('table', class_='vk_tbl vk_gy')
            except: pass

            try: self.synolist = [ i.text for i in thesrus[0].find_all('a')]
            except: pass

            try: self.antolist = [ i.text for i in thesrus[1].find_all('a')]
            except: pass

    def __repr__(self):
        return __name__

def main():
    class bcolors:
        HEADER = '\033[95m'
        OKBLUE = '\033[94m'
        OKGREEN = '\033[92m'
        WARNING = '\033[93m'
        FAIL = '\033[91m'
        ENDC = '\033[0m'

    parser = argparse.ArgumentParser()
    parser.add_argument("search", nargs='+', help="type the word you want to search here")
    args = parser.parse_args()
    searchword = " ".join(args.search)
    defi = definition(get_soup(searchword))

    try:
        if defi.syllabic:
            if hasattr(defi, "suggested") and defi.suggested != searchword:
                head = "%s (suggested) . %s . %s" % (defi.suggested, defi.syllabic, defi.phonetic)
            else:
                head = "%s . %s . %s" % (searchword, defi.syllabic, defi.phonetic)

            print(bcolors.HEADER + head + bcolors.ENDC)
            print(bcolors.OKGREEN + " - " + "\n - ".join(defi.fulldefs) + bcolors.ENDC)

            if hasattr(defi, "forms"):
                print("\n" + bcolors.HEADER + "forms:" + bcolors.ENDC)
                print(bcolors.OKGREEN + " - " + "\n -".join(defi.forms.split(';')) + bcolors.ENDC)
            if hasattr(defi, "sentence"):
                print("\n" + bcolors.HEADER + "usage:" + bcolors.ENDC)
                print(bcolors.OKGREEN + " " + defi.sentence + bcolors.ENDC)
            if hasattr(defi, "synolist"):
                print("\n" + bcolors.HEADER + "synonyms:" + bcolors.ENDC)
                print(bcolors.OKGREEN + " " + ", ".join(defi.synolist) + bcolors.ENDC)
            if hasattr(defi, "antolist"):
                print("\n" + bcolors.HEADER + "antonyms:" + bcolors.ENDC)
                print(bcolors.OKGREEN + " " + ", ".join(defi.antolist) + bcolors.ENDC)
            print()
    except:
        print('No definition found')


if __name__ == "__main__":
    main()