2019-05-10 使用python生成NE05E IP地址池脚本

逐渐在写一些工作中用的到的脚本，不太熟练，高手请提建议。
目的： 为一批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()