防止 Paramiko 退出

Question

说明

我正在尝试编写一个 python 脚本以通过 SSH 连接到我的虚拟机并执行一些命令。

• 如果我手动执行此操作，我的会话将保持打开状态，并且我仍然可以看到我的实时日志。

• 如果我使用脚本，在 SSH 进入并自动执行一些命令后，它会不断退出并返回到我的用户提示符。我正在努力防止这种情况发生。

---

视频

手动 = https://dl.dropboxusercontent.com/u/56134944/ssh_wag_manually.mov

通过 Python 脚本 = https://dl.dropboxusercontent.com/u/56134944/ssh_wag_py.mov

---

这就是我现在拥有的

import paramiko
import time
import sys

# Note
# sudo pip install --user paramiko


def ssh_con (ip, un, pw):
    global client
    client = paramiko.SSHClient()
    client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    print ("Connecting to device/VM: %s" % ip)
    client.connect(ip, username=un, password=pw)


def cmd_io (command):
    global client_cmd
    client_cmd.send("%s \n" %command)
    time.sleep(1)
    output = client_cmd.recv(10000).decode("utf-8")
    print (output)

# ip = raw_input("Enter WAG IP : ")
# ip  = sys.argv[1]

ip = '172.168.200.300'
un = 'xyz'
pw = 'abc'

ssh_con(ip,un,pw)
client_cmd = client.invoke_shell()

print ("SSH CONNECTION ESTABLISHED TO vMEG %s" % ip)
cmd_io ("en")
cmd_io ("terminal monitor")
cmd_io ("debug wag https")
cmd_io ("debug wag httpc")
cmd_io ("debug https")
cmd_io ("debug httpc")
cmd_io ("debug wag kafka")

---

结果

[local]site#
terminal monitor 
[local]site#
debug wag https 
[local]site#
debug wag httpc 
[local]site#
debug https 
[local]site#
debug httpc 
[local]site#
debug wag kafka 
[local]site#

.... Quitting and return the command prompt ....

──[/Applications/MAMP/htdocs/code/python] 
└── 

---

如何防止这种情况发生？

Answer 1

将此函数放在脚本的顶部

def write_and_print_logs(log_message):
    with open("debugLogs.txt", 'a') as logs:
        logs.write("{}\n".format(log_message))
    print(log_message)

现在将每个print(<message>) 命令替换为write_and_print_logs(<message>)。现在所有内容都将写入文件，无论您是否实时查看日志，您都可以打开它并查看发生了什么。

Answer 2

您的主要问题是您只是停止读取 paramiko 的输出。如果您希望程序继续等待 VM 的进一步输出，您需要告诉它这样做。例如，这应该可以解决问题：

import paramiko
import time
import sys
import socket

# Note
# sudo pip install --user paramiko


def ssh_con (ip, un, pw):
    global client
    client = paramiko.SSHClient()
    client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    print ("Connecting to device/VM: %s" % ip)
    client.connect(ip, username=un, password=pw)


def cmd_io (command):
    global client_cmd
    client_cmd.send("%s \n" %command)
    time.sleep(1)
    output = client_cmd.recv(10000).decode("utf-8")
    print (output)

# ip = raw_input("Enter WAG IP : ")
# ip  = sys.argv[1]

ip = '172.168.200.300'
un = 'xyz'
pw = 'abc'

ssh_con(ip,un,pw)
client_cmd = client.invoke_shell()

print ("SSH CONNECTION ESTABLISHED TO vMEG %s" % ip)
cmd_io ("en")
cmd_io ("terminal monitor")
cmd_io ("debug wag https")
cmd_io ("debug wag httpc")
cmd_io ("debug https")
cmd_io ("debug httpc")
cmd_io ("debug wag kafka")

client_cmd.settimeout(1.0)
while True:
    try:
        output = client_cmd.recv(10000).decode("utf-8")
        print (output)
    except socket.timeout:
        pass

但是，这并不能解决所有问题。您的脚本还应该更严格地处理可能的错误情况。例如，它应该读取 stderr 流，检查通道是否实际上已经关闭，并且做的不仅仅是等待每个命令完成 1 秒。

如果它只是一个快速脚本，那么缺少错误处理可能一切都很好。但是，如果您需要它 100% 可靠地工作，您将需要在某个时候解决它们。