Question

我有一个简单的脚本可以通过SSH连接到网络交换机并运行命令并将输出保存到文件中。它适用于即时显示的输出，但是当我运行＆＃34; show iproute＆＃34;它不捕获任何输出。原因是当我直接在开关上运行相同的命令时，它认为是5-6秒，显示一堆线并再次思考并显示更多的线然后结束。它没有正确等待整个命令执行我有问题修复：

str_prompt = ' # '
command = "sh iproute"
device_name = "switch1.test.com"

# Spawn SSH session
ssh_command = 'ssh {}@{}'.format(username, device_name)
session = pexpect.spawn(ssh_command, timeout=5)

# Send the password
session.sendline(password)

# Expect the switch prompt (successful login)
expect_index = session.expect([pexpect.TIMEOUT, str_prompt])
# Success
if expect_index == 1:
    # Disable clipaging so that all the output is shown (not in pages) | same as term len 0 in Cisco
    session.sendline('disable clip')
    # Expect the switch prompt if command is successful
    expect_index = session.expect([pexpect.TIMEOUT, str_prompt])

    # Send show iproute command
    session.sendline(command)
    # < This is where it needs to wait >
    #session.expect(pexpect.EOF) - Tried this and wait() but that broke the scipt
    #session.wait()
    # Expect the switch prompt if command is successful
    session.expect(str_prompt)

    # Save output of "sh iproute" to a variable
    output = session.before
    # Save results to a file
    fp = open(host + '-route.txt', "w")
    fp.write(output)
    fp.close()

这是一个示例输出。输出确实有＆＃34;＃＆＃34;但不是＆＃34; ＃＆＃34;。

#oa  10.10.10.0/24      10.0.0.1    4    UG-D---um--f- V-BB1 99d:0h:14m:49s
#oa  10.10.20.0/24      10.0.0.1    4    UG-D---um--f- V-BB2 99d:0h:14m:49s
#oa  10.10.30.0/24      10.0.0.1    4    UG-D---um--f- V-BB3 99d:0h:14m:49s
#oa  10.10.40.0/24      10.0.0.1    4    UG-D---um--f- V-BB4 99d:0h:14m:49s
and many more line ....

任何帮助将不胜感激。谢谢

编辑：我添加了睡眠（60），这似乎是诀窍，但我不想使用它，因为我发送多个命令，有些是超快。我不想为每个命令等待1分钟，脚本将永远运行。

Answer 1

因此您需要将超时与命令相关联。我今天这样做的方式是我的代码解析的xml文件格式，xml标签将有一个命令属性，另一个用于超时，另一个用于命令的end_prompt等等。

我的代码读取命令及其超时，并在发送命令之前相应地设置相应的变量。

session.sendline(command) #command is read from a xml file
session.expect(end_prompt, timeout=int(tmout)) # end_prompt, tmout for the command read form the same file

对于您的情况，如果您不想解析文件以获取命令及其相关参数，您可以将它们作为字典添加到脚本中并使用它

command_details_dict = { "cmd_details" :[
                                         {'cmd': 'pwd',
                                          'timeout': 5,
                                          },
                                         {'cmd': 'iproute',
                                          'timeout': 60,
                                          }
                                         ]
                       }

字典内的

cmd_details是一个字典列表，这样你可以在迭代时维护命令的顺序，每个命令都是一个包含相关细节的字典，你可以在命令字典中添加更多的键，如提示，唯一标识符etcc。

但如果你有时间我会建议使用配置文件

Pexpect没有等待整个输出 - Ubuntu

1 个答案: