WebSVN – HelenOS – Diff – /trunk/tools/xtui.py


#
# Copyright (c) 2009 Martin Decky
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
#
# - Redistributions of source code must retain the above copyright
#   notice, this list of conditions and the following disclaimer.
# - Redistributions in binary form must reproduce the above copyright
#   notice, this list of conditions and the following disclaimer in the
#   documentation and/or other materials provided with the distribution.
# - The name of the author may not be used to endorse or promote products
#   derived from this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
# OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
# IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
# NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
# THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#
"""
Text User Interface wrapper
"""
 
import sys
import os
 
def call_dlg(dlgcmd, *args, **kw):
    "Wrapper for calling 'dialog' program"
   
    indesc, outdesc = os.pipe()
    pid = os.fork()
    if (not pid):
        os.close(2)
        os.dup(outdesc)
        os.close(indesc)
       
        dlgargs = [dlgcmd]
        for key, val in kw.items():
            dlgargs.append('--' + key)
            dlgargs.append(val)
       
        dlgargs += args
        os.execlp(dlgcmd, *dlgargs)
   
    os.close(outdesc)
   
    try:
        errout = os.fdopen(indesc, 'r')
        data = errout.read()
        errout.close()
        pid, status = os.wait()
    except:
        # Reset terminal
        os.system('reset')
        raise
   
    if (not os.WIFEXITED(status)):
        # Reset terminal
        os.system('reset')
        raise EOFError
   
    status = os.WEXITSTATUS(status)
    if (status == 255):
        raise EOFError
   
    return (status, data)
 
try:
    import snack
    newt = True
    dialog = False
except ImportError:
    newt = False
   
    dlgcmd = os.environ.get('DIALOG', 'dialog')
    if (call_dlg(dlgcmd, '--print-maxsize')[0] != 0):
        dialog = False
    else:
        dialog = True
 
width_extra = 13
height_extra = 11
 
def width_fix(screen, width):
    "Correct width to screen size"
   
    if (width + width_extra > screen.width):
        width = screen.width - width_extra
   
    if (width <= 0):
        width = screen.width
   
    return width
 
def height_fix(screen, height):
    "Correct height to screen size"
   
    if (height + height_extra > screen.height):
        height = screen.height - height_extra
   
    if (height <= 0):
        height = screen.height
   
    return height
 
def screen_init():
    "Initialize the screen"
   
    if (newt):
        return snack.SnackScreen()
   
    return None
 
def screen_done(screen):
    "Cleanup the screen"
   
    if (newt):
        screen.finish()
 
def choice_window(screen, title, text, options, position):
    "Create options menu"
   
    maxopt = 0
    for option in options:
        length = len(option)
        if (length > maxopt):
            maxopt = length
   
    width = maxopt
    height = len(options)
   
    if (newt):
        width = width_fix(screen, width + width_extra)
        height = height_fix(screen, height)
       
        if (height > 3):
            large = True
        else:
            large = False
       
        buttonbar = snack.ButtonBar(screen, ('Done', 'Cancel'))
        textbox = snack.TextboxReflowed(width, text)
        listbox = snack.Listbox(height, scroll = large, returnExit = 1)
       
        cnt = 0
        for option in options:
            listbox.append(option, cnt)
            cnt += 1
       
        if (position != None):
            listbox.setCurrent(position)
       
        grid = snack.GridForm(screen, title, 1, 3)
        grid.add(textbox, 0, 0)
        grid.add(listbox, 0, 1, padding = (0, 1, 0, 1))
        grid.add(buttonbar, 0, 2, growx = 1)
       
        retval = grid.runOnce()
       
        return (buttonbar.buttonPressed(retval), listbox.current())
    elif (dialog):
        args = []
        cnt = 0
        for option in options:
            args.append(str(cnt + 1))
            args.append(option)
           
            cnt += 1
       
        kw = {}
        if (position != None):
            kw['default-item'] = str(position + 1)
       
        status, data = call_dlg(dlgcmd, '--title', title, '--extra-button', '--extra-label', 'Done', '--menu', text, str(height + height_extra), str(width + width_extra), str(cnt), *args, **kw)
       
        if (status == 1):
            return ('cancel', None)
       
        try:
            choice = int(data) - 1
        except ValueError:
            return ('cancel', None)
       
        if (status == 0):
            return (None, choice)
       
        return ('done', choice)
   
    sys.stdout.write("\n *** %s *** \n%s\n\n" % (title, text))
   
    maxcnt = len(str(len(options)))
    cnt = 0
    for option in options:
        sys.stdout.write("%*s. %s\n" % (maxcnt, cnt + 1, option))
        cnt += 1
   
    sys.stdout.write("\n%*s. Done\n" % (maxcnt, '0'))
    sys.stdout.write("%*s. Quit\n\n" % (maxcnt, 'q'))
   
    while True:
        if (position != None):
            sys.stdout.write("Selection[%s]: " % str(position + 1))
        else:
            sys.stdout.write("Selection: ")
        inp = sys.stdin.readline()
       
        if (not inp):
            raise EOFError
       
        if (not inp.strip()):
            if (position != None):
                return (None, position)
            continue
       
        if (inp.strip() == 'q'):
            return ('cancel', None)
       
        try:
            choice = int(inp.strip())
        except ValueError:
            continue
       
        if (choice == 0):
            return ('done', 0)
       
        if (choice < 1) or (choice > len(options)):
            continue
       
        return (None, choice - 1)
 
