python的mako、jinja2模板库，确实好用！这里做个笔记，好记性不如烂笔头。

#!/usr/bin/env python  
#encoding=utf-8
import sys,yaml                       # 配置文件使用yaml格式
from mako.template import Template      # 加载mako库的Template Class
from jinja2 import Environment,FileSystemLoader  # 加载jinja2的Environment,FileSystemLoader Class

‘‘‘

    解析配置文件，返回值是一个dict

‘‘‘

def parse_pxe_config(filename):
    config = yaml.load(file(filename,‘r‘))
    return config

‘‘‘

    这个函数实现的功能

    举例来说，掩码前缀为27转换为255.255.255.224

‘‘‘

def int2mask(mask_str):
    mask_int = int(mask_str)
    mask_array = [‘0‘ for i in xrange(32)]   # python for循环，吊吧！一口气生成32个list的元素
    for i in xrange(mask_int):
        mask_array[i] = ‘1‘        
    temp_mask = [‘‘.join(mask_array[i*8:(i+1)*8]) for i in xrange(4)]   # 每8位组成一个list元素
    temp_mask = [str(int(i,2)) for i in temp_mask]    # int(i,2) 二进制转换为十进制
    return ‘.‘.join(temp_mask)

‘‘‘

    根据网关地址和掩码前缀得到管理网网段

‘‘‘ 

def get_manage_net(gateway,prefix_str):
    prefix = int(prefix_str)
    int2bin = [bin(int(i,10)).split(‘0b‘)[1] for i in gateway.split(‘.‘)]   # bin(int(i,10)) 十进制转化为二进制
    for i in xrange(4):
        if len(int2bin[i]) < 8:      
            int2bin[i] = ‘0‘*(8 - len(int2bin[i])) + int2bin[i]      
    int2bin = ‘‘.join(int2bin)
    int2bin = int2bin[0:prefix]+‘0‘*(32-prefix)
    int2bin_list = [int2bin[8*i:(i+1)*8] for i in xrange(4)]
    manage_net = [str(int(i,2)) for i in int2bin_list]           
    return ‘.‘.join(manage_net)

‘‘‘

       利用python模板库渲染配置文件

‘‘‘

def create_pxe_config(config):
#  读取配置文件中的配置项
    system_common = config[‘system common‘]
    manage_prefix = system_common[‘manage_prefix‘]
    manage_gateway = system_common[‘manage_gateway‘]
    manage_mask = int2mask(manage_prefix)
    manage_net = get_manage_net(manage_gateway,manage_prefix)
    dns = system_common[‘dns‘]
    fqdn = system_common[‘fqdn‘]
    repo_url = system_common[‘repo_url‘]
    password = system_common[‘password‘]
    manage_nic = system_common[‘manage_nic‘]
    storage_nic = system_common[‘storage_nic‘]
    public_nic = system_common[‘public_nic‘]
    data_nic = system_common[‘data_nic‘]
    deploy_node = config[‘deploy node‘]
    deploy_node_ip = deploy_node[‘ip‘]
    deploy_node_hostname = deploy_node[‘hostname‘]
    dhcp_range_start = deploy_node[‘dhcp_range_start‘]
    dhcp_range_end = deploy_node[‘dhcp_range_end‘]
    
    compute_node = config[‘compute node‘]
    dhcp_template = Template(                                         # 创建一个Template对象
                            filename=‘./pxe_template/dhcpd.conf‘,
                            module_directory=‘/tmp/mako_modules‘      # 为了提高性能，从文件加载的 Template 还可以在文件系统中将生成的模块缓存为一般的Python模块文件，
                            )                                         # 下次同样参数的Template 创建时，自动重用/tmp/mako_modules/目录下的模块文件。
                            
    dhcp_content = dhcp_template.render(                              # 传给 Template 的文本参数被编译为一个Python模块。模块包含一个 render_body() 函数，它产生模板的输出。
                            manage_gateway = manage_gateway,          # 调用render() 方法时，Mako建立了一个模板的运行环境，并调用 render_body() 函数，把输出保存到缓冲，返回它的字符串内容
                            dns = dns,
                            manage_mask = manage_mask,
                            fqdn = fqdn,
                            deploy_node_ip = deploy_node_ip,
                            manage_net = manage_net,
                            dhcp_range_start = dhcp_range_start,
                            dhcp_range_end = dhcp_range_end
                                    )
    fp = open(‘./pxe_config/dhcpd.conf‘,‘w‘)     # 生成dhcpd.conf配置文件
    fp.write(dhcp_content)   
    if fp != None:
        fp.close()                              
    ip_mac_template = Template(                                         # ip-mac 绑定
                            filename=‘./pxe_template/ip_mac.conf‘,
                            module_directory=‘/tmp/mako_modules‘
                            )
    for compute in compute_node:
        ip_mac_content = ip_mac_template.render(          
                            compute_hostname = compute[‘hostname‘],
                            manage_mac = compute[‘manage_mac‘],
                            manage_ip = compute[‘manage_ip‘]
                                                )
        fp = open(‘./pxe_config/dhcpd.conf‘,‘a‘)                      # 以append方式打开文件
        fp.write(‘\n‘+ip_mac_content)                                 # 追加内容
    
    env = Environment(
                loader = FileSystemLoader(‘./pxe_template‘)          # 这个类的实例被用于存储配置信息, 全局对象, 从文件系统或其他位置加载模板，使用的加载器loader是FileSystemLoader类型, 
                    )                                                # 可以加载的模板是当前工作目录下的templates目录下的模板文件
    ks_template = env.get_template("puppet.cfg")                     # 使用env的模板环境加载名为puppet.cfg的模板文件.
    ks_content = ks_template.render(                                 # 渲染模板template
                            repo_url = repo_url,
                            password = password,
                            deploy_node_hostname = deploy_node_hostname,
                            fqdn = fqdn,
                            deploy_node_ip = deploy_node_ip,
                            public_nic = public_nic,
                            storage_nic = storage_nic,
                            data_nic = data_nic,
                            manage_gateway = manage_gateway,
                            manage_mask = manage_mask
                                    )
    fp = file(‘./pxe_config/puppet.cfg‘,‘w‘)
    fp.write(ks_content)
     
    if fp != None:
        fp.close()

if __name__ == ‘__main__‘:
    if len(sys.argv) != 2 or sys.argv[1] != ‘config.yaml‘:
        print "Usage:pxe-init.py config.yaml"
        sys.exit(-1)
    config = parse_pxe_config(sys.argv[1])
    create_pxe_config(config)

参考链接

http://www.yeolar.com/note/2012/08/26/mako-usage/

http://blog.csdn.net/lgg201/article/details/4647471

本文出自 “the-way-to-cloud” 博客，请务必保留此出处http://iceyao.blog.51cto.com/9426658/1566384

python模板渲染配置文件