從0開始實現自動化運維工具（三）

時間 2019-11-09

原文原文鏈接

完整的代碼已經在github開源：github.com/CTC-maxiao/…python

讀取配置文件

ansible的一些基礎配置信息保存在ansible.cfg文件中，包括ssh鏈接的默認host，默認port等。以前咱們設計的程序是先從配置文件中讀取信息，而後添加到執行字典中。爲了方便讀寫，咱們以json文件來保存配置信息。python讀寫json文件很是簡單，只須要使用json庫。git

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庫。github

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]來獲取這個字典元素。json

生成執行列表

一般一個playbook中包含多個task。在讀取yaml文件生成的字典中，這些task以字典形式保存在一個list中。因此咱們能夠先將配置文件和yaml文件中除task部分的內容填入到臨時字典中，而後將分別將task字典中的內容填入臨時字典，而後將臨時字典中的內容保存到一個list裏，這樣就得到了最終的task執行列表。bash

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.copy()                          #將模板字典複製到臨時字典中
            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就是咱們在第一節中設計的執行字典，格式以下：app

template_dict={   
            "task_name": "",            
            "task_host": "",               
            "become": "",         
            "become_method": "",          
            "become_user": "",            
            "become_pwd": "",             
            "ssh_user": "",              
            "ssh_pwd": "",                
            "register": "",  
            "action_type": "",                
            "action": ""                
            }
複製代碼