def pause(n=None):
r"""Waits for either user input or a specific number of seconds.
Examples:
.. doctest::
:skipif: branch_dev
>>> with context.local(log_level="INFO"):
... pause(1)
[x] Waiting
[x] Waiting: 1...
[+] Waiting: Done
>>> pause("whatever")
Traceback (most recent call last):
...
ValueError: pause(): n must be a number or None
Tests:
.. doctest::
:skipif: branch_dev
>>> saved_stdin = sys.stdin
>>> try:
... sys.stdin = io.TextIOWrapper(io.BytesIO(b"\n"))
... with context.local(log_level="INFO"):
... pause()
... finally:
... sys.stdin = saved_stdin
[*] Paused (press enter to continue)
>>> p = testpwnproc("pause()")
>>> b"Paused" in p.recvuntil(b"press any")
True
>>> p.send(b"x")
>>> _ = p.recvall()
"""
if n is None:
if term.term_mode:
log.info('Paused (press any to continue)')
term.getkey()
else:
log.info('Paused (press enter to continue)')
raw_input('')
elif isinstance(n, six.integer_types):
with log.waitfor("Waiting") as l:
for i in range(n, 0, -1):
l.status('%d... ' % i)
time.sleep(1)
l.success()
else:
raise ValueError('pause(): n must be a number or None')
def pause(n=None):
"""Waits for either user input or a specific number of seconds."""
if n == None:
if term.term_mode:
log.info('Paused (press any to continue)')
term.getkey()
else:
log.info('Paused (press enter to continue)')
raw_input('')
elif isinstance(n, six.integer_types):
with log.waitfor("Waiting") as l:
for i in range(n, 0, -1):
l.status('%d... ' % i)
time.sleep(1)
l.success()
else:
raise ValueError('pause(): n must be a number or None')
def yesno(prompt, default=None):
"""Presents the user with prompt (typically in the form of question) which
the user must answer yes or no.
Arguments:
prompt (str): The prompt to show
default: The default option; `True` means "yes"
Returns:
`True` if the answer was "yes", `False` if "no"
"""
if default is not None and not isinstance(default, bool):
raise ValueError('yesno(): default must be a boolean or None')
if term.term_mode:
term.output(' [?] %s [' % prompt)
yesfocus, yes = term.text.bold('Yes'), 'yes'
nofocus, no = term.text.bold('No'), 'no'
hy = term.output(yesfocus if default == True else yes)
term.output('/')
hn = term.output(nofocus if default == False else no)
term.output(']\n')
cur = default
while True:
k = term.key.get()
if k in ('y', 'Y', '<left>') and cur != True:
cur = True
hy.update(yesfocus)
hn.update(no)
elif k in ('n', 'N', '<right>') and cur != False:
cur = False
hy.update(yes)
hn.update(nofocus)
elif k == '<enter>':
if cur is not None:
return cur
else:
prompt = ' [?] %s [%s/%s] ' % (
prompt,
'Yes' if default == True else 'yes',
'No' if default == False else 'no',
)
while True:
opt = raw_input(prompt).lower()
if opt == '' and default != None:
return default
elif opt in ('y', 'yes'):
return True
elif opt in ('n', 'no'):
return False
print('Please answer yes or no')
def yesno(prompt, default=None):
r"""Presents the user with prompt (typically in the form of question)
which the user must answer yes or no.
Arguments:
prompt (str): The prompt to show
default: The default option; `True` means "yes"
Returns:
`True` if the answer was "yes", `False` if "no"
Examples:
>>> yesno("A number:", 20)
Traceback (most recent call last):
...
ValueError: yesno(): default must be a boolean or None
>>> saved_stdin = sys.stdin
>>> try:
... sys.stdin = io.TextIOWrapper(io.BytesIO(b"x\nyes\nno\n\n"))
... yesno("is it good 1")
... yesno("is it good 2", True)
... yesno("is it good 3", False)
... finally:
... sys.stdin = saved_stdin
[?] is it good 1 [yes/no] Please answer yes or no
[?] is it good 1 [yes/no] True
[?] is it good 2 [Yes/no] False
[?] is it good 3 [yes/No] False
Tests:
>>> p = testpwnproc("print(yesno('is it ok??'))")
>>> b"is it ok" in p.recvuntil("??")
True
>>> p.sendline(b"x\nny")
>>> b"True" in p.recvall()
True
"""
if default is not None and not isinstance(default, bool):
raise ValueError('yesno(): default must be a boolean or None')
if term.term_mode:
term.output(' [?] %s [' % prompt)
yesfocus, yes = term.text.bold('Yes'), 'yes'
nofocus, no = term.text.bold('No'), 'no'
hy = term.output(yesfocus if default is True else yes)
term.output('/')
hn = term.output(nofocus if default is False else no)
term.output(']\n')
cur = default
while True:
k = term.key.get()
if k in ('y', 'Y', '<left>') and cur is not True:
cur = True
hy.update(yesfocus)
hn.update(no)
elif k in ('n', 'N', '<right>') and cur is not False:
cur = False
hy.update(yes)
hn.update(nofocus)
elif k == '<enter>':
if cur is not None:
return cur
else:
prompt = ' [?] %s [%s/%s] ' % (
prompt,
'Yes' if default is True else 'yes',
'No' if default is False else 'no',
)
while True:
opt = raw_input(prompt).strip().lower()
if not opt and default is not None:
return default
elif opt in (b'y', b'yes'):
return True
elif opt in (b'n', b'no'):
return False
print('Please answer yes or no')
def options(prompt, opts, default=None):
r"""Presents the user with a prompt (typically in the
form of a question) and a number of options.
Arguments:
prompt (str): The prompt to show
opts (list): The options to show to the user
default: The default option to choose
Returns:
The users choice in the form of an integer.
Examples:
>>> options("Select a color", ("red", "green", "blue"), "green")
Traceback (most recent call last):
...
ValueError: options(): default must be a number or None
Tests:
>>> p = testpwnproc("print(options('select a color', ('red', 'green', 'blue')))")
>>> p.sendline(b"\33[C\33[A\33[A\33[B\33[1;5A\33[1;5B 0310")
>>> _ = p.recvall()
>>> saved_stdin = sys.stdin
>>> try:
... sys.stdin = io.TextIOWrapper(io.BytesIO(b"\n4\n\n3\n"))
... with context.local(log_level="INFO"):
... options("select a color A", ("red", "green", "blue"), 0)
... options("select a color B", ("red", "green", "blue"))
... finally:
... sys.stdin = saved_stdin
[?] select a color A
1) red
2) green
3) blue
Choice [1] 0
[?] select a color B
1) red
2) green
3) blue
Choice [?] select a color B
1) red
2) green
3) blue
Choice [?] select a color B
1) red
2) green
3) blue
Choice 2
"""
if default is not None and not isinstance(default, six.integer_types):
raise ValueError('options(): default must be a number or None')
if term.term_mode:
numfmt = '%' + str(len(str(len(opts)))) + 'd) '
print(' [?] ' + prompt)
hs = []
space = ' '
arrow = term.text.bold_green(' => ')
cur = default
for i, opt in enumerate(opts):
h = term.output(arrow if i == cur else space, frozen=False)
num = numfmt % (i + 1)
term.output(num)
term.output(opt + '\n', indent=len(num) + len(space))
hs.append(h)
ds = ''
while True:
prev = cur
was_digit = False
k = term.key.get()
if k == '<up>':
if cur is None:
cur = 0
else:
cur = max(0, cur - 1)
elif k == '<down>':
if cur is None:
cur = 0
else:
cur = min(len(opts) - 1, cur + 1)
elif k == 'C-<up>':
cur = 0
elif k == 'C-<down>':
cur = len(opts) - 1
elif k in ('<enter>', '<right>'):
if cur is not None:
return cur
elif k in tuple(string.digits):
was_digit = True
d = str(k)
n = int(ds + d)
if 0 < n <= len(opts):
ds += d
cur = n - 1
elif d != '0':
ds = d
n = int(ds)
cur = n - 1
if prev != cur:
if prev is not None:
hs[prev].update(space)
if was_digit:
hs[cur].update(term.text.bold_green('%5s> ' % ds))
else:
hs[cur].update(arrow)
else:
linefmt = ' %' + str(len(str(len(opts)))) + 'd) %s'
if default is not None:
default += 1
while True:
print(' [?] ' + prompt)
for i, opt in enumerate(opts):
print(linefmt % (i + 1, opt))
s = ' Choice '
if default:
s += '[%s] ' % str(default)
try:
x = int(raw_input(s) or default)
except (ValueError, TypeError):
continue
if x >= 1 and x <= len(opts):
return x - 1
def options(prompt, opts, default=None):
"""Presents the user with a prompt (typically in the
form of a question) and a number of options.
Arguments:
prompt (str): The prompt to show
opts (list): The options to show to the user
default: The default option to choose
Returns:
The users choice in the form of an integer.
"""
if default is not None and not isinstance(default, six.integer_types):
raise ValueError('options(): default must be a number or None')
if term.term_mode:
numfmt = '%' + str(len(str(len(opts)))) + 'd) '
print(' [?] ' + prompt)
hs = []
space = ' '
arrow = term.text.bold_green(' => ')
cur = default
for i, opt in enumerate(opts):
h = term.output(arrow if i == cur else space, frozen=False)
num = numfmt % (i + 1)
term.output(num)
term.output(opt + '\n', indent=len(num) + len(space))
hs.append(h)
ds = ''
prev = 0
while True:
prev = cur
was_digit = False
k = term.key.get()
if k == '<up>':
if cur is None:
cur = 0
else:
cur = max(0, cur - 1)
elif k == '<down>':
if cur is None:
cur = 0
else:
cur = min(len(opts) - 1, cur + 1)
elif k == 'C-<up>':
cur = 0
elif k == 'C-<down>':
cur = len(opts) - 1
elif k in ('<enter>', '<right>'):
if cur is not None:
return cur
elif k in tuple('1234567890'):
was_digit = True
d = str(k)
n = int(ds + d)
if n > 0 and n <= len(opts):
ds += d
elif d != '0':
ds = d
n = int(ds)
cur = n - 1
if prev != cur:
if prev is not None:
hs[prev].update(space)
if was_digit:
hs[cur].update(term.text.bold_green('%5s> ' % ds))
else:
hs[cur].update(arrow)
else:
linefmt = ' %' + str(len(str(len(opts)))) + 'd) %s'
while True:
print(' [?] ' + prompt)
for i, opt in enumerate(opts):
print(linefmt % (i + 1, opt))
s = ' Choice '
if default:
s += '[%s] ' % str(default)
try:
x = int(raw_input(s) or default)
except (ValueError, TypeError):
continue
if x >= 1 and x <= len(opts):
return x
