python ssh 库 - paramiko and fabric

标题提到两个第三方库,都是可以实现在 python 中执行 ssh 命令。fabric 是在 paramiko 的基础上封装开发的。所以一般场景下 fabric 会更加容易使用。

paramiko

paramiko 最直接的是提供 SSHClient,呈现同服务器的一个会话,基本满足我们执行远程命令,文件上下传的操作。client 连接远端服务可以通过提供 key 或者秘钥的方式,如果 使用 ssh 秘钥登录(本地生成 ssh 公秘钥, 将公钥追加到服务器登录用户目录的 .ssh/authorized_keys 中),在建立连接时,将传递密码的参数改为传递秘钥参数既可。

先用 pip 安装 paramiko。

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

import paramiko

class ssh_client():

    def __init__(self, host=None,port=22, username=None, password=None, pkey_path=None, timeout=10):
        self.client = paramiko.SSHClient()
        """
        使用xshell登录机器,对于第一次登录的机器会提示允许将陌生主机加入host_allow列表
        需要connect 前调用,否则可能有异常。
        """
        self.client.set_missing_host_key_policy(paramiko.AutoAddPolicy())

        if password != None:
            self.client.connect(hostname=host,port=port, password=password, timeout=timeout)
        elif  pkey_path != None:
            """使用秘钥登录"""
            self.pkey = paramiko.RSAKey.from_private_key_file(pkey_path)
            self.client.connect(hostname=host,port=port,pkey=self.pkey, timeout=timeout)

        self.sftp = self.client.open_sftp()


    def run_cmd(self, cmd):
        _in, _out, _error = self.client.exec_command(cmd)
        return _out.read(),_error.read()

    def put_file(self, local, remote):
        return self.sftp.put(localpath=local, remotepath=remote)

    def get_file(self, local, remote):
        return self.sftp.get(localpath=local, remotepath=remote)

    def __del__(self):
        self.client.close()
        self.sftp.close()


if __name__ == "__main__":

    print("------ paramiko ssh client test -------")
    ## 密码登录
    #client = ssh_client(host='192.168.37.134', username='lcd', password='jklfds')
    ## 秘钥登录
    client = ssh_client(host='192.168.37.134', username='lcd',pkey_path="/home/lcd/.ssh/id_rsa")

    out, err = client.run_cmd('update')
    print("cO : %s"%out.strip('\n'))
    print("cE : %s"%err.strip('\n'))
    if len(err) != 0:
        print("cmd exec error")

    out, err = client.run_cmd('pwd; cd /home/')
    print("cO : %s"%out.strip('\n'))
    print("cE : %s"%err.strip('\n'))
    if len(err) != 0:
        print("cmd exec error")

    print client.get_file(local="./aa", remote="/home/lcd/.bashrc")
    print client.put_file(local="./aa", remote="/home/lcd/bb")

fabric

简单地介绍了 paramiko 后,稍微详细地来了解下 fabric 的使用方法和场景。支持 python(2.7 3.4)版本。

我直接在自己机器上通过 pip 安装,默认的版本是
Successfully installed fabric-2.1.3
然后发现在公司正常运行的脚本到这边是会报错的,import api 失败之类的问题,看了下原因,确实是版本不同导致的,坑爹。。。
可以通过 pip 指定安装旧版本

pip install fabric==1.14.0

1.14.0 版本的 fabric

通过 virtualenv 弄个低版本的临时环境,基本操作如下:

$ mkdir fabric_old-1   
$ cd ./fabric_old-1/
$ virtualenv --no-site-packages old_fabric
$ source old_fabric/bin/activate   ## 切到临时环境
$ pip install fabric==1.14.0            
# ...  fabric 版本为1.14.0 环境
$ deactivate

按照基本需求,封装了几个基本函数,分别是在本地和远程执行命令,上传和下载文件。

#!/usr/bin/env python
# coding=utf-8
# by orientlu
## fabric version 1.14.0
import fabric
from fabric.api import *

class ssh_client(object):

    def __init__(self, hosts_list=None, roles_list=None):

        env.hosts=[]
        env.passwords={}
        env.roledefs={}

        if hosts_list != None:
            for key in hosts_list.keys():
                env.hosts.append(key)
                env.passwords[key] = hosts_list[key]
        if roles_list != None:
            for key in roles_list.keys():
                env.roledefs[key] = roles_list[key]

    def _run_cmd(self, cmd, path=None):
        result = None
        if path != None:
            with cd(path):
                result = run(cmd, shell=False,warn_only=True,quiet=True)
        else:
            result = run(cmd, shell=False,warn_only=True,quiet=True)
        return result


    def remote_cmd(self, cmd, path=None,roles=None, is_parallel=0):
        if roles != None and len(roles) > 0:
            func=fabric.api.roles(*roles)(self._run_cmd)
        else:
            func=self._run_cmd

        if is_parallel == 1:
            func = fabric.api.parallel(func)

        result=execute(func,cmd)
        return result


    def local_cmd(self, cmd, path=None):
        result = None
        if path == None:
            path = './'
        with lcd(path):
            with settings(warn_only=True):
                result = local(cmd, capture=1)
        return result

    def _get_file(self, remote_path=None, file_name=None, local_path='./', alias=None):
        if remote_path != None and file_name != None:
            with cd(remote_path):
                with lcd(local_path):
                    if alias == None:
                        alias = file_name
                    result = get(file_name, alias)
        else:
            result = None
        return result

    def get_file(self, remote_path=None, file_name=None, local_path='./', alias=None, roles=None):
        if roles != None and len(roles) > 0:
            func=fabric.api.roles(*roles)(self._get_file)
        else:
            func=self._get_file

        result=execute(func,remote_path, file_name, local_path, alias)
        return result


    def _put_file(self, src_path='./', file_name=None, dst_path=None, alias=None):
        if file_name != None and dst_path != None:
            with lcd(src_path):
                with cd(dst_path):
                    if alias == None:
                        alias = file_name
                    result = put(file_name, alias)
        else:
            result = None
        return result

    def put_file(self, src_path='./', file_name=None, dst_path=None, alias=None, roles=None, is_parallel=0):
        if roles != None and len(roles) > 0:
            func=fabric.api.roles(*roles)(self._put_file)
        else:
            func=self._put_file

        if is_parallel == 1:
            func = fabric.api.parallel(func)

        result=execute(func,src_path, file_name, dst_path, alias)
        return result



if __name__ == "__main__":
    ## 主机登录信息
    hosts_list = {
            "[email protected]" : "jklfds",
            "[email protected]" : "jklfds",
            "[email protected]" : "jklfds",
            "[email protected]" : "jklfds",
    }

    ## 主机分组,按组执行任务
    roles_list = {
            "role1" : ["[email protected]"],
            "role2" : ["[email protected]", "[email protected]"],
            "role3" : ["[email protected]", "[email protected]"],
            "role4" : ["[email protected]"],
    }


    client = ssh_client(hosts_list, roles_list)

    result = client.remote_cmd("ls | wc -l", roles=["role1", "role4"])
    print result

    result = client.local_cmd("rm ./aa")
    print result

    client.get_file("/home/lcd", ".bashrc", local_path='./', alias='aa', roles=['role1'])

    client.put_file(file_name='aa', dst_path='/home/lcd', alias='cc', roles=['role1'])

参考

  • paramiko 官方主页
  • fabric 官方主页 版本2
  • fabric 中文参考 版本1

你可能感兴趣的:(python ssh 库 - paramiko and fabric)