mbrun/define.py at master · mbfraga/mbrun · 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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
#!/bin/env python3

from sys import argv, exit
from subprocess import Popen, PIPE
from helpers import mbrofi

# user variables

# application variables
mbconfig = mbrofi.parse_config()
dictionary_file = 'dictd_www.dict.org_gcide'
thesaurus_file = 'Moby Thesaurus II'
BIND_TOGGLE = 'alt-t'
BIND_OPENWEB = 'alt-o'
rofi_lines = None
script_ident = 'define'
if script_ident in mbconfig:
    dictionary_file = mbconfig[script_ident].get("dictionary_file",
                                        fallback='dictd_www.dict.org_gcide')
    thesaurus_file = mbconfig[script_ident].get("thesaurus_file",
                                        fallback='Moby Thesaurus II')
    BIND_TOGGLE = mbconfig[script_ident].get("bind_toggle", fallback="alt-t")
    BIND_OPENWEB = mbconfig[script_ident].get("bind_openweb", fallback="alt-o")
    rofi_lines = mbconfig[script_ident].get("rofi_lines", fallback=None)
    BROWSER = mbconfig['mbmain'].get('browser', fallback='firefox')

if rofi_lines == "None":
    rofi_lines = None

# launcher variables
prompt = "define:"
filt = ""

# run correct launcher with prompt and help message
launcher_args = {}
launcher_args['prompt'] = prompt
launcher_args['filter'] = filt
launcher_args['format'] = 'f'
launcher_args['bindings'] = [BIND_TOGGLE, BIND_OPENWEB]


def define(word, thesaurus=False):
    """Send word to sdcv dictionary and return results"""
    if thesaurus:
        proc = Popen(['sdcv', '-n', '-u', thesaurus_file, word], stdout=PIPE)
    else:
        proc = Popen(['sdcv', '-n', '-u', dictionary_file, word], stdout=PIPE)
    ans = proc.stdout.read().decode('utf-8')
    exit_code = proc.wait()
    if exit_code == 1:
        print("sdcv failed...exiting")
        exit(1)
    else:
        entries = []
        templist = ans.strip().split('\n')
        for entry in templist:
            if entry.startswith('-'):
                continue
            entries.append(entry)
        return(entries)


def main(launcher_args, query=None, thesaurus=False, rofi_lines=None):
    """Main function."""
    while True:
        if thesaurus:
            launcher_args['prompt'] = "thesaurus:"
            launcher_args['mesg'] = "Press '" + BIND_TOGGLE
            launcher_args['mesg'] += "' to show definition,"
            launcher_args['mesg'] += " '" + BIND_OPENWEB + "' to open in web."
        else:
            launcher_args['prompt'] = "define"
            launcher_args['mesg'] = "Press '" + BIND_TOGGLE
            launcher_args['mesg'] += "' to show thesaurus,"
            launcher_args['mesg'] += " '" + BIND_OPENWEB + "' to open in web."
        if query is None:
                answer, exit_code = mbrofi.rofi([], launcher_args)
        else:
            if rofi_lines is None:
                theme_args = ["-theme-str", "#window {width: 38em;}"]
                theme_args.append("-i")
                theme_args.append("-markup-rows")
                answer, exit_code = mbrofi.rofi(define(query, thesaurus),
                                                launcher_args, theme_args)
            else:
                theme_args = ["-theme-str",
                              "#window {width: 38em;}" +
                              " #window.mainbox.listview { lines: " +
                              rofi_lines + ";}"]
                theme_args.append("-i")
                theme_args.append("-markup-rows")
                answer, exit_code = mbrofi.rofi(define(query, thesaurus),
                                                launcher_args, theme_args)
        if (exit_code == 1):
            break
        if not answer:
            if exit_code == 10:
                thesaurus = not(thesaurus)
                continue
            if exit_code == 11:
                if query:
                    if thesaurus:
                        squery="http://www.thesaurus.com/browse/" + query
                    else:
                        squery="http://www.dictionary.com/browse/" + query
                    Popen([BROWSER, squery]).communicate()
                    break
            else:
                break
        elif (exit_code == 0):
            filt = answer.strip()
            query = filt
        elif (exit_code == 10):
            thesaurus = not(thesaurus)
            filt = answer.strip()
            query = filt
        elif (exit_code == 11):
            filt = answer.strip()
            if not filt:
                continue
            if thesaurus:
                squery="http://www.thesaurus.com/browse/" + filt
            else:
                squery="http://www.dictionary.com/browse/" + filt
            Popen([BROWSER, squery]).communicate()
            break
        else:
            break


if __name__ == '__main__':
    if (len(argv) > 1):
        if argv[1] == '-t':
            thesaurus = True
            query = ''.join(str(e) + ' ' for e in argv[2:]).strip()
        else:
            thesaurus = False
            query = ''.join(str(e) + ' ' for e in argv[1:]).strip()
        main(launcher_args, query, thesaurus, rofi_lines=rofi_lines)
    else:
        query = None
        main(launcher_args, query, rofi_lines=rofi_lines)