批量主机管理工具_python

一、主要功能

本服务器管理工具主要实现以下两种功能：
（1）批量执行服务器操作命令
（2）批量上传下载文件

二、程序目录

 三、代码

start.py:

#/usr/bin/env python
#_*_ coding:utf-8 _*_
#Author:chenxiaozan
import re,os,sys
import paramiko
BASE_DIR=os.path.dirname(os.path.dirname(__file__))
split=os.path.sep
sys.path.append(BASE_DIR)

from core import servermanage


if __name__ == '__main__':
    obj=servermanage.Servermanage()
    obj.interaction()

hosts:

#_*_coding:utf-8 _*_
#[gruop_name]
#host_ip password

[test]
192.168.1.1  123456
192.168.102.1  123456

[test3]
192.168.102.1  123456

settings:

#/usr/bin/env python
#_*_ coding:utf-8 _*_
#Author:chenxiaozan
import re,os,sys

BASE_DIR=os.path.dirname(os.path.dirname(__file__))
split=os.path.sep
sys.path.append(BASE_DIR+split)



#主机配置文件
host_file=BASE_DIR+split+'conf'+split+'hosts'

msg.py:

#/usr/bin/env python
#_*_ coding:utf-8 _*_
#Author:chenxiaozan
import re,os,sys
BASE_DIR=os.path.dirname(os.path.dirname(__file__))
split=os.path.sep
sys.path.append(BASE_DIR+split)
from conf import settings

def read_msg():
    '''
    读取主机配置文件
    :return:
    '''
    count = 1
    f=open(settings.host_file,'r')
    host_data_dict={}
    for line in f:
        line=line.strip()
        if line!='': #去除空行
            begin_wiht=re.match('#',line)  #去除注释行
            if not begin_wiht:
                group=re.findall('[(.+)]', line) #判断是否是组名
                if group:
                    group_name=group[0]
                    host_data_dict[group_name] = {}
                else:
                    line_list=re.split(' +',line)
                    if len(line_list)==2:
                        if line_list[0]  not in host_data_dict[group_name]:
                            #print('本行可添加到主机信息')
                            host_data_dict[group_name][line_list[0]]=line_list[1]
                    elif len(line_list)==0:
                        pass
                    else :
                        print('hosts文件第%s行服务器信息为空行或填写格式错误,本行相关服务器不执行本次操作'%count)
        count+=1
    return host_data_dict




if __name__ == '__main__':

    host_data_dict=read_msg()
    print(host_data_dict)

servermanage.py:

#/usr/bin/env python
#_*_ coding:utf-8 _*_
#Author:chenxiaozan
import re,os,sys
import paramiko
BASE_DIR=os.path.dirname(os.path.dirname(__file__))
split=os.path.sep
sys.path.append(BASE_DIR+split)
import threading
import paramiko

from conf import settings
from core import msg


class Servermanage(object):
    def __init__(self):
        self.host_data_dict=msg.read_msg()
        self.lock=threading.Lock()
    def interaction(self):
        print('根据您填写的hosts文件，有以下分组：')
        for i in self.host_data_dict:
            print(i)
        while True:
            self.group_choice=input('请选择你需要操作的组名>>').strip()
            if self.group_choice in self.host_data_dict:
                    print('您所选择的主机组拥有以下主机：')
                    for host_ip in self.host_data_dict[self.group_choice]:
                        print(host_ip)
                    print('''用法：
                    1、直接输入linux命令：例如 df              (注：不能执行动态命令，如top)
                    2、上传下载文件： put 上传文件 目标文件名   （注：本地上传先存放数据在db文件夹）
                                    get 下载文件 目标文件名    （注：远程下载存放数据在db文件夹,并命名追加ip）
                                      ''')
                    while True:
                        self.cmd_input=input('请输入命令(r返回，q退出)>>').strip()
                        if self.cmd_input=='':
                            continue
                        elif self.cmd_input=='r':
                            break
                        elif self.cmd_input=='q':
                            exit()
                        else:
                            self.cmd_input_list=self.cmd_input.split(' ')
                            if self.cmd_input_list[0]=='put' or self.cmd_input_list[0]=='get' and len(self.cmd_input_list)==3:
                                self.mul_thread(self.scp)
                            elif self.cmd_input_list[0]=='put' or self.cmd_input_list[0]=='get' and len(self.cmd_input_list)!=3:
                                print('您输入的scp命令不符合格式')

                            else:
                                self.mul_thread(self.cmd)

            elif self.group_choice=='':
                continue
            else:
                print('您输入的组名不存在')


    def mul_thread(self,run):
        thread_list=[]
        for ip in self.host_data_dict[self.group_choice]:
            t=threading.Thread(target=run,args=(ip,self.host_data_dict[self.group_choice][ip]))
            t.start()
            thread_list.append(t)
        for i in thread_list:
            i.join()

    def cmd(self,ip,passwd):
        ssh=paramiko.SSHClient()
        ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy)
        ssh.connect(hostname=ip,port=22,username='root',password=passwd)
        stdin, stdout, stderr = ssh.exec_command(command=self.cmd_input)
        res = stdout if stdout else stderr
        self.lock.acquire()
        print('===========%s==========='%ip)
        print(res.read().decode()) #在这里进行decode()的时候不同服务器可能出现不同code
        self.lock.release()
        ssh.close()

    def scp(self,ip,passwd):
        db_path=BASE_DIR+split+'db'
        transport=paramiko.Transport((ip,22))
        transport.connect(username='root',password=passwd)
        sftp=paramiko.SFTPClient.from_transport(transport)
        try:
            if self.cmd_input_list[0]=='get':
                sftp.get(self.cmd_input_list[1],'%s%s%s_%s'%(db_path,split,self.cmd_input_list[2],ip))
            else:
                sftp.put('%s%s%s'%(db_path,split,self.cmd_input_list[1]), self.cmd_input_list[2])
        except Exception as e:
            print('请检查要上传或下载的文件存不存在')

if __name__ == '__main__':
    obj=Servermanage()
    obj.interaction()

readme：

#/usr/bin/env python
#_*_ coding:utf-8 _*_
#Author:chenxiaozan

功能说明：
本服务器管理工具主要实现以下两种功能：
（1）批量执行服务器操作命令
（2）批量上传下载文件

使用说明：
（1）批量执行命令时，只需要直接敲如命令即可。不能执行动态命令，如top
（2）使用批量上传下载文件时，上传之前需要将你要上传的文件放在db文件夹内。下载文件会统一下载在db文件夹内，并将文件名后面追加主机ip进行区分