python系列：远程SSH登录并执行命令，获取结果

最新推荐文章于 2024-07-08 01:37:07 发布

原创最新推荐文章于 2024-07-08 01:37:07 发布 · 1.7k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#ssh #运维

【Python常见的1000个问题】专栏收录该内容

7 篇文章

订阅专栏

本文介绍如何使用Python通过paramiko库实现在机器A上以用户名'nio4444'登录机器B（IP为10.10.10.10，端口22），并在B上执行指定的shell命令，如'cd /tmp/ && tar czf fstab.tgz fstab'。

需求：

机器A上执行Python代码登录连接到机器B，然后在机器B上执行命令

实现

#!/usr/bin/env python
#coding=utf-8

import paramiko, getpass,sys,traceback

class sshDemo():
    def login_by_passwd(self, ip, port, username, passwd):
        self.ip = ip
        self.port = port
        self.username = username
        self.passwd = passwd
        self.pkey = None
    
    def login(self, username, key_path, passwd):
        try:
            self.pkey=paramiko.RSAKey.from_private_key_file(key_path)
        except paramiko.PasswordRequiredException:
            if not passwd:
                passwd = getpass.getpass('RSA key password: ')
            self.pkey = paramiko.RSAKey.from_private_key_file(key_path, passwd) 
         
    def ssh(self,shell):
        try:
            ssh = paramiko.SSHClient()
            ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
            if self.pkey:
                ssh.connect(self.ip, self.port, self.username, compress = True, pkey= self.pkey)
            else:
                if not self.passwd:
                    self.passwd = getpass.getpass('input password: ')
                ssh.connect(self.ip,self.port,self.username, self.passwd)
            stdin, stdout, stderr = ssh.exec_command(shell)
            res = stdout.readlines()
            ssh.close()
            return res
        except:
            type, value, tb = sys.exc_info()
            return traceback.format_exception(type, value, tb)
        
 

if __name__ == '__main__': 
    myssh = sshDemo()
    myssh.login("10.10.10.10",22,"nio4444","nio4444") 
    myssh.ssh("cd /tmp/ && tar -zcf fstab.tgz fstab")