Python Telnetlib read_until'#'或'>',多个字符串确定?

时间:2014-02-27 16:46:57

标签: python port automated-tests telnet telnetlib

if (tn.read_until('>')):
    action1
else:
    action2

if (tn.read_until() == '>'):
    action1
else:
    action2

我只希望read_until()检查哪个String首先出现,然后执行不同的操作。 或者有任何等效的方法吗?

2 个答案:

答案 0 :(得分:5)

查看docs。读取直到想要期望的字符串作为位置参数和可选的超时。我会这样做:

>>> try:
...     response = tn.read_until(">", timeout=120) #or whatever timeout you choose.
... except EOFError as e:
...     print "Connection closed: %s" % e

>>> if ">" in response:
...    action1
... else:
...    action2

如果您想要多个不同的字符,可以使用read_some()

>>> while True: #really you should set some sort of a timeout here.
...    r = tn.read_some()
...    if any(x in r for x in ["#", ">"]):
...        break

答案 1 :(得分:0)

更好地使用:

buff_tpl = tn.expect(["#", ">"])

这将返回一个包含三个项目的元组,第一个(索引为0)是匹配的列表项的索引,第二个是和对象匹配的,第三个是输出文本。

您可以使用buff_tpl [2]打印缓冲区输出,并使用buff_tpl [0]知道匹配的项目