从0开始实现自动化运维工具(三)

栏目: 服务器 · 发布时间: 5年前

内容简介:ansible的一些基础配置信息保存在ansible.cfg文件中,包括ssh连接的默认host,默认port等。之前我们设计的程序是先从配置文件中读取信息,然后添加到执行字典中。为了方便读写,我们以json文件来保存配置信息。python读写json文件非常简单,只需要使用json库。playbook是ansible的精髓所在,python读写yaml文件也是非常简单便捷的,只需要使用yaml库。读取后的结果是一个只有一个字典元素的list,我们直接可以通过read_yaml_file()[0]来获取这个

ansible的一些基础配置信息保存在ansible.cfg文件中,包括ssh连接的默认host,默认port等。之前我们设计的程序是先从配置文件中读取信息,然后添加到执行字典中。为了方便读写,我们以json文件来保存配置信息。python读写json文件非常简单,只需要使用json库。

def read_cfg_file(cfg_file):
        if os.path.isfile(cfg_file):
            with open(cfg_file,"r") as f:
                file_content = json.load(f)

            if file_content:
                return file_content
            else:
                print "config file format error"
                sys.exit(1)

        else:
            print "config file not existed"
            sys.exit(1)
复制代码

读取yaml格式playbook

playbook是ansible的精髓所在,python读写yaml文件也是非常简单便捷的,只需要使用yaml库。

def read_yaml_file(filename):
        if os.path.isfile(filename):
            with open(filename,"r") as f:
                file_content = yaml.load(f)

            if file_content:
                return file_content
            else:
                print "yaml file format error"
                sys.exit(1)

        else:
            print "yaml file not existed"
            sys.exit(1)
复制代码

读取后的结果是一个只有一个字典元素的list,我们直接可以通过read_yaml_file()[0]来获取这个字典元素。

生成执行列表

通常一个playbook中包含多个task。在读取yaml文件生成的字典中,这些task以字典形式保存在一个list中。所以我们可以先将配置文件和yaml文件中除task部分的内容填入到临时字典中,然后将分别将task字典中的内容填入临时字典,然后将临时字典中的内容保存到一个list里,这样就获得了最终的task执行列表。

def get_final_task_list(ssh_user,ssh_pwd,yaml_file):
        cfg_dict = read_cfg_file()
        yaml_dict = read_yaml_file(yaml_file)[0]

        final_task_list=[]

        for key,value in cfg_dict.items():                    #将配置文件中的内容填入到模板字典中
            if template_dict.has_key(key):
                template_dict[key] = value

        for key,value in yaml_dict.items():                   #将yaml文件中除task部分外的内容填入到模板字典中
            if template_dict.has_key(key):
                if key != "tasks":
                    self.template_dict[key] = value

        template_dict["ssh_user"] = ssh_user                  #将ssh的用户和密码填入到模板字典中
        template_dict["ssh_pwd"] = ssh_pwd

    
        for task_dict in yaml_dict["tasks"]:
            tmp_dict = {}
            tmp_dict = template_dict                          #将模板字典复制到临时字典中
            for key,value in task_dict.items():            
                if tmp_dict.has_key(key):
                    tmp_dict[key] = value                     #将yaml文件中task部分分别填入到临时字典中
                    if "delegate_to" in task_dict.keys():
                        tmp_dict["task_host"] = task_dict["delegate_to"]     #处理delegate_to关键字,赋值给task_host

            final_task_list.append(tmp_dict.copy())           #将临时字典的浅拷贝添加到最终的执行列表中

        return final_task_list
复制代码

其中template_dict就是我们在第一节中设计的执行字典,格式如下:

template_dict={   
            "task_name": "",            
            "task_host": "",               
            "become": "",         
            "become_method": "",          
            "become_user": "",            
            "become_pwd": "",             
            "ssh_user": "",              
            "ssh_pwd": "",                
            "register": "",  
            "action_type": "",                
            "action": ""                
            }
复制代码

在写好这些代码之后,只需要调用get_final_task_list()函数,传入ssh登录的用户和密码,以及要使用的yaml文件后,就得到了最终的执行字典列表。


以上所述就是小编给大家介绍的《从0开始实现自动化运维工具(三)》,希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对 码农网 的支持!

查看所有标签

猜你喜欢:

本站部分资源来源于网络,本站转载出于传递更多信息之目的,版权归原作者或者来源机构所有,如转载稿涉及版权问题,请联系我们

Large-Scale Inference

Large-Scale Inference

Bradley Efron / Cambridge University Press / 2010-8-5 / GBP 48.00

We live in a new age for statistical inference, where modern scientific technology such as microarrays and fMRI machines routinely produce thousands and sometimes millions of parallel data sets, each ......一起来看看 《Large-Scale Inference》 这本书的介绍吧!

XML、JSON 在线转换
XML、JSON 在线转换

在线XML、JSON转换工具

XML 在线格式化
XML 在线格式化

在线 XML 格式化压缩工具

UNIX 时间戳转换
UNIX 时间戳转换

UNIX 时间戳转换