• python10作业思路及源码:类Fabric主机管理程序开发(仅供参考)


    类Fabric主机管理程序开发

    一,作业要求

    1, 运行程序列出主机组或者主机列表(已完成)

    2,选择指定主机或主机组(已完成)

    3,选择主机或主机组传送文件(上传/下载)(已完成)

    4,充分使用多线程或多进程(已完成)

    5,不同主机的用户名,密码,端口可以不同(已完成)

    6,可向主机或主机组批量发布命令(已完成)

    7,可一次性执行多条操作命令(已完成)

    二,程序文件清单

    三,程序流程简图

    四,程序测试样图

     五,程序核心源码

    #!usr/bin/env python
    # -*- coding:utf-8 -*-
    # auther:Mr.chen
    # 描述:只能运行在MacOS或linux系统下
    
    
    import os,sys,pickle,time
    import paramiko,threading
    
    
    
    sys.path.append('..')
    PATH = os.path.dirname(os.path.abspath(__file__))
    PUT_PATH = PATH.replace('core','Folder/')
    GET_PATH = PATH.replace('core','download/')
    CONF_PATH = PATH.replace('core','conf/')
    semaphore = threading.BoundedSemaphore(1) # 进程锁
    
    def new_Host():
        while True:
            hostname = raw_input("请输入服务器的主机名(输入n=返回上级):")
            if os.path.exists(CONF_PATH+hostname+'_conf'):
                print ("主机名已存在,请重新输入!")
                continue
            if hostname =='n':
                return
            port = raw_input("请输入服务器的ssh端口号(输入n=返回上级):")
            if port == 'n':
                return
            username = raw_input("请输入登陆的用户名(输入n=返回上级):")
            if username == 'n':
                return
            password = raw_input("请输入用户的密码(输入n=返回上级):")
            if password == 'n':
                return
            dic = {
                'hostname':hostname,    # 主机名
                'port':port,            # 端口
                'username':username,    # 用户名
                'password':password,    # 密码
                'status':0              # 状态(0:未激活 1:已激活 2:激活失败)
            }
            if os.path.isdir(GET_PATH + hostname) == False:
                command = 'mkdir ' + GET_PATH + hostname
                os.system(command)
            re = hostmessage_Write(dic)
            if re == True:
                return
            else:
                print ("主机信息存储失败,请检查原因!")
    
    
    
    def delete_Host():
        List = Traverse_folder()
        while True:
            dic = {}
            num = 0
            print ("已存在的主机列表如下:")
            for i in List:
                print ("{0},主机名:{1}".format(str(num+1),i))
                dic[str(num+1)] = i
                num += 1
            choose = raw_input("请输入你想删除的主机索引(输入n=返回上级):")
            if choose == 'n':
                return
            elif choose in dic:
                hostname = dic[choose]
                command = 'rm -f '+CONF_PATH+hostname+'_conf'
                os.system(command)
                print ("删除成功!")
                break
            else:
                print ("您的输入有误!")
    
    
    
    def auto_activeHost():
        text = """
                警告!程序准备开启多线程模式激活主机,请确保:
                1,远程服务器处于开启状态
                2,DNS或本地hosts映射能够解析远程服务器主机名
        """
        while True:
            print (text)
            choose = raw_input("是否确定开始激活远程主机(y/n)?:")
            if choose == 'n':
                return
            elif choose == 'y':
                break
            else:
                print ("你的输入有误!")
        print ("程序开始自动激活远程主机,请稍后...")
        List = Traverse_folder()
        if len(List) == 0:
            print ("请先创建主机!")
            return
        for i in List:
            dic = hostmessage_Read(i)
            t = threading.Thread(target=auto_Active,args=(dic,)) # 创建多线程对象
            t.setDaemon(True)  # 将对象设置为守护线程
            t.start() # 线程开启
        while threading.activeCount() != 1:    # 当前活跃的线程数
            pass
        else:
            print ("所有主机激活完毕!")
            time.sleep(2)
    
    
    def auto_Active(dic):
        ssh = paramiko.SSHClient()  # 创建ssh对象
        ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) #允许连接不在know_hosts文件中的主机
        try:
            ssh.connect(hostname=dic['hostname'],port=int(dic['port']),username=dic['username'],password=dic['password'],timeout=10)
        except Exception,e:
            print ("主机名:{0}的主机激活失败!失败原因:{1}".format(dic['hostname'],e))
            dic['status'] = 2
            hostmessage_Write(dic)
        else:
            print ("主机名:{0}的主机激活成功!".format(dic['hostname']))
            dic['status'] = 1
            hostmessage_Write(dic)
        finally:
            ssh.close()
    
    
    def remote_Host():
        List = Traverse_folder(key='status',value=1)
        if len(List) == 0:
            print ("请先激活主机!")
            return
        while True:
            dic = {}
            print ("已激活主机如下:")
            num = 0
            for i in List:
                print ("{0},主机名:{1}".format(str(num+1),i))
                dic[str(num+1)] = i
                num += 1
            choose = raw_input("请输入你想操控的主机的索引(可多选,n=返回上级):")
            if choose == 'n':
                return
            choose = list(set(choose)) #去重复
            if set(choose) & set(dic.keys()) == set(choose):
                LIST = []
                for i in choose:
                    LIST.append(dic[i])
                remote_Host_control(LIST)
            else:
                print ("您的输入有误!")
    
    
    def remote_Host_control(List):
        help = """
                        help帮助提示:
                1.程序的Folder目录是本地文件目录
                2,输入put向远程主机上传文件
                3,输入get向远程主机下载文件
                4,输入其他直接向远程主机发布命令
        """
        while True:
            print ("正在操控{0}台主机,如下:".format(len(List)))
            for i in List:
                print ("主机名:{0}".format(i))
            command = raw_input("请输入你想执行的命令(输入n=返回上级,输入help获取帮助):>>")
            if command == 'n':
                return
            elif command == 'help':
                print help
            elif command == 'get':
                print ("程序准备下载文件...")
                remote_path = raw_input("请输入想要下载的远程服务器文件绝对路径(例如:/etc/hosts):")
                LIST = remote_path.split('/')
                filename = LIST[len(LIST)-1]
                for i in List:
                    local_path = GET_PATH + i + '/'+filename
                    t = threading.Thread(target=get_Method,args=(i,[remote_path,local_path]))
                    t.setDaemon(True)
                    t.start()
                while threading.activeCount() != 1:
                    pass
                else:
                    print ("命令执行完毕!")
            elif command == 'put':
                print ("程序准备上传文件...")
                while True:
                    filename = raw_input("请输入想上传的文件的文件名:")
                    if os.path.exists(PUT_PATH+filename) == False:
                        print ("文件没有找到,请重新输入!")
                        continue
                    local_path= PUT_PATH+filename
                    remote_path = raw_input("你想将文件保存到远程服务器的哪里?(例如:/etc/):")
                    remote_path = remote_path + '/' + filename
                    for i in List:
                        t = threading.Thread(target=put_Method, args=(i, [local_path,remote_path]))
                        t.setDaemon(True)
                        t.start()
                    while threading.activeCount() != 1:
                        pass
                    else:
                        print ("命令执行完毕!")
                        break
            else:
                for i in List:
                    t = threading.Thread(target=execute_Command,args=(i,command))
                    t.setDaemon(True)
                    t.start()
                while threading.activeCount() != 1:
                    pass
                else:
                    print ("命令执行完毕!")
    
    def put_Method(hostname,Path):
        dic = hostmessage_Read(hostname)
        transport = paramiko.Transport((hostname, int(dic['port'])))
        try:
            transport.connect(username=dic['username'], password=dic['password'])
            sftp = paramiko.SFTPClient.from_transport(transport)
            sftp.put(Path[0], Path[1])
        except Exception, e:
            print ("主机名:{0},上传失败!错误原因:{1}".format(hostname, e))
        else:
            pass
        finally:
            transport.close()
    
    
    def get_Method(hostname,Path):
        dic = hostmessage_Read(hostname)
        transport = paramiko.Transport((hostname,int(dic['port'])))
        try:
            transport.connect(username=dic['username'],password=dic['password'])
            sftp = paramiko.SFTPClient.from_transport(transport)
            sftp.get(Path[0],Path[1])
        except Exception,e:
            print ("主机名:{0},下载失败!错误原因:{1}".format(hostname,e))
        else:
            pass
        finally:
            transport.close()
    
    def execute_Command(hostname,command):
        dic = hostmessage_Read(hostname)
        ssh = paramiko.SSHClient()
        ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        try:
            ssh.connect(hostname=hostname,port=int(dic['port']),username=dic['username'],password=dic['password'])
            command_list = command.strip().split(';')
        except Exception,e:
            semaphore.acquire()     # 线程锁开启
            print ("主机:{0}连接出现异常:{1}".format(hostname,e))
            semaphore.release()     # 线程锁释放
            return
        for com in command_list:
            stdin, stdout, stderr = ssh.exec_command(com)
            error = stderr.read()
            output = stdout.read()
            semaphore.acquire()     # 线程锁开启
            if len(error) != 0:
                print ("主机:{0} 执行{1}命令时出错:{2}".format(hostname,com, error))
            if len(output) != 0:
                print ("主机:{0},执行{1}命令的结果如下:".format(hostname,com))
                print (output)
            semaphore.release()     # 线程锁释放
    
    
    
    
    def Traverse_folder(key = None,value = None):
        '''
        根据条件遍历某文件夹里的全部文件内容,
        找出符合条件的文件返回包含主机名的列表
        如果无条件,则返回包含所有主机名的列表
        :return:LIST
        '''
        LIST = []
        List = os.listdir(CONF_PATH)
        for i in List:
            if i == '__init__.py' or i == '__init__.pyc':
                continue
            else:
                with open(CONF_PATH+i,'r') as f:
                    dic = pickle.load(f)
                if key != None:
                    if dic[key] == value:
                        LIST.append(dic['hostname'])
                else:
                    LIST.append(dic['hostname'])
        return LIST
    
    
    
    def hostmessage_Write(dic):
        with open(CONF_PATH+dic['hostname']+'_conf','w') as f:
            pickle.dump(dic,f)
            return True
    
    
    def hostmessage_Read(hostname):
        if os.path.exists(CONF_PATH+hostname+'_conf'):
            with open(CONF_PATH+hostname+'_conf','r') as f:
                dic = pickle.load(f)
                return dic
    
    
    
    def Main():
        text = """
                欢迎来到Fabric主机管理界面
                    1,创建主机
                    2,删除主机
                    3,自动激活所有主机
                    4,开始远程操控
                    5,退出程序
        """
        while True:
            print text
            choose = raw_input("请输入你的选择:")
            dic = {'1':new_Host,'2':delete_Host,'3':auto_activeHost,'4':remote_Host,'5':Exit}
            if choose in dic:
                dic[choose]()
            else:
                print ("你的输入有误!")
    
    
    
    def Exit():
        print ("程序退出!")
        exit()
    
    
    
    
    
    
    
    if __name__ == "__main__":
        Main()
    

     

  • 相关阅读:
    QuickSort(Java)
    MergeSort(Java)
    Silverlight中Datagrid添加Button列用于控制单行对象
    二叉搜索树(BST)demo
    svn ignore使用方法
    海量数据处理面试题及解决方法
    Android中的单元测试
    Ubuntu下配置Intellij的Android开发环境
    urlrewrite地址重写之后丢失css和js解决方案
    修改后 简单的 TCP server
  • 原文地址:https://www.cnblogs.com/chensiqiqi/p/6241810.html
Copyright © 2020-2023  润新知