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:? # 命令加參數(shù)長(zhǎng)度不能小于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()
