python大佬养成计划----paramiko实现SSH

摘要：模块为的缩写，由的网络小组所制定为建立在应用层基础上的安全协议。是目前较可靠，专为远程登录会话和其他网络服务提供安全性的协议。利用该模块，可以方便的进行连接和协议进行文件传输。

SSH 为 Secure Shell 的缩写，由 IETF 的网络小组（Network Working Group）所制定；SSH 为建立在应用层基础上的安全协议。SSH 是目前较可靠，专为远程登录会话和其他网络服务提供安全性的协议。利用 SSH 协议可以有效防止远程管理过程中的信息泄露问题.

paramiko是用python语言写的一个模块，遵循SSH2协议，支持以加密和认证的方式，进行远程服务器的连接。paramiko支持Linux, Solaris, BSD, MacOS X, Windows等平台通过SSH从一个平台连接到另外一个平台。利用该模块，可以方便的进行ssh连接和sftp协议进行sftp文件传输。

#基于ssh,用于连接远程服务器做操作：远程执行命令，上传或下载文件
import paramiko

#创建一个ssh对象
client = paramiko.SSHClient()
#2.解决问题：首次连接，会出现
# Are you sure you want to continue connecting (yes/no)? yes
# 自动选择yes

# 允许连接不在know_hosts文件中的主机
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())

#3.连接服务器
client.connect(hostname="172.25.254.19",port=22,username="root",password="westos")

#4.执行操作
stdin,stdout,stderr = client.exec_command("hostname")#标准输入，标准输出，标准错误输出。
#Execute a command on the SSH server.  A new `.Channel` is opened and
# the requested command is executed.  The command"s input and output
# streams are returned as Python ``file``-like objects representing
# stdin, stdout, and stderr.

#5.获取命令的执行结果
res = stdout.read().decode("utf-8")#使结果具有可读性

print(res)

#6.断开连接
client.close()

批量连接host.txt文件中的主机，返回执行结果
格式：172.25.254.1:22:root:westos

import paramiko
with open("host.txt") as f: #保证host.txt文件在当前目录下
    hostinfos = f.readlines()   #列表形式，["172.25.254.1:22:root:westos
", "172.25.254.2:22:root:westos
", "172.25.254.3:22:root:westos
", "172.25.254.19:22:root:westos
"]

    for hostinfo in hostinfos:
        hostinfo = hostinfo.strip() #去掉空格,字符串格式,172.25.254.2:22:root:westos
        print("正在连接%s主机" %(hostinfo.split(":")[0]))
        hostname,port,username,passwd = hostinfo.split(":")
        try:
            client = paramiko.SSHClient()
            client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
            client.connect(hostname=hostname,port=port,username=username,password=passwd)
            stdin,stdout,stderr = client.exec_command("hostname")
            res = stdout.read().decode("utf-8")
            print("结果为：",res)
        except Exception as e :
            print("Connection is failed,the reason is :",e)
        finally:
            client.close()
            print("连接结束")

免密登录远程主机
首先在需要连接的主机上生成一对公钥和私钥，本机获取到需要连接的主机的私钥时，就可以通过公私钥配对，登陆远程主机。
这里需要id_rsa存放你所连接的主机的私钥

import paramiko
from paramiko.ssh_exception import NoValidConnectionsError, AuthenticationException

def conn(cmd,hostname,port=22,username="root"):
    client = paramiko.SSHClient()
    private_key = paramiko.RSAKey.from_private_key_file("id_rsa")#id_rsa存放私钥
    client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    try:
        client.connect(hostname=hostname,
            port=port,username=username,pkey=private_key)
    except NoValidConnectionsError as e:
        print("...连接失败...")
    except AuthenticationException as e:
        print("...密码错误...")
    else:
        stdin, stdout, stderr = client.exec_command(cmd)
        result = stdout.read().decode("utf-8")
        print(result)

    finally:
        client.close()


if __name__=="__main__":
    for count in range(13,20):
        hostname = "172.25.254.%s" %(count)
        print("正在连接主机：",hostname)
        conn("hostname",hostname)
    print("...连接结束...")

sftp是Secure File Transfer Protocol的缩写，安全文件传送协议。可以为传输文件提供一种安全的网络的加密方法。

import paramiko
tran = paramiko.Transport("172.25.254.19",22)
tran.connect(username="root",password="westos")


