通过python脚本实现ansible远程命令

import paramiko

import getpass

import sys

import threading

import os

def rcmd(host, port=22, user='root', passwd=None, cmd=None):

    ssh = paramiko.SSHClient()

    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())

    ssh.connect(host, port=port, username=user, password=passwd)

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

    out = stdout.read()

    err = stderr.read()

    if out:

        print(f'[{host}] \033[32;1mOUT\033[0m:\n{out.decode()}')

    if err:

        print(f'[{host}] \033[31;1mERROR\033[0m:\n{err.decode()}')

if __name__ == '__main__':

    # rcmd('127.0.0.1', passwd='redhat', cmd='id root; id zhangsan')

    if len(sys.argv) < 3:  # 命令加参数长度不能小于3

        print(f'Usage: {sys.argv[0]} ipfile commands')

        exit(1)

    if not os.path.isfile(sys.argv[1]):  # ip地址文件必须存在

        print(f'No such file: {sys.argv[1]}')

        exit(2)

    ipfile = sys.argv[1]

    cmd = ' '.join(sys.argv[2:])  # 拼接出命令

    passwd = getpass.getpass()

    with open(ipfile) as fobj:

        for line in fobj:

            ip = line.strip()  # 去除行尾的的\n

            t = threading.Thread(target=rcmd, args=(ip, 22, 'root', passwd, cmd))

            t.start()