在Python中使用Paramiko？在ssh上实现交互式shell,

ar7v8xwq 于 2023-02-05 发布在 Shell

关注(0)|答案(2)|浏览(233)

我想写一个程序（在Windows 7上的Python 3.x中），它通过ssh在远程shell上执行多个命令。在查看paramikos的exec_command()函数后，我意识到它不适合我的用例（因为通道在命令执行之后关闭），因为这些命令取决于环境变量（由先前命令设置），并且不能被连接成一个exec_command()调用，因为它们将在程序中的不同时间被执行。
因此，我希望在同一个通道中执行命令，我研究的下一个选项是使用paramikos的invoke_shell()函数实现交互式shell：

ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect(host, username=user, password=psw, port=22)

channel = ssh.invoke_shell()

out = channel.recv(9999)

channel.send('cd mivne_final\n')
channel.send('ls\n')

while not channel.recv_ready():
    time.sleep(3)

out = channel.recv(9999)
print(out.decode("ascii"))

channel.send('cd ..\n')
channel.send('cd or_fail\n')
channel.send('ls\n')

while not channel.recv_ready():
    time.sleep(3)

out = channel.recv(9999)
print(out.decode("ascii"))

channel.send('cd ..\n')
channel.send('cd simulator\n')
channel.send('ls\n')

while not channel.recv_ready():
    time.sleep(3)

out = channel.recv(9999)
print(out.decode("ascii"))

ssh.close()

此代码存在一些问题：
1.第一个print并不总是打印ls输出（有时只在第二个print上打印）。
1.第一个cd和ls命令总是出现在输出中（我通过recv命令获得它们，作为输出的一部分），而后面的所有cd和ls命令有时会打印出来，有时则不会。
1.第二个和第三个cd和ls命令（打印时）始终出现在第一个ls输出之前。
我对这种“非决定论”感到困惑，非常感谢你的帮助。

shell

来源：https://stackoverflow.com/questions/35821184/implement-an-interactive-shell-over-ssh-in-python-using-paramiko

2条答案

按热度按时间

643ylb081#

import paramiko
import re

class ShellHandler:

    def __init__(self, host, user, psw):
        self.ssh = paramiko.SSHClient()
        self.ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        self.ssh.connect(host, username=user, password=psw, port=22)

        channel = self.ssh.invoke_shell()
        self.stdin = channel.makefile('wb')
        self.stdout = channel.makefile('r')

    def __del__(self):
        self.ssh.close()

    def execute(self, cmd):
        """

        :param cmd: the command to be executed on the remote computer
        :examples:  execute('ls')
                    execute('finger')
                    execute('cd folder_name')
        """
        cmd = cmd.strip('\n')
        self.stdin.write(cmd + '\n')
        finish = 'end of stdOUT buffer. finished with exit status'
        echo_cmd = 'echo {} $?'.format(finish)
        self.stdin.write(echo_cmd + '\n')
        shin = self.stdin
        self.stdin.flush()

        shout = []
        sherr = []
        exit_status = 0
        for line in self.stdout:
            if str(line).startswith(cmd) or str(line).startswith(echo_cmd):
                # up for now filled with shell junk from stdin
                shout = []
            elif str(line).startswith(finish):
                # our finish command ends with the exit status
                exit_status = int(str(line).rsplit(maxsplit=1)[1])
                if exit_status:
                    # stderr is combined with stdout.
                    # thus, swap sherr with shout in a case of failure.
                    sherr = shout
                    shout = []
                break
            else:
                # get rid of 'coloring and formatting' special characters
                shout.append(re.compile(r'(\x9B|\x1B\[)[0-?]*[ -/]*[@-~]').sub('', line).
                             replace('\b', '').replace('\r', ''))

        # first and last lines of shout/sherr contain a prompt
        if shout and echo_cmd in shout[-1]:
            shout.pop()
        if shout and cmd in shout[0]:
            shout.pop(0)
        if sherr and echo_cmd in sherr[-1]:
            sherr.pop()
        if sherr and cmd in sherr[0]:
            sherr.pop(0)

        return shin, shout, sherr

赞(0）回复(0）举报 2023-02-05

yvfmudvl2#

我尝试了上面的答案，但它不起作用，因为ECHO命令在Python CLI中返回了错误，我在SSH中使用了该命令。
所以我编写了另一个适用于Python CLI的代码，假设输出在一行中。
我也认为像f“print（'{finish}'）”这样的函数可以做和上面的ECHO一样的事情（终止符？？），但是我没有使用它，因为我的输出总是在一行中。

class MusicPlayer:
def __init__(self, host='', username='pi', password=''):
    self.ssh = paramiko.SSHClient()
    self.ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    self.ssh.connect(host, username=username, password=password)
    channel = self.ssh.invoke_shell()
    self.stdin = channel.makefile('wb')
    self.stdout = channel.makefile('r')
    self.in_history = []
    self.out_history = []
    self.init_vlc()
    self.print()
    # atexit.register(self.__del__)

def __del__(self):
    self.ssh.close()

def execute(self, cmd):
    self.in_history.append(cmd)
    self.stdin.write(cmd + '\n')

def print(self, lines=1):
    for line in self.stdout:
        lined = line.strip()
        print(lined)
        self.out_history.append(lined)
        if self.in_history[-1] in lined:
            next_one = self.stdout.__next__().strip()
            print(next_one)
            self.out_history.append(next_one)
            return next_one

def init_vlc(self):
    for command in ['python', 'import vlc', 'import time', 'media_player = vlc.MediaPlayer()']:
        self.execute(command)

赞(0）回复(0）举报 2023-02-05

我来回答

在Python中使用Paramiko？在ssh上实现交互式shell,

2条答案

相关问题

热门标签

最新问答