111 lines
4.9 KiB
Python
Executable file
111 lines
4.9 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
from configparser import ConfigParser
|
|
import socket
|
|
|
|
from re2oapi import Re2oAPIClient
|
|
|
|
from jinja2 import Environment, FileSystemLoader
|
|
|
|
config = ConfigParser()
|
|
config.read('config.ini')
|
|
|
|
api_hostname = config.get('Re2o', 'hostname')
|
|
api_password = config.get('Re2o', 'password')
|
|
api_username = config.get('Re2o', 'username')
|
|
|
|
api_client = Re2oAPIClient(api_hostname, api_username, api_password)
|
|
|
|
client_hostname = socket.gethostname().split('.', 1)[0]
|
|
|
|
print("get switchs conf")
|
|
all_switchs = api_client.list("switchs/ports-config/")
|
|
|
|
# Création de l'environnement Jinja
|
|
ENV = Environment(loader=FileSystemLoader('.'))
|
|
|
|
|
|
# Création du template final avec les valeurs contenues dans le dictionnaire "valeurs" - Ces valeurs sont positionnées dans un objet "temp", qui sera utilisé par le moteur, et que l'on retrouve dans le template.
|
|
|
|
|
|
class Switch:
|
|
def __init__(self):
|
|
self.additionnal = None
|
|
self.all_vlans = api_client.list("machines/vlan/")
|
|
self.all_roles = api_client.list("machines/role/")
|
|
# Import du fichier template dans une variable "template"
|
|
self.hp_tpl = ENV.get_template("templates/hp.tpl")
|
|
self.conf = None
|
|
self.name = None
|
|
self.switch = None
|
|
|
|
def get_conf_file_name(self):
|
|
return self.switch["short_name"] + ".conf"
|
|
|
|
def preprocess_hp(self):
|
|
"""Prérempli certains valeurs renvoyées directement à jinja, pour plus de simplicité"""
|
|
|
|
def add_to_vlans(vlans, vlan, port, tagged=True):
|
|
if not vlan['vlan_id'] in vlans:
|
|
if not tagged:
|
|
vlans[vlan['vlan_id']] = {'ports_untagged' : [str(port['port'])], 'ports_tagged' : [], 'name' : vlan['name']}
|
|
else:
|
|
vlans[vlan['vlan_id']] = {'ports_tagged' : [str(port['port'])], 'ports_untagged' : [], 'name' : vlan['name']}
|
|
else:
|
|
if not tagged:
|
|
vlans[vlan['vlan_id']]['ports_untagged'].append(str(port['port']))
|
|
else:
|
|
vlans[vlan['vlan_id']]['ports_tagged'].append(str(port['port']))
|
|
|
|
vlans = dict()
|
|
|
|
for port in self.switch['ports']:
|
|
if port['get_port_profil']['vlan_untagged']:
|
|
add_to_vlans(vlans, port['get_port_profil']['vlan_untagged'], port, tagged=False)
|
|
if port['get_port_profil']['vlan_tagged']:
|
|
for vlan in port['get_port_profil']['vlan_tagged']:
|
|
add_to_vlans(vlans, vlan, port)
|
|
|
|
arp_protect_vlans = [vlan["vlan_id"] for vlan in self.all_vlans if vlan["arp_protect"]]
|
|
dhcp_snooping_vlans = [vlan["vlan_id"] for vlan in self.all_vlans if vlan["dhcp_snooping"]]
|
|
dhcpv6_snooping_vlans = [vlan["vlan_id"] for vlan in self.all_vlans if vlan["dhcpv6_snooping"]]
|
|
igmp_vlans = [vlan["vlan_id"] for vlan in self.all_vlans if vlan["igmp"]]
|
|
mld_vlans = [vlan["vlan_id"] for vlan in self.all_vlans if vlan["mld"]]
|
|
ntp_servers = [server["servers"] for server in self.all_roles if server["role_type"] == "ntp-server"][0]
|
|
log_servers = [server["servers"] for server in self.all_roles if server["role_type"] == "log-server"][0]
|
|
dhcp_servers = [server["servers"] for server in self.all_roles if server["role_type"] == "dhcp"][0]
|
|
radius_servers = [server["servers"] for server in self.all_roles if server["role_type"] == "radius-server"][0]
|
|
ra_guarded = [str(port['port']) for port in self.switch['ports'] if port['get_port_profil']['ra_guard']]
|
|
loop_protected = [str(port['port']) for port in self.switch['ports'] if port['get_port_profil']['loop_protect']]
|
|
|
|
self.additionals = {'ra_guarded' : ra_guarded, 'loop_protected' : loop_protected, 'vlans' : vlans, 'arp_protect_vlans' : arp_protect_vlans, 'dhcp_snooping_vlans' : dhcp_snooping_vlans, 'dhcpv6_snooping_vlans' : dhcpv6_snooping_vlans, 'ntp_servers': ntp_servers, 'log_servers': log_servers, 'dhcp_servers' : dhcp_servers, 'radius_servers' : radius_servers, 'igmp_vlans' : igmp_vlans, 'mld_vlans': mld_vlans}
|
|
|
|
|
|
def gen_conf_hp(self):
|
|
"""Génère la config pour ce switch hp"""
|
|
self.preprocess_hp()
|
|
self.conf = self.hp_tpl.render(switch=self.switch, additionals=self.additionals)
|
|
|
|
|
|
def gen_conf_and_write(self):
|
|
"""Génère la conf suivant le bon constructeur et l'écrit"""
|
|
if self.switch["model"]:
|
|
constructor = self.switch["model"]["constructor"].lower()
|
|
if "hp" in constructor or "aruba" in constructor:
|
|
self.gen_conf_hp()
|
|
self.write_conf()
|
|
|
|
def write_conf(self):
|
|
"""Ecriture de la conf du switch dans le fichier qui va bien"""
|
|
with open("generated/" + self.get_conf_file_name(), 'w+') as f:
|
|
f.write(self.conf)
|
|
|
|
print("gen tpl")
|
|
|
|
|
|
sw = Switch()
|
|
for switch in all_switchs:
|
|
sw.switch = switch
|
|
# sw.gen_conf_and_write()
|
|
sw.gen_conf_hp()
|
|
sw.write_conf()
|
|
|