cleaning up code per suggestions

This commit is contained in:
adam j hartz 2016-05-19 21:00:36 -04:00
parent 2f300e5b68
commit af2123e0a5
2 changed files with 7 additions and 7 deletions

View file

@ -212,9 +212,9 @@ class StateFile(Input):
given file name. This node type is likely not useful on its own.
"""
attrs = ('default_file', 'check')
attrs = ('default_file', 'check', 'ask_filename')
def __init__(self, default_file=None, check=True, ask=True):
def __init__(self, default_file=None, check=True, ask_filename=True):
"""
Parameters
----------
@ -231,7 +231,7 @@ class StateFile(Input):
self._df = None
super().__init__(prompt='filename: ', converter=None,
confirm=False, path=None)
self.ask_filename = ask
self.ask_filename = ask_filename
self.default_file = default_file
self.check = check
@ -388,8 +388,8 @@ class PrettyFormatter(Visitor):
return s
def visit_statefile(self, node):
s = '{0}(default_file={1!r}, check={2})'
s = s.format(node.__class__.__name__, node.default_file, node.check)
s = '{0}(default_file={1!r}, check={2}, ask_filename={3})'
s = s.format(node.__class__.__name__, node.default_file, node.check, node.ask_filename)
return s
def visit_while(self, node):

View file

@ -280,9 +280,9 @@ def make_wizard(default_file=None, confirm=False):
"1. Yes\n2. No, but ask me later.\n3. No, and don't ask me again."
"\n\n1, 2, or 3 [default: 2]? ")
passer = Pass()
saver = Save(check=False, ask=False, default_file=default_file)
saver = Save(check=False, ask_filename=False, default_file=default_file)
wiz = Question(q, {1: wiz, 2: passer, 3: saver},
converter=lambda x: int(x.strip()) if x !='' else 2)
converter=lambda x: int(x) if x != '' else 2)
return wiz