【Python-自动化】paramiko模块

文章目录

  • 运维自动化Python
  • 一、模块介绍
  • 二、模块应用
  • 1.使用paramiko模块,通过ssh协议连接服务器
  • 2.解决首次连接known_hosts问题
  • 3、执行命令exec_command方法
  • 扩展:
  • 使用try异常捕获
  • 4、多台服务器执行命令
  • 5、从服务器上传下载文件–SFTPClient方法
  • 6、多台服务器上传下载文件
  • 扩展-文件操作
  • 可扩展练习

  • 运维自动化Python

    paramiko模块


    一、模块介绍

    模块:paramiko

    模块作用:
    1、通过ssh协议远程执行命令

    2、文件上传下载

    安装模块:

    pip install paramiko
    

    二、模块应用

    1.使用paramiko模块,通过ssh协议连接服务器

    #导入paramiko,(导入前需要先在环境里安装该模块)
    import paramiko
    
    #定义函数ssh,把操作内容写到函数里
    def sshExeCMD():
        #定义一个变量ssh_clint
        ssh_client=paramiko.SSHClient()
        #使用cnnect类来连接服务器
        ssh_client.connect(hostname="192.168.1.110", port="22", username="songxk", password="123123")
    
    #通过判断模块名运行上边函数
    if __name__ == '__main__':
        sshExeCMD()
    
    

    这时会报错,提示在服务器的known_hosts中没有,这个就是连接服务器的时候那个首次连接需要输入一个yes保存证书。

    2.解决首次连接known_hosts问题

    通过这个set_missing_host_key_policy方法用于实现登录是需要确认输入yes,否则保存

    #导入paramiko,(导入前需要先在环境里安装该模块)
    import paramiko
    
    #定义函数ssh,把操作内容写到函数里
    def sshExeCMD():
        #定义一个变量ssh_clint
        ssh_client=paramiko.SSHClient()
        #通过这个set_missing_host_key_policy方法用于实现登录是需要确认输入yes,否则保存
        ssh_client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        #使用cnnect类来连接服务器
        ssh_client.connect(hostname="192.168.1.110", port="22", username="songxk", password="123123")
    
    
    #通过判断模块名运行上边函数
    if __name__ == '__main__':
        sshExeCMD()
        
    

    连接成功, 没有报错:

    3、执行命令exec_command方法

    使用exec_command执行命令会返回三个信息:

    1、标准输入内容(用于实现交互式命令)

    2、标准输出(保存命令的正常执行结果)

    3、标准错误输出(保存命令的错误信息)

    可以通过三个变量来接受,然后使用print输出到屏幕查看

    #导入paramiko,(导入前需要先在环境里安装该模块)
    import paramiko
    
    #定义函数ssh,把操作内容写到函数里
    def sshExeCMD():
        #定义一个变量ssh_clint使用SSHClient类用来后边调用
        ssh_client=paramiko.SSHClient()
        #通过这个set_missing_host_key_policy方法用于实现登录是需要确认输入yes,否则保存
        ssh_client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        #使用cnnect类来连接服务器
        ssh_client.connect(hostname="192.168.1.110", port="22", username="songxk", password="123123")
        #使用exec_command方法执行命令,并使用变量接收命令的返回值并用print输出
        stdin, stdout, stderr = ssh_client.exec_command("hostname")
        print(stdout.read())
    #通过判断模块名运行上边函数
    if __name__ == '__main__':
        sshExeCMD()
    


    展示内容以python的格式b’内容\n’的格式展示,如果后边需要对回显处理,可以直接用str把内容输出为字符串格式如下:

    print(str(stdout.read()))
    

    扩展:

    使用try异常捕获

    import paramiko
    import sys
    #定义函数ssh,把操作内容写到函数里
    def sshExeCMD():
        #定义一个变量ssh_clint使用SSHClient类用来后边调用
        ssh_client=paramiko.SSHClient()
        #通过这个set_missing_host_key_policy方法用于实现登录是需要确认输入yes,否则保存
        ssh_client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        #使用try做异常捕获
        try:
        #使用cnnect类来连接服务器
            ssh_client.connect(hostname="192.168.1.110", port="22", username="songx1k", password="123123")
        #如果上边命令报错吧报错信息定义到err变量,并输出。
        except Exception as err:
            print("服务器链接失败!!!")
            print(err)
            #如果报错使用sys的exit退出脚本
            sys.exit()
        #使用exec_command方法执行命令,并使用变量接收命令的返回值并用print输出
        stdin, stdout, stderr = ssh_client.exec_command("df -hT")
        print(str(stdout.read()))
    #通过判断模块名运行上边函数
    if __name__ == '__main__':
        sshExeCMD()
    

    4、多台服务器执行命令

    注意:给函数传参,需要在函数括号里写上接收的参数

    #导入paramiko,(导入前需要先在环境里安装该模块)
    import paramiko
    import sys
    #定义函数ssh,把操作内容写到函数里,函数里接收参数(写在括号里),其中port=是设置一个默认值如果没传就用默认
    def sshExeCMD(ip, username, password, port=22):
        #定义一个变量ssh_clint使用SSHClient类用来后边调用
        ssh_client=paramiko.SSHClient()
        #通过这个set_missing_host_key_policy方法用于实现登录是需要确认输入yes,否则保存
        ssh_client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        #使用try做异常捕获
        try:
        #使用cnnect类来连接服务器
            ssh_client.connect(hostname=ip, port=port, username=username, password=password)
        #如果上边命令报错吧报错信息定义到err变量,并输出。
        except Exception as err:
            print("服务器链接失败!!!"% ip)
            print(err)
            #如果报错使用sys的exit退出脚本
            sys.exit()
        #使用exec_command方法执行命令,并使用变量接收命令的返回值并用print输出
        #这里也可以把命令做成参数传进来
        stdin, stdout, stderr = ssh_client.exec_command("hostname")
        #使用decode方法可以指定编码
        print(stdout.read().decode("utf-8"))
    #通过判断模块名运行上边函数
    if __name__ == '__main__':
        #定义一个字典,写服务器信息
        servers = {
            #以服务器IP为键,值为服务器的用户密码端口定义的字典
            "192.168.1.110": {
                "username": "songxk",
                "password": "123123",
                "port": 22,
            },
            "192.168.1.123": {
                "username": "root",
                "password": "123123",
                "port": 22,
            },
        }
        #使用items方法遍历,使用ip 和info把字典里的键和值遍历赋值,传给函数sshExeCMD
        for ip, info in servers.items():
            # 这里的info也就是上边定义的ip对应值的字典,使用get获取这个字典里对应username键对应的值,赋值给变量username传给函数中使用
            sshExeCMD(
                ip=ip,
                username=info.get("username"),
                password=info.get("password"),
                port=info.get("port")
            )
    

    5、从服务器上传下载文件–SFTPClient方法

    '''
    通过ssh协议在服务器上传下载文
    时间:2022-04-09
    '''
    import paramiko
    
    def sshfileftp():
        #与服务器创建ssh连接,transport方法建立通道,以元组的方式歇服务器信息
        ssh_conn = paramiko.Transport(("192.168.1.110", 60317))
        ssh_conn.connect(username="songxk", password="123123")
    
        #创建连接后,使用sftpclient类和from_transport(括号里写上边创建的Transport通道)基于上边ssh连接创建一个sftp连接,定义成ftp_client变量后边方便引用
        ftp_client = paramiko.SFTPClient.from_transport(ssh_conn)
        #下载文件
        #ftp_client.get("目标文件", r"保存位置,写到文件名")
        ftp_client.get("/etc/fstab", r"C:\Users\Administrator.USER-CH3G0KO3MG\Desktop\test\fstab")
        '''
        上传文件
        ftp_client.put(r"C:\Users\Administrator.USER-CH3G0KO3MG\Desktop\test\fstab", "/etc/fstab")
        '''
    
        #关闭ssh连接
        ssh_conn.close()
    
    if __name__ == '__main__':
        sshfileftp()
    

    6、多台服务器上传下载文件

    和批量对服务器执行命令原理一样,使用字典写服务器信息,通过for循环处理把变量分别传给写好的上传函数。

    '''
    批量通过ssh协议在服务器上传文件
    时间:2022-04-09
    localfile:本地文件名
    remotedir:服务器目录名
    '''
    
    import paramiko
    #后续需要用到os模块的方法
    import os
    
    #定义函数并且接收变量
    def sshPutfile(ip, port, username, password, localfile, remotedir):
        #获取源文件的文件名,把进来的localfile变量的值处理只剩文件名
        file_name = os.path.basename(localfile)
    
        #处理服务器目录名,如果输入的目录名没有带后边的/(不是以/结尾)则添加一个,方便后边拼接文件名
        if not remotedir.endswith("/"):
            remotedir = remotedir + "/"
        dest_file_name = remotedir + file_name
    
        #创建ssh连接
        ssh_conn = paramiko.Transport((ip, port))
        ssh_conn.connect(username=username, password=password)
        #创建ftp工具(变量)
        ftp_client = paramiko.SFTPClient.from_transport(ssh_conn)
        #上传文件
        ftp_client.put(localfile, dest_file_name)
        # 关闭ssh连接
        ssh_conn.close()
    
    if __name__ == '__main__':
        #定义一个字典,写服务器的信息
        servers = {
            #以服务器IP为键,值为服务器的用户密码端口定义的字典
            "192.168.1.110": {
                "username": "songxk",
                "password": "123123",
                "port": 60317,
            },
            "192.168.106.71": {
                "username": "root",
                "password": "123123",
                "port": 22,
            },
        }
        source_file = input("请输入源文件路径(绝对路径):")
        remote_dir = input("服务器路径(绝对路径):")
        for ip, info in servers.items():
            sshPutfile(
                ip=ip,
                port=info.get("port"),
                username=info.get("username"),
                password=info.get("password"),
                localfile=source_file,
                remotedir=remote_dir
            )
    

    效果:

    扩展-文件操作

    对文件的操作除了上传下载还有其他很多操作


    可扩展练习

    1、 输入的文件是否存在

    2、服务器目录是否存在

    3、上传后文件是否完整(利用md5判断,本地文件的md5和Linux命令md5sum)

    来源:宋晓坤

    物联沃分享整理
    物联沃-IOTWORD物联网 » 【Python-自动化】paramiko模块

    发表回复