一 安装paramiko库

(venv) E:\WebSpider>pip install paramiko

二 代码

import json
import paramiko

# connect函数中,参数是一个主机的IP地址或者是主机名称,
# 在执行这个方法之后,如果成功的连接到服务器,那么就会返回一个sshclient对象
def connect(host):
    # 建立一个SSHClient的对象
    ssh = paramiko.SSHClient()

    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())

    try:
        # 然后就尝试连接服务器,在连接服务器的时候,可以使用两种方式:
        # 一种方式是使用秘钥的方式,也就是参数look_for_keys。
        # 也可以直接使用密码的方式,也就是直接使用参数password,从而最后返回一个连接的对象。
        # ssh.connect(host,username='root',allow_agent=True,look_for_keys=True)
        ssh.connect(host,2022, username='root', password='123456')
        return ssh
    except:
        return None

# 在参数中,一个是args,一个outpath,args表示命令的参数,而outpath表示为可执行文件的路径,
# 例如/usr/bin/ls -l。在其中outpath也就是/usr/bin/ls ,而参数为-l
# 这个方法主要是用来组合命令,将分开的参数作为命令的一部分进行组装。
def command(args, outpath):
    cmd = '%s %s' % (outpath, args)
    return cmd

# 传入的参数一个为连接的对象conn,一个为需要执行的命令cmd,
# 最后得到执行的结果,也就是stdout.read(),最后返回得到的结果
def exec_commands(conn, cmd):
    stdin, stdout, stderr = conn.exec_command(cmd)
    results = stdout.read()
    return results

# 上传文件
# 一个是连接对象conn,一个是上传的文件名称,一个上传之后的文件名称,在此必须写入完整的文件名称包括路径。
# 做法主要是打开一个sftp对象,然后使用put方法进行上传文件,最后关闭sftp连接,最后返回一个上传的文件名称的完整路径
def copy_module(conn, inpath, outpath):
    ftp = conn.open_sftp()
    ftp.put(inpath, outpath)
    ftp.close()
    return outpath

# 执行命令得到结果
# 首先,进行连接服务器,得到一个连接对象,
# 如果连接不成功,那么返回主机名和None,表示没有连接成功,
# 如果连接成功,那么修改文件的执行权限,从而可以执行文件,
# 然后得到执行的命令,
# 最后,进行执行命令,得到结果,
# 将结果用json格式表示返回,从而结果能得到一个美观的json格式,
# 最后和主机名一起返回相关的信息
def excutor(host, outpath, args):
    conn = connect(host)
    if not conn:
        return [host, None]
    # exec_commands(conn,'chmod +x %s' % outpath)
    cmd = command(args, outpath)
    result = exec_commands(conn, cmd)
    result = json.dumps(result.decode(encoding="utf-8"),indent=4,ensure_ascii=False)
    return [host, result]

# 第一步测试命令执行,
# 第二步测试上传文件,
# 第三部测试修改上传文件的权限。
if __name__ == '__main__':
    print(json.dumps(excutor('127.0.0.1', 'ls', ' -l'), indent=4, sort_keys=True))
    print(copy_module(connect('127.0.0.1'), '3_1_1.py', '/root/3_1_1.py'))
    exec_commands(connect('127.0.0.1'), 'chmod +x %s' % '/root/3_1_1.py')

三 结果

[root@master ~]# ll /root/3_1_1.py
-rwxr-xr-x 1 root root 7332 Jan 16 21:36 /root/3_1_1.py
[root@master ~]#

 

Logo

更多推荐