MINI Sh3ll
# snack.py: maps C extension module _snack to proper python types in module
# snack.
# The first section is a very literal mapping.
# The second section contains convenience classes that amalgamate
# the literal classes and make them more object-oriented.
"""
This module provides the NEWT Windowing toolkit API for Python
This is a lightweight text-mode windowing library, based on slang.
Classes:
- Widget
- Button
- CompactButton
- Checkbox
- SingleRadioButton
- Listbox
- Textbox
- TextboxReflowed
- Label
- Scale
- Entry
- Form
- Grid
- SnackScreen
- RadioGroup
- RadioBar
- ButtonBar
- GridFormHelp
- GridForm
- CheckboxTree
- Clistbox
Functions:
- ListboxChoiceWindow
- ButtonChoiceWindow
- EntryWindow
"""
from __future__ import absolute_import, print_function, unicode_literals
import _snack
import string
import sys
from _snack import FLAG_DISABLED, FLAGS_SET, FLAGS_RESET, FLAGS_TOGGLE, FD_READ, FD_WRITE, FD_EXCEPT
LEFT = (-1, 0)
DOWN = (-1, -1)
CENTER = (0, 0)
UP = (1, 1)
RIGHT = (1, 0)
snackArgs = {"append":-1}
class Widget:
"""Base class for NEWT toolkit - Do not use directly
methods:
- Widget(self)
- setCallback(self, obj, data = None) :
The callback for when object activated.
data is passed to obj.
"""
def setCallback(self, obj, data = None):
if data:
self.w.setCallback(obj, data)
else:
self.w.setCallback(obj)
def __init__(self):
raise NotImplementedError
class Button(Widget):
"""Basic button class, takes button text as parameter
method:
- Button(self, text): returns a button
"""
def __init__(self, text):
self.w = _snack.button(text)
class CompactButton(Widget):
"""Compact Button class (less frilly button decoration).
methods:
- CompactButton(self,text) : create button, with text.
"""
def __init__(self, text):
self.w = _snack.compactbutton(text)
class Checkbox(Widget):
"""A checkbox.
methods:
- Checkbox(self, text, isOn = 0) : text, and boolean as to default value
- setValue(self) : set value
- value(self, value) : return checkbox value
- selected(self) : returns boolean
- setFlags(self, flag, sense) : set flags
flags: FLAG_DISABLED, FLAGS_SET, FLAGS_RESET
"""
def value(self):
return self.w.checkboxValue
def selected(self):
return self.w.checkboxValue != 0
def setFlags (self, flag, sense):
return self.w.checkboxSetFlags(flag, sense)
def setValue (self, value):
return self.w.checkboxSetValue(value)
def __init__(self, text, isOn = 0):
self.w = _snack.checkbox(text, isOn)
class SingleRadioButton(Widget):
"""Single Radio Button.
methods:
- SingleRadioButton(text, group, isOn = 0) : create button
- selected(self) : returns bool, whether or not is selected.
"""
def selected(self):
return self.w.key == self.w.radioValue;
def __init__(self, text, group, isOn = 0):
if group:
self.w = _snack.radiobutton(text, group.w, isOn)
else:
self.w = _snack.radiobutton(text, None, isOn)
class Listbox(Widget):
"""Listbox class.
methods:
- Listbox(self, height, scroll = 0, returnExit = 0, width = 0, showCursor = 0, multiple = 0, border = 0)
- insert(self, text, item, before) : insert element; before = key to item to insert before, or None.
- delete(self, item) : delete item from list.
- replace(self, text,item) : Replace a given item's text
- current(self) : returns currently selected item
- getSelection(self) : returns a list of selected items
- setCurrent(self,i tem) : select current.
- clear(self) : clear listbox
"""
def append(self, text, item):
key = self.w.listboxAddItem(text)
self.key2item[key] = item
self.item2key[item] = key
def insert(self, text, item, before):
if (not before):
key = self.w.listboxInsertItem(text, 0)
else:
key = self.w.listboxInsertItem(text, self.item2key[before])
self.key2item[key] = item
self.item2key[item] = key
def delete(self, item):
self.w.listboxDeleteItem(self.item2key[item])
del self.key2item[self.item2key[item]]
del self.item2key[item]
def replace(self, text, item):
key = self.w.listboxInsertItem(text, self.item2key[item])
self.w.listboxDeleteItem(self.item2key[item])
del self.key2item[self.item2key[item]]
self.item2key[item] = key
self.key2item[key] = item
def current(self):
return self.key2item[self.w.listboxGetCurrent()]
def getSelection(self):
selection = []
list = self.w.listboxGetSelection()
for key in list:
selection.append(self.key2item[key])
return selection
def setCurrent(self, item):
self.w.listboxSetCurrent(self.item2key[item])
def clear(self):
self.key2item = {}
self.item2key = {}
self.w.listboxClear()
def __init__(self, height, scroll = 0, returnExit = 0, width = 0, showCursor = 0, multiple = 0, border = 0):
self.w = _snack.listbox(height, scroll, returnExit, showCursor, multiple, border)
self.key2item = {}
self.item2key = {}
if (width):
self.w.listboxSetWidth(width)
class Textbox(Widget):
"""Textbox, container for text.
methods:
- Textbox(self, width, height, scroll = 0, wrap = 0): scroll, wrap are flags
include scroll bars, or text wrap.
- setText(text) : set text.
- setHeight(height): set height.
"""
def setText(self, text):
self.w.textboxText(text)
def setHeight(self, height):
self.w.textboxHeight(height)
def __init__(self, width, height, text, scroll = 0, wrap = 0):
self.w = _snack.textbox(width, height, text, scroll, wrap)
class TextboxReflowed(Textbox):
def __init__(self, width, text, flexDown = 5, flexUp = 10, maxHeight = -1):
(newtext, width, height) = reflow(text, width, flexDown, flexUp)
if maxHeight != -1 and height > maxHeight:
Textbox.__init__(self, width, maxHeight, newtext, 1)
else:
Textbox.__init__(self, width, height, newtext, 0)
class Label(Widget):
"""A Label (simple text).
methods:
- Label(self,text) : create label
- setText(self,text) : change text.
- setColors(self, colorset) : change individual colors
"""
def setText(self, text):
self.w.labelText(text)
def __init__(self, text):
self.w = _snack.label(text)
def setColors(self, colorset):
self.w.labelSetColors(colorset)
class Scale(Widget):
"""A Scale (progress bar).
methods:
- Scale(self,width, total) : create scale; width: size on screen, fullamount: integer.
- set(self,amount) : set amount to integer.
"""
def set(self, amount):
self.w.scaleSet(amount)
def __init__(self, width, total):
self.w = _snack.scale(width, total)
class Entry(Widget):
"""Entry widget.
methods:
- Entry(self, width, text = "", hidden = 0, password = 0, scroll = 1, returnExit = 0)
constructor. hidden doesn't show text, password stars it out,
scroll includes scroll bars;
if returnExit is set, return from Form when exiting this element, else
proceed to next entry widget.
- value(self): return value.
- set(text, cursorAtEnd = 1) : set the text
- setFlags (flag, sense) : flags can be FLAG_DISABLED, FLAGS_SET, FLAGS_RESET, FLAGS_TOGGLE
"""
def value(self):
return self.w.entryValue
def set(self, text, cursorAtEnd = 1):
return self.w.entrySetValue(text, cursorAtEnd)
def setFlags (self, flag, sense):
return self.w.entrySetFlags(flag, sense)
def __init__(self, width, text = "", hidden = 0, password = 0, scroll = 1,
returnExit = 0):
self.w = _snack.entry(width, text, hidden, password, scroll, returnExit)
# Form uses hotkeys
hotkeys = { "F1" : _snack.KEY_F1, "F2" : _snack.KEY_F2, "F3" : _snack.KEY_F3,
"F4" : _snack.KEY_F4, "F5" : _snack.KEY_F5, "F6" : _snack.KEY_F6,
"F7" : _snack.KEY_F7, "F8" : _snack.KEY_F8, "F9" : _snack.KEY_F9,
"F10" : _snack.KEY_F10, "F11" : _snack.KEY_F11,
"F12" : _snack.KEY_F12, "ESC" : _snack.KEY_ESC,
"ENTER": _snack.KEY_ENTER, "SUSPEND" : _snack.KEY_SUSPEND,
"BACKSPACE": _snack.KEY_BACKSPACE, "DELETE": _snack.KEY_DELETE,
"INSERT": _snack.KEY_INSERT, "RESIZE": _snack.KEY_RESIZE,
" " : ord(" ") }
for n in list(hotkeys.keys()):
hotkeys[hotkeys[n]] = n
for o,c in [ (ord(c),c) for c in string.ascii_letters+string.digits ]:
hotkeys[c] = o
hotkeys[o] = c
class Form:
""" Base Form class, from which Grid, etc. inherit
methods:
- Form(self, helpArg = None) : constructor.
- addHotKey(self, keyname) : keynames of form "F1" through "F12", "ESC"
- add(self, widget) : Add a widget
- run(self): run a form, expecting input
- draw(self): draw form.
- setTimer(self, timer) : add a timer
- watchFile(self, file, flags) : watch a named file
- setCurrent (self, co): Set a given widget as the current focus
"""
def addHotKey(self, keyname):
self.w.addhotkey(hotkeys[keyname])
def add(self, widget):
if 'hotkeys' in widget.__dict__:
for key in widget.hotkeys.keys():
self.addHotKey(key)
if 'gridmembers' in widget.__dict__:
for w in widget.gridmembers:
self.add(w)
elif 'w' in widget.__dict__:
self.trans[widget.w.key] = widget
return self.w.add(widget.w)
return None
def run(self):
(what, which) = self.w.run()
if (what == _snack.FORM_EXIT_WIDGET):
return self.trans[which]
elif (what == _snack.FORM_EXIT_TIMER):
return "TIMER"
elif (what == _snack.FORM_EXIT_FDREADY):
return self.filemap[which]
elif (what == _snack.FORM_EXIT_HOTKEY):
return hotkeys[which]
raise RuntimeError("EOF or IO error")
def draw(self):
self.w.draw()
return None
def __init__(self, helpArg = None):
self.trans = {}
self.filemap = {}
self.w = _snack.form(helpArg)
# we do the reference count for the helpArg in python! gross
self.helpArg = helpArg
def setCurrent (self, co):
self.w.setcurrent (co.w)
def setTimer (self, timer):
self.w.settimer (timer)
def watchFile (self, file, flags):
self.filemap[file.fileno()] = file
self.w.watchfd (file.fileno(), flags)
class Grid:
"""Grid class.
methods:
- place(self,x,y): Return what is placed at (x,y)
- setField(self, what, col, row, padding = (0, 0, 0, 0),
anchorLeft = 0, anchorTop = 0, anchorRight = 0,
anchorBottom = 0, growx = 0, growy = 0):
used to add widget 'what' to grid.
- Grid(self, *args): eg. g = Grid(2,3) for 2x3 grid
"""
def place(self, x, y):
return self.g.place(x, y)
def setField(self, what, col, row, padding = (0, 0, 0, 0),
anchorLeft = 0, anchorTop = 0, anchorRight = 0,
anchorBottom = 0, growx = 0, growy = 0):
self.gridmembers.append(what)
anchorFlags = 0
if (anchorLeft):
anchorFlags = _snack.ANCHOR_LEFT
elif (anchorRight):
anchorFlags = _snack.ANCHOR_RIGHT
if (anchorTop):
anchorFlags = anchorFlags | _snack.ANCHOR_TOP
elif (anchorBottom):
anchorFlags = anchorFlags | _snack.ANCHOR_BOTTOM
gridFlags = 0
if (growx):
gridFlags = _snack.GRID_GROWX
if (growy):
gridFlags = gridFlags | _snack.GRID_GROWY
if 'g' in what.__dict__:
return self.g.setfield(col, row, what.g, padding, anchorFlags,
gridFlags)
else:
return self.g.setfield(col, row, what.w, padding, anchorFlags)
def __init__(self, *args):
self.g = _snack.grid(*args)
self.gridmembers = []
colorsets = { "ROOT" : _snack.COLORSET_ROOT,
"BORDER" : _snack.COLORSET_BORDER,
"WINDOW" : _snack.COLORSET_WINDOW,
"SHADOW" : _snack.COLORSET_SHADOW,
"TITLE" : _snack.COLORSET_TITLE,
"BUTTON" : _snack.COLORSET_BUTTON,
"ACTBUTTON" : _snack.COLORSET_ACTBUTTON,
"CHECKBOX" : _snack.COLORSET_CHECKBOX,
"ACTCHECKBOX" : _snack.COLORSET_ACTCHECKBOX,
"ENTRY" : _snack.COLORSET_ENTRY,
"LABEL" : _snack.COLORSET_LABEL,
"LISTBOX" : _snack.COLORSET_LISTBOX,
"ACTLISTBOX" : _snack.COLORSET_ACTLISTBOX,
"TEXTBOX" : _snack.COLORSET_TEXTBOX,
"ACTTEXTBOX" : _snack.COLORSET_ACTTEXTBOX,
"HELPLINE" : _snack.COLORSET_HELPLINE,
"ROOTTEXT" : _snack.COLORSET_ROOTTEXT,
"EMPTYSCALE" : _snack.COLORSET_EMPTYSCALE,
"FULLSCALE" : _snack.COLORSET_FULLSCALE,
"DISENTRY" : _snack.COLORSET_DISENTRY,
"COMPACTBUTTON" : _snack.COLORSET_COMPACTBUTTON,
"ACTSELLISTBOX" : _snack.COLORSET_ACTSELLISTBOX,
"SELLISTBOX" : _snack.COLORSET_SELLISTBOX }
class SnackScreen:
"""A Screen;
methods:
- Screen(self) : constructor
- finish(self)
- resume(self)
- suspend(self)
- doHelpCallback(self,arg) call callback with arg
- helpCallback(self,cb): Set help callback
- suspendcallback(self,cb, data=None) : set callback. data=data to pass to cb.
- openWindow(self,left, top, width, height, title): Open a window.
- pushHelpLine(self,text): put help line on screen. Returns current help line if text=None
- setColor(self, colorset, fg, bg): Set foreground and background colors;
colorset = key from snack.colorsets,
fg & bg = english color names defined by S-Lang
(ref: S-Lang Library C Programmer's Guide section:
8.4.4. Setting Character Attributes)
"""
def __init__(self):
_snack.init()
(self.width, self.height) = _snack.size()
self.pushHelpLine(None)
def finish(self):
return _snack.finish()
def resume(self):
_snack.resume()
def suspend(self):
_snack.suspend()
def doHelpCallback(self, arg):
self.helpCb(self, arg)
def helpCallback(self, cb):
self.helpCb = cb
return _snack.helpcallback(self.doHelpCallback)
def suspendCallback(self, cb, data = None):
if data:
return _snack.suspendcallback(cb, data)
return _snack.suspendcallback(cb)
def openWindow(self, left, top, width, height, title):
return _snack.openwindow(left, top, width, height, title)
def pushHelpLine(self, text):
if (not text):
return _snack.pushhelpline("*default*")
else:
return _snack.pushhelpline(text)
def popHelpLine(self):
return _snack.pophelpline()
def drawRootText(self, left, top, text):
return _snack.drawroottext(left, top, text)
def centeredWindow(self, width, height, title):
return _snack.centeredwindow(width, height, title)
def gridWrappedWindow(self, grid, title, x = None, y = None):
if x and y:
return _snack.gridwrappedwindow(grid.g, title, x, y)
return _snack.gridwrappedwindow(grid.g, title)
def popWindow(self, refresh = True):
if refresh:
return _snack.popwindow()
return _snack.popwindownorefresh()
def refresh(self):
return _snack.refresh()
def setColor(self, colorset, fg, bg):
if colorset in colorsets:
return _snack.setcolor(colorsets[colorset], fg, bg)
else:
# assume colorset is an integer for the custom color set
return _snack.setcolor(colorset, fg, bg)
def reflow(text, width, flexDown = 5, flexUp = 5):
""" returns a tuple of the wrapped text, the actual width, and the actual height
"""
return _snack.reflow(text, width, flexDown, flexUp)
# combo widgets
class RadioGroup(Widget):
""" Combo widget: Group of Radio buttons
methods:
- RadioGroup(self): constructor.
- add(self,title, value, default = None): add a button. Returns button.
- getSelection(self) : returns value of selected button | None
"""
def __init__(self):
self.prev = None
self.buttonlist = []
def add(self, title, value, default = None):
if not self.prev and default == None:
# If the first element is not explicitly set to
# not be the default, make it be the default
default = 1
b = SingleRadioButton(title, self.prev, default)
self.prev = b
self.buttonlist.append((b, value))
return b
def getSelection(self):
for (b, value) in self.buttonlist:
if b.selected(): return value
return None
class RadioBar(Grid):
""" Bar of Radio buttons, based on Grid.
methods:
- RadioBar(self, screen, buttonlist) : constructor.
- getSelection(self): return value of selected button
"""
def __init__(self, screen, buttonlist):
self.list = []
self.item = 0
self.group = RadioGroup()
Grid.__init__(self, 1, len(buttonlist))
for (title, value, default) in buttonlist:
b = self.group.add(title, value, default)
self.list.append((b, value))
self.setField(b, 0, self.item, anchorLeft = 1)
self.item = self.item + 1
def getSelection(self):
return self.group.getSelection()
# you normally want to pack a ButtonBar with growx = 1
class ButtonBar(Grid):
""" Bar of buttons, based on grid.
methods:
- ButtonBar(screen, buttonlist,buttonlist, compact = 0):
- buttonPressed(self, result): Takes the widget returned by Form.run and looks to see
if it was one of the widgets in the ButtonBar.
"""
def __init__(self, screen, buttonlist, compact = 0):
self.list = []
self.hotkeys = {}
self.item = 0
Grid.__init__(self, len(buttonlist), 1)
for blist in buttonlist:
if isinstance(blist, str if sys.version >= '3' else basestring):
title = blist
value = blist.lower()
elif len(blist) == 2:
(title, value) = blist
else:
(title, value, hotkey) = blist
self.hotkeys[hotkey] = value
if compact:
b = CompactButton(title)
else:
b = Button(title)
self.list.append((b, value))
self.setField(b, self.item, 0, (1, 0, 1, 0))
self.item = self.item + 1
def buttonPressed(self, result):
if result in self.hotkeys:
return self.hotkeys[result]
for (button, value) in self.list:
if result == button:
return value
return None
class GridFormHelp(Grid):
""" Subclass of Grid, for the help form text.
methods:
- GridFormHelp(self, screen, title, help, *args) :
- add (self, widget, col, row, padding = (0, 0, 0, 0),
anchorLeft = 0, anchorTop = 0, anchorRight = 0,
anchorBottom = 0, growx = 0, growy = 0):
- runOnce(self, x = None, y = None): pop up the help window
- addHotKey(self, keyname):
- setTimer(self, keyname):
- create(self, x = None, y = None):
- run(self, x = None, y = None):
- draw(self):
- runPopup(self):
- setCurrent (self, co):
"""
def __init__(self, screen, title, help, *args):
self.screen = screen
self.title = title
self.form = Form(help)
self.childList = []
self.form_created = 0
args = list(args)
args[:0] = [self]
Grid.__init__(*tuple(args))
def add(self, widget, col, row, padding = (0, 0, 0, 0),
anchorLeft = 0, anchorTop = 0, anchorRight = 0,
anchorBottom = 0, growx = 0, growy = 0):
self.setField(widget, col, row, padding, anchorLeft,
anchorTop, anchorRight, anchorBottom,
growx, growy);
self.childList.append(widget)
def runOnce(self, x = None, y = None):
result = self.run(x, y)
self.screen.popWindow()
return result
def addHotKey(self, keyname):
self.form.addHotKey(keyname)
def setTimer(self, keyname):
self.form.setTimer(keyname)
def create(self, x = None, y = None):
if not self.form_created:
self.place(1,1)
for child in self.childList:
self.form.add(child)
self.screen.gridWrappedWindow(self, self.title, x, y)
self.form_created = 1
def run(self, x = None, y = None):
self.create(x, y)
return self.form.run()
def draw(self):
self.create()
return self.form.draw()
def runPopup(self):
self.create()
self.screen.gridWrappedWindow(self, self.title)
result = self.form.run()
self.screen.popWindow()
return result
def setCurrent (self, co):
self.form.setCurrent (co)
class GridForm(GridFormHelp):
""" GridForm class (extends GridFormHelp):
methods:
- GridForm(self, screen, title, *args):
"""
def __init__(self, screen, title, *args):
myargs = (self, screen, title, None) + args
GridFormHelp.__init__(*myargs)
class CheckboxTree(Widget):
""" CheckboxTree combo widget,
methods:
- CheckboxTree(self, height, scroll = 0, width = None, hide_checkbox = 0, unselectable = 0)
constructor.
- append(self, text, item = None, selected = 0):
- addItem(self, text, path, item = None, selected = 0):
- getCurrent(self):
- getSelection(self):
- setEntry(self, item, text):
- setCurrent(self, item):
- setEntryValue(self, item, selected = 1):
- getEntryValue(self, item):
"""
def append(self, text, item = None, selected = 0):
self.addItem(text, (snackArgs['append'], ), item, selected)
def addItem(self, text, path, item = None, selected = 0):
if item is None:
item = text
key = self.w.checkboxtreeAddItem(text, path, selected)
self.key2item[key] = item
self.item2key[item] = key
def getCurrent(self):
curr = self.w.checkboxtreeGetCurrent()
return self.key2item[curr]
def __init__(self, height, scroll = 0, width = None, hide_checkbox = 0, unselectable = 0):
self.w = _snack.checkboxtree(height, scroll, hide_checkbox, unselectable)
self.key2item = {}
self.item2key = {}
if (width):
self.w.checkboxtreeSetWidth(width)
def getSelection(self):
selection = []
list = self.w.checkboxtreeGetSelection()
for key in list:
selection.append(self.key2item[key])
return selection
def setEntry(self, item, text):
self.w.checkboxtreeSetEntry(self.item2key[item], text)
def setCurrent(self, item):
self.w.checkboxtreeSetCurrent(self.item2key[item])
def setEntryValue(self, item, selected = 1):
self.w.checkboxtreeSetEntryValue(self.item2key[item], selected)
def getEntryValue(self, item):
return self.w.checkboxtreeGetEntryValue(self.item2key[item])
def ListboxChoiceWindow(screen, title, text, items,
buttons = ('Ok', 'Cancel'),
width = 40, scroll = 0, height = -1, default = None,
help = None):
"""
- ListboxChoiceWindow(screen, title, text, items,
buttons = ('Ok', 'Cancel'),
width = 40, scroll = 0, height = -1, default = None,
help = None):
"""
if (height == -1): height = len(items)
bb = ButtonBar(screen, buttons)
t = TextboxReflowed(width, text)
l = Listbox(height, scroll = scroll, returnExit = 1)
count = 0
for item in items:
if type(item) == tuple:
(text, key) = item
else:
text = item
key = count
if (default == count):
default = key
elif (default == item):
default = key
l.append(text, key)
count = count + 1
if (default != None):
l.setCurrent (default)
g = GridFormHelp(screen, title, help, 1, 3)
g.add(t, 0, 0)
g.add(l, 0, 1, padding = (0, 1, 0, 1))
g.add(bb, 0, 2, growx = 1)
rc = g.runOnce()
return (bb.buttonPressed(rc), l.current())
def ButtonChoiceWindow(screen, title, text,
buttons = [ 'Ok', 'Cancel' ],
width = 40, x = None, y = None, help = None):
"""
- ButtonChoiceWindow(screen, title, text,
buttons = [ 'Ok', 'Cancel' ],
width = 40, x = None, y = None, help = None):
"""
bb = ButtonBar(screen, buttons)
t = TextboxReflowed(width, text, maxHeight = screen.height - 12)
g = GridFormHelp(screen, title, help, 1, 2)
g.add(t, 0, 0, padding = (0, 0, 0, 1))
g.add(bb, 0, 1, growx = 1)
return bb.buttonPressed(g.runOnce(x, y))
def EntryWindow(screen, title, text, prompts, allowCancel = 1, width = 40,
entryWidth = 20, buttons = [ 'Ok', 'Cancel' ], help = None):
"""
EntryWindow(screen, title, text, prompts, allowCancel = 1, width = 40,
entryWidth = 20, buttons = [ 'Ok', 'Cancel' ], help = None):
"""
bb = ButtonBar(screen, buttons);
t = TextboxReflowed(width, text)
count = 0
for n in prompts:
count = count + 1
sg = Grid(2, count)
count = 0
entryList = []
for n in prompts:
if type(n) == tuple:
(n, e) = n
if isinstance(e, str if sys.version >= '3' else basestring):
e = Entry(entryWidth, e)
else:
e = Entry(entryWidth)
sg.setField(Label(n), 0, count, padding = (0, 0, 1, 0), anchorLeft = 1)
sg.setField(e, 1, count, anchorLeft = 1)
count = count + 1
entryList.append(e)
g = GridFormHelp(screen, title, help, 1, 3)
g.add(t, 0, 0, padding = (0, 0, 0, 1))
g.add(sg, 0, 1, padding = (0, 0, 0, 1))
g.add(bb, 0, 2, growx = 1)
result = g.runOnce()
entryValues = []
count = 0
for n in prompts:
entryValues.append(entryList[count].value())
count = count + 1
return (bb.buttonPressed(result), tuple(entryValues))
class CListbox(Grid):
"""Clistbox convenience class.
methods:
- Clistbox(self, height, cols, cols_widths, scroll = 0) : constructor
- colFormText(self, col_text, align = None, adjust_width = 0) : column text.
- append(self, col_text, item, col_text_align = None) :
- insert(self, col_text, item, before, col_text_align = None)
- delete(self, item)
- replace(self, col_text, item, col_text_align = None)
- current(self) : returns current item
- setCurrent(self, item): sets an item as current
- clear(self): clear the listbox
Alignments may be LEFT, RIGHT, CENTER, None
"""
def __init__(self, height, cols, col_widths, scroll = 0,
returnExit = 0, width = 0, col_pad = 1,
col_text_align = None, col_labels = None,
col_label_align = None, adjust_width=0):
self.cols = cols
self.col_widths = col_widths[:]
self.col_pad = col_pad
self.col_text_align = col_text_align
if col_labels != None:
Grid.__init__(self, 1, 2)
box_y = 1
lstr = self.colFormText(col_labels, col_label_align,
adjust_width=adjust_width)
self.label = Label(lstr)
self.setField(self.label, 0, 0, anchorLeft=1)
else:
Grid.__init__(self, 1, 1)
box_y = 0
self.listbox = Listbox(height, scroll, returnExit, width)
self.setField(self.listbox, 0, box_y, anchorRight=1)
def colFormText(self, col_text, align = None, adjust_width=0):
i = 0
str = ""
c_len = len(col_text)
while (i < self.cols) and (i < c_len):
cstr = col_text[i]
cstrlen = _snack.wstrlen(cstr)
if self.col_widths[i] < cstrlen:
if adjust_width:
self.col_widths[i] = cstrlen
else:
cstr = cstr[:self.col_widths[i]]
delta = self.col_widths[i] - _snack.wstrlen(cstr)
if delta > 0:
if align == None:
a = LEFT
else:
a = align[i]
if a == LEFT:
cstr = cstr + (" " * delta)
if a == CENTER:
cstr = (" " * (delta / 2)) + cstr + \
(" " * ((delta + 1) / 2))
if a == RIGHT:
cstr = (" " * delta) + cstr
if i != c_len - 1:
pstr = (" " * self.col_pad)
else:
pstr = ""
str = str + cstr + pstr
i = i + 1
return str
def append(self, col_text, item, col_text_align = None):
if col_text_align == None:
col_text_align = self.col_text_align
text = self.colFormText(col_text, col_text_align)
self.listbox.append(text, item)
def insert(self, col_text, item, before, col_text_align = None):
if col_text_align == None:
col_text_align = self.col_text_align
text = self.colFormText(col_text, col_text_align)
self.listbox.insert(text, item, before)
def delete(self, item):
self.listbox.delete(item)
def replace(self, col_text, item, col_text_align = None):
if col_text_align == None:
col_text_align = self.col_text_align
text = self.colFormText(col_text, col_text_align)
self.listbox.replace(text, item)
def current(self):
return self.listbox.current()
def setCurrent(self, item):
self.listbox.setCurrent(item)
def clear(self):
self.listbox.clear()
def customColorset(x):
return 30 + x
OHA YOOOO