tornado+ansible+twisted+mongodb运维自动化系统开发(二)

esdghj

源码:

    #!/usr/bin/env python  
    #coding:utf-8  
    import os.path  
    import tornado.locale  
    import tornado.httpserver  
    import tornado.ioloop  
    import tornado.options  
    import tornado.web  
    from tornado.options import define, options  
    import pymongo  
      
    define("port", default=8000, help="run on the given port", type=int)  
      
    class Application(tornado.web.Application):  
        def __init__(self):  
            #初始化一些东西  
            handlers = [  
                #url匹配  
                (r"/", MainHandler),  
                (r"/index.html", MainHandler),  
                (r"/add.html", AddHandler),  
                (r"/listhost.html",List_hostHandler),  
                (r"/delete.html", delete_hostHandler),  
                (r"/module_action.html", Module_actionHandler),  
      
            ]  
            settings = dict(  
                #程序设置,字典形式  
                template_path=os.path.join(os.path.dirname(__file__), "templates"),  
                #设置模板文件路径  
                static_path=os.path.join(os.path.dirname(__file__), "static"),  
                #设置静态文件路径,如css\jpg\gif等  
                # ui_modules={"Book": BookModule},  
                #设置ui模块,可以用字典添加多个  
                debug=True,  
            )  
            conn = pymongo.Connection("localhost", 27017)  
            #初始化数据库连接  
            self.db = conn["waitfish"]  
            #选择mongodb集合  
            tornado.web.Application.__init__(self, handlers, **settings)  
            #传入设置配置  
      
    class MainHandler(tornado.web.RequestHandler):  
        #主页函数方法  
        def get(self):  
            #设置httpget方法函数  
            self.render(  
                "index.html",  
            )  
      
      
    class AddHandler(tornado.web.RequestHandler):  
        #添加主机页面  
        def get(self):  
            self.render(  
                "add.html",  
            )  
      
      
      
      
    class List_hostHandler(tornado.web.RequestHandler):  
        #主机列表页面,get方式现实全部主机  
        def get(self, *args, **kwargs):  
            coll = self.application.db.waitfish  
            hosts = coll.find()  
            self.render(  
                "listhost.html",  
                hosts = hosts  
            )  
      
        def post(self):  
            #post方法现实post的主机  
            coll = self.application.db.waitfish  
            #初始化数据库连接  
            hostname = self.get_argument('hostname')  
            #从post中获取主机名  
            ipadd = self.get_argument('ipadd')  
            #获取主机ip地址  
            username = self.get_argument('username')  
            #获取主机用户名  
            password = self.get_argument('password')  
            #获取密码  
            post_dic = {'hostname':hostname, 'ipadd':ipadd, 'username':username, 'password':password}  
            #生成要存入数据库的内容  
            hosts = coll.find({'hostname':hostname})  
            #根据主机名判断是否已经存在该主机  
            if hosts:  
                #如果不存在  
                import ansible.runner  
                #对主机进行初始化,复制公钥到受管主机,(添加ip地址和主机名对到本机的hosts文件和ansible的hosts文件)  
                runner_copy_autherized_keys = ansible.runner.Runner(  
                        module_name = 'copy',  
                        module_args = "src=~/.ssh/id_rsa.pub  dest=~/.ssh/authorized_keys owner=%s group=%s mode=644 backup=yes" %(username, username),  
                        remote_user = username,  
                        remote_pass = password,  
                        sudo = 'yes',  
                        sudo_pass =password,  
                        pattern = hostname,  
                )  
                b = runner_copy_autherized_keys.run()  
                print b  
      
                runner = ansible.runner.Runner(  
                    module_name = 'shell',  
                    module_args = "echo '%s' >>/etc/ansible/hosts"% ipadd,  
                    sudo = 'yes',  
                    sudo_pass = 'xxxxxxx',  
                    transport = 'local',  
                    pattern = '127.0.0.1',  
                )  
                #异步执行该操作,防止web页面被卡住  
                runner.run_async(30)  
                coll.save(post_dic)  
                #保存主机信息到数据库  
                self.render(  
                "listhost.html",  
                #调用主机列表模板显示被添加的主机信息  
                hosts = hosts,  
            )  
            else:  
                #如果存在,则更新主机信息  
                coll.update(post_dic,post_dic)  
                self.render(  
                # "listhost.html",  
                # hosts = hosts,  
            )  
      
      
    class delete_hostHandler(tornado.web.RequestHandler):  
        #定义删除主机的函数  
        def post(self, *args, **kwargs):  
            hostnames = self.get_arguments('hostname')  
            # 根据checkbox得到hostname的列表  
            coll = self.application.db.waitfish  
            #获得数据库游标  
            for host in hostnames:  
                coll.remove({"hostname":host})  
                #根据主机名删除  
            self.render(  
                "delete_info.html",  
                message = "%s is removed!"% hostnames,  
                #给出消息  
            )  
      
      
    class Module_actionHandler(tornado.web.RequestHandler):  
        #定义模块操作函数方法  
        def get(self, *args, **kwargs):  
            coll = self.application.db.waitfish  
            #初始化数据库连接  
            hosts = coll.find({}, {'hostname':1,'ipadd':1,"_id":0})  
            #这里hostname:1 表示返回hostname列,由于_id列每次都返回所以用0禁用掉,模板还可以一样  
            modulenames = ['ping', 'setup', 'copy','shell']  
            #现实我们定义的操作  
            self.render(  
                "module_action.html",  
                hosts = hosts,  
                modulenames = modulenames,  
            )  
      
        def post(self, *args, **kwargs):  
            ipadd = self.get_arguments('ipadd')[0]  
            #获取主机名  
            module = self.get_arguments('modulename')[0]  
            #获取模块名  
            arg = self.get_arguments('args')[0]  
            #获取参数  
            coll = self.application.db.waitfish  
            #初始化数据库  
            user = coll.find_one({'ipadd':'%s'%ipadd})['username']  
            hostname = coll.find_one({'ipadd':'%s'%ipadd})['hostname']  
            #从数据库找到主机的用户名信息  
            import ansible.runner  
            runner = ansible.runner.Runner(  
                #根据ansible的api来运行脚本  
                module_name = module,  
                module_args = arg,  
                remote_user = user,  
                #设定操作远程受管主机的用户名  
                pattern = ipadd,  
                #设定要操作主机名  
            )  
            result = runner.run()  
            #得到返回结果,这里是同步执行,下个版本改进异步  
            def pars_result(result):  
                # 定义一个判断结果的函数  
                if len(result['dark'])>0:  
                    # dark返回不为空则表示操作失败了  
                    return result['dark'],'失败!'  
                else:  
                    return result['contacted'],'成功!'  
            result = pars_result(result)  
            self.render(  
                "message.html",  
                hostname = hostname,  
                message = result[0],  
                jieguo = result[1]  
            )  
      
    if __name__ == "__main__":  
        tornado.options.parse_command_line()  
        http_server = tornado.httpserver.HTTPServer(Application())  
        http_server.listen(options.port)  
        tornado.ioloop.IOLoop.instance().start()