逐渐在写一些工作中用的到的脚本，不太熟练，高手请提建议。
目的： 为一批NE05E写地址池的配置文件，由于站点比较多，很多重复性的工作，考虑用python生成。
用到的知识点：yaml，jinja2，excel，字符操作，字典等。
思路这样：
1、IP地址段是固定的，生成一个C类地址段的列表，需要的话可以考虑输入或从外部读取。
#这里是/18的地址段，共64个可用C类地址
list_0 = [("135.36." + str(x) + ".0") for x in range(192,256)]
2、读取yaml文件，生成含有站点名称的字典
curPath = os.path.dirname(os.path.realpath(__file__))
# 获取yaml文件路径
yamlPath = os.path.join(curPath, "create_pool.yaml")
#yaml文件名叫create_pool.yaml
# open方法打开直接读出来
f = open(yamlPath, 'r', encoding='utf-8')
cfg = f.read()
# 用load方法转字典
d = yaml.load(cfg)
#这里取得字典数量，即站点数，待会循环用
con1 = (len(d))
yaml文件的格式如下：
- pop_name: ol01.bes.lon
net_4_pool:
pool_gatway:
pool_start:
pool_end:
读出来是 这样的字典构成的一个列表
{'pop_name': 'ol01.bes.lon', 'net_4_pool': None, 'pool_gatway': None, 'pool_start': None, 'pool_end': None}
3、现在进行数据处理，根据地址段，算出几个参数，网关，地址池的开始地址，结束地址，替换到字典里去。
#写个函数
def main():
#循环，更新字典内容
for x2 in range(0,(con1)):
d[x2].update({'net_4_pool':list_0[x2], 'pool_gatway':(list_0[x2][:-1] + "1"), 'pool_start':(list_0[x2][:-1] + "2"), 'pool_end':(list_0[x2][:-1] + "254")})
#生成一个excel文件，把生成的数据写入表格
name=['pop_name', 'net_4_pool', 'pool_gatway', 'pool_start', 'pool_end']
content_0=pd.DataFrame(columns=name,data=d)
print(content_0)
content_0.to_csv('pop_pool.csv',encoding='gbk')
4、函数的下半部分，调用jinja2的模板，渲染之后输出为一个完整的配置文件。
#调用templates目录下的net_pool.tmpl模板文件
env = Environment(loader = FileSystemLoader('templates'), trim_blocks=True, lstrip_blocks=True)
template = env.get_template('net_pool.tmpl')
#写个文件名
today = datetime.date.today()
todaystr = today.strftime('%Y-%m-%d')
filename = 'output/' + todaystr + '_pools'
#创建一个空文件
with open(filename, "w") as f:
f.write('')
#循环添加每个站点的配置到文档
for x3 in range(0,(con1)):
output = template.render(d[x3])
print(output)
with open(filename, "a") as f:
f.write(output)
f.write('\n')
5、完整脚本
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
本脚本用于为站点配置IP地址池
首先读取yaml文件，获取站点数据，放入列表
生成可用的/24地址段，放入列表2
迭代处理，将站点与列表对应
为每个站点生成jinja2需要的变量，放入字典
将这些字典放入列表
最终生成列表n
包含的字典结构：站点名称，地址池名称，地址池起始IP，地址池结束IP
读取列表n，逐个渲染模板，放入列表
写入文件
"""
import yaml
import os
import datetime
import pandas as pd
from jinja2 import Environment, FileSystemLoader
from pprint import pprint
list_0 = [("135.36." + str(x) + ".0") for x in range(192,256)]
#print(list_0)
#print(len(list_0))
# 获取当前脚本所在文件夹路径
curPath = os.path.dirname(os.path.realpath(__file__))
# 获取yaml文件路径
yamlPath = os.path.join(curPath, "create_pool.yaml")
# open方法打开直接读出来
f = open(yamlPath, 'r', encoding='utf-8')
cfg = f.read()
#print(type(cfg)) # 读出来是字符串
#print(cfg)
d = yaml.load(cfg) # 用load方法转字典
#print(type(d))
con1 = (len(d))
#print(type(con1),con1)
#print(d)
def main():
for x2 in range(0,(con1)):
#print(x2, d[x2])
#print(list_0[x2], (list_0[x2][:-1] + "1"), (list_0[x2][:-1] + "2"), (list_0[x2][:-1] + "254"))
d[x2].update({'net_4_pool':list_0[x2], 'pool_gatway':(list_0[x2][:-1] + "1"), 'pool_start':(list_0[x2][:-1] + "2"), 'pool_end':(list_0[x2][:-1] + "254")})
print(d[x2])
name=['pop_name', 'net_4_pool', 'pool_gatway', 'pool_start', 'pool_end']
content_0=pd.DataFrame(columns=name,data=d)
print(content_0)
content_0.to_csv('pop_pool.csv',encoding='gbk')
env = Environment(loader = FileSystemLoader('templates'), trim_blocks=True, lstrip_blocks=True)
template = env.get_template('net_pool.tmpl')
today = datetime.date.today()
todaystr = today.strftime('%Y-%m-%d')
filename = 'output/' + todaystr + '_pools'
print(filename)
with open(filename, "w") as f:
f.write('')
for x3 in range(0,(con1)):
output = template.render(d[x3])
print(output)
with open(filename, "a") as f:
f.write(output)
f.write('\n')
if __name__ == "__main__":
main()
