是否可以从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)

应该能得到你想要的东西。

确实如此。