python传文件给堡垒机上远程的另一个机器_python3 实现堡垒机功能（并发执行命令及上传下载文件）...

#Author by Guangboli#_*_ coding:utf-8 _*_

importconfigparser,threading,paramiko,time,os,sys

base_dir= os.path.dirname(os.path.dirname(os.path.abspath(__file__)))

sys.path.append(base_dir)from core importColor_set

color=Color_set.Colorset()

cf=configparser.ConfigParser()

cf.read('config',encoding='utf-8')

secs=cf.sections()defget_group_info():'''获取主机组信息'''

print("主机组列表：")for i insecs:

opts=cf.options(i)

quantity= int(len(opts) / 4)print(i,'[%s]'%quantity)defget_host_ip_list(group_name,quantity):'''获取主机ip列表信息'''i= 1

while i <=quantity:print(cf.get(group_name,'ip%s'%i))

i+=1

defcmd_run(ip,username,password,cmd):

ssh=paramiko.SSHClient()

ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())try:

ssh.connect(hostname=ip, port=22, username=username, password=password)

stdin, stdout, stderr= ssh.exec_command(cmd) #执行命令,不可执行类似vim，top watch命令

result = stdout.read().decode() #获取结果

tips = color.red('- - - - - %s - - - - - -'.center(20) %ip)print(tips)print(result,stderr.read().decode())exceptException as e:print('%s 主机发生异常：'%ip,e)

ssh.close()deftransport_put_file(ip,username,password,cmd):try:

transport= paramiko.Transport((ip, 22))

transport.connect(username=username, password=password)

sftp=paramiko.SFTPClient.from_transport(transport)

sftp.put(local_file,remote_dir+'/'+put_filename)exceptException as e:print(print('%s 主机发生异常：'%ip,e))

transport.close()deftransport_get_file(ip,username,password,cmd):

transport= paramiko.Transport((ip, 22))try:

transport.connect(username=username, password=password)

sftp=paramiko.SFTPClient.from_transport(transport)

sftp.get(remote_file,local_dir+'\\'+ip+'-'+get_filename) #Linux 需要将\\改为/

exceptException as e:print(print('%s 主机发生异常：'%ip,e))

transport.close()defmake_treading(func,quantity,group_name,cmd):

i= 1

while i <=quantity:

ip= cf.get(group_name, 'ip%s' %i)

username= cf.get(group_name, 'username%s' %i)

password= cf.get(group_name, 'password%s' %i)

thread= threading.Thread(target=func,args=(ip,username,password,cmd))

thread.setDaemon(True)

thread.start()

i+=1

defwelcome():'''show group and host ip information'''get_group_info()globalopts,quantity,group_namewhileTrue:

group_name= input('请输入组名(退出请按q or Q)：').strip().upper()if group_name == 'Q':print('欢迎再次使用，谢谢....')

exit()elif group_name not insecs:print("Wrong input,please confirm!")else:break

print(group_name,'组下的主机：')

opts=cf.options(group_name)

quantity= int(len(opts) / 4) #hosts quantity

get_host_ip_list(group_name,quantity)defmenu_line2():globallocal_dir,local_file,remote_file,remote_dir,get_filename,put_filenameprint('''usage:

上传文件：put 本地文件 远程目录

下载文件：get 远程文件 本地目录

退出输入q or Q

examples：

put test.txt /tmp

get /tmp/test.txt /data''')whileTrue:

cmd= input("请输入操作：").strip()

args=cmd.split()if cmd == 'q' or cmd == 'Q':

menu_line1()elif len(args) != 3:print("输入有误，请重新输入！")

menu_line2()else:if args[0] == 'put':

local_file= args[1]

put_filename=args[1].split('/')[-1]

remote_dir= args[2]

func=transport_put_file

make_treading(func, quantity, group_name, cmd)while threading.active_count() != 1:

time.sleep(0.1)else:continue

elif args[0] == 'get':

remote_file= args[1]

get_filename= args[1].split('/')[-1]

local_dir= args[2]

func=transport_get_file

make_treading(func, quantity, group_name, cmd)while threading.active_count() != 1:

time.sleep(0.1)else:continue

else:print("输入格式有误，请重新输入！")

menu_line2()defmenu_line1():'''commadns menu'''func=cmd_runwhileTrue:print('1、执行命令\n2、传输文件\n')

choice= input("请选择操作：").strip()if choice == '1':

cmd= input("请输入命令：").strip()print('%s 命令执行结果如下：' %cmd)

make_treading(func,quantity,group_name,cmd)while threading.active_count() != 1:#print(threading.active_count())

time.sleep(0.1)else:continue

elif choice == '2':

menu_line2()elif choice == 'q' or choice == 'Q':

welcome()else:continue

defrun():'''main function'''

print('welcome.....')

welcome()

menu_line1()if __name__ == '__main__':

run()