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])?

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)

針對上面代碼的改良版，寫的不好的地方，歡迎大家指出！#!/usr/bin/env?python

#encoding=utf-8

import?sys,yaml

import?subprocess?????????????????????????????????#?用來創建一個子進程

from?jinja2?import?Environment,FileSystemLoader???#?統一使用jinja2模板

env?=?Environment(

loader?=?FileSystemLoader('./pxe_template')???#?env設為全局變量

)

def?parse_pxe_config(filename):

config?=?yaml.load(file(filename,'r'))

return?config

'''

'1'*4?==?'1111'

int('111',2)?==?7?二進制轉換十進制

從<

'''

def?int2mask(prefix):

return?'.'.join(str(int(('1'*(int(prefix))+'0'?\

*(32-int(prefix)))[i*8:(i+1)*8],2))?\

for?i?in?xrange(4))

'''

使用zip對兩個list同時遍歷，然后對各個元素進行&運算

192???168??2??????254

255???255??255??224

不知道通過ip和mask計算網絡號，google下！

'''

def?get_manage_net(gateway,prefix_str):

mask?=?int2mask(int(prefix_str))

return?'.'.join(str(int(m)?&?int(n))?\

for?m,n?in?zip(gateway.split('.'),mask.split('.')))

'''

模板渲染抽象為一個函數

渲染的時候可以傳進一個dict搞定，dict的key要和模板中的變量名對應(我老大實驗得到的)

'''

def?create_config_template(tem_name,tem_dict):

template?=?env.get_template(tem_name)

content?=?template.render(tem_dict)

return?content

'''

創建配置文件

'''

def?create_config_file(config_name,content,mode):

try:

fp?=?open(config_name,mode)

fp.write(content+'\n')

except?IOError:

print?"Error:?can\'t?find?file?or?read?data"

else:

fp.close()

'''

這部分內容比上面的搜身了不少吧，當然還有很多地方要改進的！

'''

def?create_pxe_config(config):

system_common?=?config.get('system?common')???????????#?字典使用get方法獲取key對應的value，如果key不存在，返回None。比config['system?common']友好多了。

manage_prefix?=?system_common.get('manage_prefix')

manage_gateway?=?system_common.get('manage_gateway')

manage_mask?=?int2mask(manage_prefix)

manage_net?=?get_manage_net(manage_gateway,manage_prefix)

deploy_node?=?config.get('deploy?node')

deploy_node['manage_gateway']?=?manage_gateway

deploy_node['manage_mask']?=?manage_mask

deploy_node['manage_net']?=?manage_net

dhcp_content?=?create_config_template('dhcpd.conf',deploy_node)

create_config_file('./pxe_config/dhcpd.conf',dhcp_content,'w')

compute_node?=?config['compute?node']

for?compute_name,compute_info?in?compute_node.items():

ip_mac_content?=?create_config_template("ip_mac.conf",compute_info)

create_config_file('./pxe_config/dhcpd.conf',ip_mac_content,'a')

subprocess.Popen('service?dhcpd?restart',shell?=?True)?????????????????#?開了子進程來重啟dhcpd服務，好讓修改的dhcp配置生效

deploy_node.update(system_common)

ks_content?=?create_config_template('puppet.cfg',deploy_node)

create_config_file('./pxe_config/puppet.cfg',ks_content,'w')

pxe_default_content?=?create_config_template('default',system_common)

create_config_file('./pxe_config/default',pxe_default_content,'w')

'''

這部分不變

'''

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)

參考鏈接

總結

以上是生活随笔為你收集整理的python渲染光线_python模板渲染配置文件的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。