client.load_system_host_keys()  # 从系统默认的"~/.ssh/known_hosts" 加载已信任的主机 host key
client.load_host_keys(filename="...")  # 从自定义路径加载
client.connect(...)

client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
client.connect(...)

from paramiko.client import SSHClient, AutoAddPolicy
from paramiko.config import SSH_PORT
from paramiko.rsakey import RSAKey
from io import StringIO
import paramiko
import getpass


def ssh_by_username_passwd(host, port, user, cmd):
    """「用户名+密码」方式进行SSH远程连接，并执行一条简单命令。
    """
    # 实例化 ssh client 对象
    client = paramiko.SSHClient()

    # 自动添加未知主机到信任主机列表
    client.set_missing_host_key_policy(paramiko.AutoAddPolicy())

    # * 创建SSH连接
    client.connect(
        hostname=host, port=port, username=user, password=getpass.getpass()
    )  # 通过getpass()，在执行脚本时交互输入密码，避免代码中写明文密码！

    # 在远程主机执行命令
    mystdin, mystdout, mystderr = client.exec_command(cmd)  # 结果放入stdout，有错误则放入stderr

    # 执行结果
    ret = [f'[{host}]']
    ret.extend(mystdout.readlines())
    # mystdout.readlines() 返回list，每行都是byte字节串，可decode('utf-8')
    # mystdout.read()  直接返回所有内容，为byte字节串，也可decode('utf-8')
    return_code = mystdout.channel.recv_exit_status()  # 命令执行结果的返回码

    # 关闭连接
    client.close()
    return ret


def ssh_by_keyfile(host, port, user, cmd, private_key_file_path=""):
    """「公私钥」方式进行SSH远程连接，并执行一条简单命令。

    该方式的前提，是需要两台主机间已经SSH授信。
    即本地机器（paramiko脚本所在）的公钥，已经添加到目标远程主机的 ～/.ssh/authorized_keys 文件中。
    所以若是无法公私钥方式远程连接，可能和 .ssh/目录、authorized_keys文件 的权限设置。
    """
    client = paramiko.SSHClient()
    client.set_missing_host_key_policy(paramiko.AutoAddPolicy())

    # 读取私钥文件
    praivate_key = paramiko.RSAKey.from_private_key_file(
        private_key_file_path
    )  # 创建密钥对时，若设置了密码，需指定passwoer参数。
    # praivate_key = paramiko.RSAKey.from_private_key_file(
    #     private_key_file_path, password=getpass.getpass()
    # )

    # 建立连接，指定pkey
    client.connect(hostname=host, port=port, username=user, pkey=praivate_key)

    stdin, stdout, stderr = client.exec_command(cmd)
    ret = stdout.read().decode('utf-8')
    client.close()
    return ret


def ssh_multiple_cmds(host, port, user, cmds=None, private_key_file_path=""):
    """一次SSH会话连接，执行多条命令。

    每条命令若比较复杂，则可以借助 pipes 模块构造复杂 shell 命令行。
        from pipes import quote
        cmd_1 = quote("cat filname.txt")
        cmd_2 = quote("cat 'filname with spaces.txt'")
        cmd_3 = quote("cat filname\ with\ spaces.txt")
        cmd_4 = quote("echo 'danger!'; rm -r /tmp/*")
        cmd_5 = quote("echo `date` && df -hl")
        cmds = [cmd_1, cmd_2, cmd_3, cmd_4]
    """
    client = paramiko.SSHClient()
    client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    praivate_key = paramiko.RSAKey.from_private_key_file(private_key_file_path)
    client.connect(hostname=host, port=port, username=user, pkey=praivate_key)

    ret = []
    for cmd in cmds:
        stdin, stdout, stderr = client.exec_command(cmd)
        stdin.close()
        ret.append(stdout.read().decode('utf-8'))
        stdout.close()
        stderr.close()

    client.close()


def sftp_by_keyfile(
    host,
    port,
    user,
    cmd,
    private_key_file="",
    local_path="",
    remote_path="",
):
    """「公私钥」方式，SFTP上传、下载文件。

    一些SSH的特性，需要借助更底层的对象Transport来实现操作：传输文件、端口转发...
    """
    # 用Transport建立连接
    trans = paramiko.Transport((host, port))
    praivate_key = paramiko.RSAKey.from_private_key_file(private_key_file)
    trans.connect(username=user, pkey=praivate_key)
    # trans.connect(username=user, password=getpass.getpass())  # 密码登录方式则替换为 password 参数进行连接

    # 实例化 ssh client 对象，并指定已连接的 Transport
    client = paramiko.SSHClient()
    client._transport = trans

    stdin, stdout, stderr = client.exec_command(cmd)

    # 从连接的Transport，实例化一个sftp对象。开启单独的 channel
    sftp = paramiko.SFTPClient.from_transport(trans)
    # 上传文件
    sftp.put(localpath=local_path, remotepath=remote_path)
    # # 下载文件
    # sftp.get(localpath=local_path, remotepath=remote_path)

    # 关闭 Transport 对象即可
    trans.close()