sftp = paramiko.SFTPClient.from_transport(tran)
#class SFTPClient(BaseSFTP, ClosingContextManager)
#SFTP client object.
# Used to open an SFTP session across an open SSH `.Transport` and perform
# remote file operations.
# Instances of this class may be used as context managers.
sftp.put("/home/kiosk/PycharmProjects/day18/07_pratice.py","/mnt/practice.py")
sftp.get("/mnt/passwd","hallo")
tran.close()

使paramiko模块执行自己想要的操作

import paramiko
import os
from paramiko.ssh_exception import NoValidConnectionsError, AuthenticationException, SSHException

class SshRrmote(object):
    def __init__(self,cmd,hostname,port,username,passwd):
        self.hostname = hostname
        self.passwd = passwd
        self.cmd = cmd
        self.username = username
        self.port = port
    def run(self):
        """默认调用的内容"""
        # cmd hostname
        # put local_file remote_file
        # get remote_file local_file
        cmd_str = self.cmd.split()[0]  # cmd
        # 类的反射, 判断类里面是否可以支持该操作?
        if hasattr(self, "do_" + cmd_str):  # do_cmd
            getattr(self, "do_" + cmd_str)()
        else:
            print("目前不支持该功能")
    def do_cmd(self):
        client = paramiko.SSHClient()
        try:
            client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
            client.connect(hostname=self.hostname,port=int(self.port),username=self.username,password=self.passwd)

        except NoValidConnectionsError as e:
            print("...连接失败...")
        except AuthenticationException as e:
            print("...密码错误...")
        else:
            cmd = " ".join(self.cmd.split()[1:])
            stdin, stdout, stderr = client.exec_command(cmd)
            result = stdout.read().decode("utf-8")
            print("执行结果",result)
        finally:
            print("断开%s的连接" %(self.hostname))
            client.close()
    def do_get(self):
        #有待改进，因为连接多个主机时，会覆盖文件
        print("开始下载")
        try:
            trans = paramiko.Transport(self.hostname,int(self.port))
            trans.connect(username=self.username,password=self.passwd)
            print("hello")
        except SSHException as e:
            print("连接失败")
        else:
            sftp = paramiko.SFTPClient.from_transport(trans)
            cmd = self.cmd.split()[1:]
            if len(cmd)==2:
                sftp.get(cmd[0],cmd[1])
                print("下载文件%s成功，并保存为%s" %(cmd[0],cmd[1]))
            else:
                print("参数有误")
            trans.close()
    def do_put(self):
        # put /tmp/passwd /tmp/passwd  # 将本机的/tmp/passwd文件上传到远程主机的/tmp/passwd;
        print("开始上传")   #注意你使用的用户是否为kiosk
        try:
            trans = paramiko.Transport(self.hostname, int(self.port))
            trans.connect(username=self.username, password=self.passwd)

        except SSHException as e:
            print("连接失败")
        else:
            sftp = paramiko.SFTPClient.from_transport(trans)
            cmd = self.cmd.split()[1:]
            if len(cmd) == 2:
                sftp.put(cmd[0],cmd[1])
                print("上传文件%s成功，并保存为%s" %(cmd[0], cmd[1]))
            else:
                print("参数有误")
            trans.close()

#1.选择要操作的主机组：mysql,web,ftp
#       主机信息怎么存？将不同的主机信息存放在不同的文件中
#2.根据选择的主机组，显示包含的主机IP/主机名
#3.让用户确认信息，选择需要批量执行的命令
#   -cmd shell命令
#   -put 本地文件 远程文件
#   -get 远程文件 本地文件

def main():
    groups = [file.rstrip(".conf") for file in os.listdir("conf")]
    print("主机组显示".center(50,"*"))
    [print("	",item) for item in groups]
    choiceGroup = input("请选择批量操作的主机组(eg:web):")
    with open("conf/"+choiceGroup+".conf") as f:
        info = f.readlines()
        print("批量执行脚本".center(50, "*"))
        while True:
            cmd = input(">>").strip()
            if cmd:
                if cmd =="exit":
                    print("连接执行结束")
                    break
                for item in info:
                    item=item.strip()
                    print(item.split(":")[0].center(50,"-"))
                    hostname,port,username,passwd = item.split(":")
                    ssh = SshRrmote(cmd,hostname,port,username,passwd)
                    ssh.run()

main()