def error_dialog(screen, title, msg):
    "Print error dialog"
   
    width = len(msg)
   
    if (newt):
        width = width_fix(screen, width)
       
        buttonbar = snack.ButtonBar(screen, ['Ok'])
        textbox = snack.TextboxReflowed(width, msg)
       
        grid = snack.GridForm(screen, title, 1, 2)
        grid.add(textbox, 0, 0, padding = (0, 0, 0, 1))
        grid.add(buttonbar, 0, 1, growx = 1)
        grid.runOnce()
    elif (dialog):
        call_dlg(dlgcmd, '--title', title, '--msgbox', msg, '6', str(width + width_extra))
   
    sys.stdout.write("\n%s: %s\n" % (title, msg))
 

Rev 3804	Rev 3805
1	#	1	#
2	# Copyright (c) 2009 Martin Decky	2	# Copyright (c) 2009 Martin Decky
3	# All rights reserved.	3	# All rights reserved.
4	#	4	#
5	# Redistribution and use in source and binary forms, with or without	5	# Redistribution and use in source and binary forms, with or without
6	# modification, are permitted provided that the following conditions	6	# modification, are permitted provided that the following conditions
7	# are met:	7	# are met:
8	#	8	#
9	# - Redistributions of source code must retain the above copyright	9	# - Redistributions of source code must retain the above copyright
10	# notice, this list of conditions and the following disclaimer.	10	# notice, this list of conditions and the following disclaimer.
11	# - Redistributions in binary form must reproduce the above copyright	11	# - Redistributions in binary form must reproduce the above copyright
12	# notice, this list of conditions and the following disclaimer in the	12	# notice, this list of conditions and the following disclaimer in the
13	# documentation and/or other materials provided with the distribution.	13	# documentation and/or other materials provided with the distribution.
14	# - The name of the author may not be used to endorse or promote products	14	# - The name of the author may not be used to endorse or promote products
15	# derived from this software without specific prior written permission.	15	# derived from this software without specific prior written permission.
16	#	16	#
17	# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR	17	# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
18	# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES	18	# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19	# OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.	19	# OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20	# IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,	20	# IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
21	# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT	21	# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22	# NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,	22	# NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23	# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY	23	# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24	# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT	24	# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25	# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF	25	# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26	# THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.	26	# THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27	#	27	#
28	"""	28	"""
29	Text User Interface wrapper	29	Text User Interface wrapper
30	"""	30	"""
31		31
32	import sys	32	import sys
33	import os	33	import os
34		34
35	def call_dlg(dlgcmd, args, *kw):	35	def call_dlg(dlgcmd, args, *kw):
36	"Wrapper for calling 'dialog' program"	36	"Wrapper for calling 'dialog' program"
37		37
38	indesc, outdesc = os.pipe()	38	indesc, outdesc = os.pipe()
39	pid = os.fork()	39	pid = os.fork()
40	if (not pid):	40	if (not pid):
41	os.close(2)	41	os.close(2)
42	os.dup(outdesc)	42	os.dup(outdesc)
43	os.close(indesc)	43	os.close(indesc)
44		44
45	dlgargs = [dlgcmd]	45	dlgargs = [dlgcmd]
46	for key, val in kw.items():	46	for key, val in kw.items():
47	dlgargs.append('--' + key)	47	dlgargs.append('--' + key)
48	dlgargs.append(val)	48	dlgargs.append(val)
49		49
50	dlgargs += args	50	dlgargs += args
51	os.execlp(dlgcmd, *dlgargs)	51	os.execlp(dlgcmd, *dlgargs)
52		52
53	os.close(outdesc)	53	os.close(outdesc)
54		54
55	try:	55	try:
56	errout = os.fdopen(indesc, 'r')	56	errout = os.fdopen(indesc, 'r')
57	data = errout.read()	57	data = errout.read()
58	errout.close()	58	errout.close()
59	pid, status = os.wait()	59	pid, status = os.wait()
60	except:	60	except:
61	# Reset terminal	61	# Reset terminal
62	os.system('reset')	62	os.system('reset')
63	raise	63	raise
64		64
65	if (not os.WIFEXITED(status)):	65	if (not os.WIFEXITED(status)):
66	# Reset terminal	66	# Reset terminal
67	os.system('reset')	67	os.system('reset')
68	raise EOFError	68	raise EOFError
69		69
70	status = os.WEXITSTATUS(status)	70	status = os.WEXITSTATUS(status)
71	if (status == 255):	71	if (status == 255):
72	raise EOFError	72	raise EOFError
73		73
74	return (status, data)	74	return (status, data)
75		75
76	try:	76	try:
77	import snack	77	import snack
78	newt = True	78	newt = True
79	dialog = False	79	dialog = False
80	except ImportError:	80	except ImportError:
81	newt = False	81	newt = False
82		82
83	dlgcmd = os.environ.get('DIALOG', 'dialog')	83	dlgcmd = os.environ.get('DIALOG', 'dialog')
84	if (call_dlg(dlgcmd, '--print-maxsize')[0] != 0):	84	if (call_dlg(dlgcmd, '--print-maxsize')[0] != 0):
85	dialog = False	85	dialog = False
86	else:	86	else:
87	dialog = True	87	dialog = True
88		88
89	width_extra = 13	89	width_extra = 13
90	height_extra = 11	90	height_extra = 11
91		91
92	def width_fix(screen, width):	92	def width_fix(screen, width):
93	"Correct width to screen size"	93	"Correct width to screen size"
94		94
95	if (width + width_extra > screen.width):	95	if (width + width_extra > screen.width):
96	width = screen.width - width_extra	96	width = screen.width - width_extra
97		97
98	if (width <= 0):	98	if (width <= 0):
99	width = screen.width	99	width = screen.width
100		100
101	return width	101	return width
102		102
103	def height_fix(screen, height):	103	def height_fix(screen, height):
104	"Correct height to screen size"	104	"Correct height to screen size"
105		105
106	if (height + height_extra > screen.height):	106	if (height + height_extra > screen.height):
107	height = screen.height - height_extra	107	height = screen.height - height_extra
108		108
109	if (height <= 0):	109	if (height <= 0):
110	height = screen.height	110	height = screen.height
111		111
112	return height	112	return height
113		113
114	def screen_init():	114	def screen_init():
115	"Initialize the screen"	115	"Initialize the screen"
116		116
117	if (newt):	117	if (newt):
118	return snack.SnackScreen()	118	return snack.SnackScreen()
119		119
120	return None	120	return None
121		121
122	def screen_done(screen):	122	def screen_done(screen):
123	"Cleanup the screen"	123	"Cleanup the screen"
124		124
125	if (newt):	125	if (newt):
126	screen.finish()	126	screen.finish()
127		127
128	def choice_window(screen, title, text, options, position):	128	def choice_window(screen, title, text, options, position):
129	"Create options menu"	129	"Create options menu"
130		130
131	maxopt = 0	131	maxopt = 0
132	for option in options:	132	for option in options:
133	length = len(option)	133	length = len(option)
134	if (length > maxopt):	134	if (length > maxopt):
135	maxopt = length	135	maxopt = length
136		136
137	width = maxopt	137	width = maxopt
138	height = len(options)	138	height = len(options)
139		139
140	if (newt):	140	if (newt):
141	width = width_fix(screen, width + width_extra)	141	width = width_fix(screen, width + width_extra)
142	height = height_fix(screen, height)	142	height = height_fix(screen, height)
143		143
-		144	if (height > 3):
-		145	large = True
-		146	else:
-		147	large = False
-		148
144	buttonbar = snack.ButtonBar(screen, ('Done', 'Cancel'))	149	buttonbar = snack.ButtonBar(screen, ('Done', 'Cancel'))
145	textbox = snack.TextboxReflowed(width, text)	150	textbox = snack.TextboxReflowed(width, text)
146	listbox = snack.Listbox(height, scroll = True, returnExit = 1)	151	listbox = snack.Listbox(height, scroll = large, returnExit = 1)
147		152
148	cnt = 0	153	cnt = 0
149	for option in options:	154	for option in options:
150	listbox.append(option, cnt)	155	listbox.append(option, cnt)
151	cnt += 1	156	cnt += 1
152		157
153	if (position != None):	158	if (position != None):
154	listbox.setCurrent(position)	159	listbox.setCurrent(position)
155		160
156	grid = snack.GridForm(screen, title, 1, 3)	161	grid = snack.GridForm(screen, title, 1, 3)
157	grid.add(textbox, 0, 0)	162	grid.add(textbox, 0, 0)
158	grid.add(listbox, 0, 1, padding = (0, 1, 0, 1))	163	grid.add(listbox, 0, 1, padding = (0, 1, 0, 1))
159	grid.add(buttonbar, 0, 2, growx = 1)	164	grid.add(buttonbar, 0, 2, growx = 1)
160		165
161	retval = grid.runOnce()	166	retval = grid.runOnce()
162		167
163	return (buttonbar.buttonPressed(retval), listbox.current())	168	return (buttonbar.buttonPressed(retval), listbox.current())
164	elif (dialog):	169	elif (dialog):
165	args = []	170	args = []
166	cnt = 0	171	cnt = 0
167	for option in options:	172	for option in options:
168	args.append(str(cnt + 1))	173	args.append(str(cnt + 1))
169	args.append(option)	174	args.append(option)
170		175
171	cnt += 1	176	cnt += 1
172		177
173	kw = {}	178	kw = {}
174	if (position != None):	179	if (position != None):
175	kw['default-item'] = str(position + 1)	180	kw['default-item'] = str(position + 1)
176		181
177	status, data = call_dlg(dlgcmd, '--title', title, '--extra-button', '--extra-label', 'Done', '--menu', text, str(height + height_extra), str(width + width_extra), str(cnt), args, *kw)	182	status, data = call_dlg(dlgcmd, '--title', title, '--extra-button', '--extra-label', 'Done', '--menu', text, str(height + height_extra), str(width + width_extra), str(cnt), args, *kw)
178		183
179	if (status == 1):	184	if (status == 1):
180	return ('cancel', None)	185	return ('cancel', None)
181		186
182	try:	187	try:
183	choice = int(data) - 1	188	choice = int(data) - 1
184	except ValueError:	189	except ValueError:
185	return ('cancel', None)	190	return ('cancel', None)
186		191
187	if (status == 0):	192	if (status == 0):
188	return (None, choice)	193	return (None, choice)
189		194
190	return ('done', choice)	195	return ('done', choice)
191		196
192	sys.stdout.write("\n * %s * \n%s\n\n" % (title, text))	197	sys.stdout.write("\n * %s * \n%s\n\n" % (title, text))
193		198
194	maxcnt = len(str(len(options)))	199	maxcnt = len(str(len(options)))
195	cnt = 0	200	cnt = 0
196	for option in options:	201	for option in options:
197	sys.stdout.write("%*s. %s\n" % (maxcnt, cnt + 1, option))	202	sys.stdout.write("%*s. %s\n" % (maxcnt, cnt + 1, option))
198	cnt += 1	203	cnt += 1
199		204
200	sys.stdout.write("\n%*s. Done\n" % (maxcnt, '0'))	205	sys.stdout.write("\n%*s. Done\n" % (maxcnt, '0'))
201	sys.stdout.write("%*s. Quit\n\n" % (maxcnt, 'q'))	206	sys.stdout.write("%*s. Quit\n\n" % (maxcnt, 'q'))
202		207
203	while True:	208	while True:
204	if (position != None):	209	if (position != None):
205	sys.stdout.write("Selection[%s]: " % str(position + 1))	210	sys.stdout.write("Selection[%s]: " % str(position + 1))
206	else:	211	else:
207	sys.stdout.write("Selection: ")	212	sys.stdout.write("Selection: ")
208	inp = sys.stdin.readline()	213	inp = sys.stdin.readline()
209		214
210	if (not inp):	215	if (not inp):
211	raise EOFError	216	raise EOFError
212		217
213	if (not inp.strip()):	218	if (not inp.strip()):
214	if (position != None):	219	if (position != None):
215	return (None, position)	220	return (None, position)
216	continue	221	continue
217		222
218	if (inp.strip() == 'q'):	223	if (inp.strip() == 'q'):
219	return ('cancel', None)	224	return ('cancel', None)
220		225
221	try:	226	try:
222	choice = int(inp.strip())	227	choice = int(inp.strip())
223	except ValueError:	228	except ValueError:
224	continue	229	continue
225		230
226	if (choice == 0):	231	if (choice == 0):
227	return ('done', 0)	232	return ('done', 0)
228		233
229	if (choice < 1) or (choice > len(options)):	234	if (choice < 1) or (choice > len(options)):
230	continue	235	continue
231		236
232	return (None, choice - 1)	237	return (None, choice - 1)
233		238
234	def error_dialog(screen, title, msg):	239	def error_dialog(screen, title, msg):
235	"Print error dialog"	240	"Print error dialog"
236		241
237	width = len(msg)	242	width = len(msg)
238		243
239	if (newt):	244	if (newt):
240	width = width_fix(screen, width)	245	width = width_fix(screen, width)
241		246
242	buttonbar = snack.ButtonBar(screen, ['Ok'])	247	buttonbar = snack.ButtonBar(screen, ['Ok'])
243	textbox = snack.TextboxReflowed(width, msg)	248	textbox = snack.TextboxReflowed(width, msg)
244		249
245	grid = snack.GridForm(screen, title, 1, 2)	250	grid = snack.GridForm(screen, title, 1, 2)
246	grid.add(textbox, 0, 0, padding = (0, 0, 0, 1))	251	grid.add(textbox, 0, 0, padding = (0, 0, 0, 1))
247	grid.add(buttonbar, 0, 1, growx = 1)	252	grid.add(buttonbar, 0, 1, growx = 1)
248	grid.runOnce()	253	grid.runOnce()
249	elif (dialog):	254	elif (dialog):
250	call_dlg(dlgcmd, '--title', title, '--msgbox', msg, '6', str(width + width_extra))	255	call_dlg(dlgcmd, '--title', title, '--msgbox', msg, '6', str(width + width_extra))
251		256
252	sys.stdout.write("\n%s: %s\n" % (title, msg))	257	sys.stdout.write("\n%s: %s\n" % (title, msg))
253		258

Subversion Repositories HelenOS

(root)/trunk/tools/xtui.py – Rev 3804 → 3805