python - 是否有可能从 pexpect.expect 中取回匹配项？

Question

我正在自动测试一个动态生成的菜单——也就是说，每个项目都可以在它之前获得几个数字之一。我可以说

a = child.expect('1\) Set Password', '2\) Set Password', '3\) Set Password')
if a == 0: child.sendline('1')
elif a == 1: child.sendline('2')
elif a == 2: child.sendline('3')

但这相当笨重。我宁愿有这样的东西:

child.expect('(\d)\) Set Password')
a = child.get_match()
child.sendline(a)

有没有这样的功能？

Answer 1

Thomas K 在评论中提供了这个答案:

 child.match.group(1)

应该让你得到你想要的。

确实如此。