a62d0a2257
Ignore-this: c1a8930338b851e26506652dd09db5d8 darcs-hash:20090916190224-bd074-087e2f6b311e24d212376f10e3fa45a6d151ab01.gz
89 lines
3.1 KiB
Python
Executable file
89 lines
3.1 KiB
Python
Executable file
#!/usr/bin/env python
|
|
#
|
|
# DIALOGWIZARD.PY--
|
|
#
|
|
# Copyright (C) 2009 Antoine Durand-Gasselin
|
|
# Author: Antoine Durand-Gasselin <adg@crans.org>
|
|
#
|
|
|
|
from wizard import Step, PreviousStep, EndScenario
|
|
from itertools import izip
|
|
import dialog, time
|
|
|
|
def really_quit(dico):
|
|
raise EndScenario("game over!", data = dico)
|
|
|
|
class DialogStepGenerator():
|
|
u"""This class defines a step, that will prompt the user for various
|
|
fields."""
|
|
|
|
def __init__(self, backtitle):
|
|
self.d = dialog.Dialog()
|
|
self.d.add_persistent_args(["--backtitle", backtitle])
|
|
|
|
def _check_rc(self, rc):
|
|
if rc == 2: really_quit(dico)
|
|
if rc == 1: raise PreviousStep
|
|
|
|
def _skim_choices(self, choices, dico):
|
|
return [ f for f in choices if f.get('show_cond', lambda x : True)(dico) ]
|
|
|
|
def form_step(self, title, enonce, form, **kw):
|
|
def fn(dico, default):
|
|
sform = self._skim_choices(form, dico)
|
|
for field in sform:
|
|
item = default.get(field['var'], dico.get(field['var'], ''))
|
|
field['item'] = item
|
|
rc, res = self.d.form(enonce, fields = sform, title=title, **kw)
|
|
self._check_rc(rc)
|
|
for field, val in izip (sform, res):
|
|
dico[field['var']] = val
|
|
return dico
|
|
return Step(fn)
|
|
|
|
def select_step(self, title, enonce, var, choix, **kw):
|
|
def fn(dico, default):
|
|
schoix = []
|
|
for c in choix:
|
|
try:
|
|
schoix.append((c[0], c[1]))
|
|
except Exception,e:
|
|
if c.get('show_cond', lambda x : True)(dico):
|
|
schoix.append((c['label'], c['item']))
|
|
print schoix
|
|
rc, res = self.d.menu(enonce, choices = schoix, title = title, **kw)
|
|
self._check_rc(rc)
|
|
dico[var] = res
|
|
return dico
|
|
return Step(fn)
|
|
|
|
def input_step(self, title, enonce, var, **kw):
|
|
def fn(dico, default):
|
|
rc, res = self.d.inputbox(enonce, title = title, init = dico.get(var, ''), **kw)
|
|
self._check_rc(rc)
|
|
dico[var] = res
|
|
return dico
|
|
return Step(fn)
|
|
|
|
def checklist_step(self, title, enonce, liste):
|
|
def fn(dico, default):
|
|
sliste = []
|
|
for c in liste:
|
|
try:
|
|
if dico.get(c[0], False): checked = 'on'
|
|
else: checked = 'off'
|
|
sliste.append((c[0], c[1], checked))
|
|
except Exception,e:
|
|
if c.get('show_cond', lambda x : True)(dico):
|
|
if dico.get(c['var'], False): checked = 'on'
|
|
else: checked = 'off'
|
|
sliste.append((c['var'], c['item'], checked))
|
|
rc, res = self.d.checklist(enonce, title = title, choices = sliste, **kw)
|
|
self._check_rc(rc)
|
|
for tag, item, status in liste:
|
|
if tag in res:
|
|
dico[tag] = True
|
|
else:
|
|
dico[tag] = False
|
|
return dico
|
|
return Step(fn)
|