mirror of
https://github.com/Telecominfraproject/oopt-gnpy.git
synced 2025-10-30 17:47:50 +00:00
104 lines
4.2 KiB
Python
104 lines
4.2 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
create_eqpt_sheet.py
|
|
====================
|
|
|
|
XLS parser that can be called to create a "City" column in the "Eqpt" sheet.
|
|
|
|
If not present in the "Nodes" sheet, the "Type" column will be implicitly
|
|
determined based on the topology.
|
|
"""
|
|
|
|
from sys import exit
|
|
try:
|
|
from xlrd import open_workbook
|
|
except ModuleNotFoundError:
|
|
exit('Required: `pip install xlrd`')
|
|
from argparse import ArgumentParser
|
|
from collections import namedtuple, defaultdict
|
|
|
|
|
|
Shortlink = namedtuple('Link', 'src dest')
|
|
|
|
Shortnode = namedtuple('Node', 'nodename eqt')
|
|
|
|
parser = ArgumentParser()
|
|
parser.add_argument('workbook', nargs='?', default='meshTopologyExampleV2.xls',
|
|
help = 'create the mandatory columns in Eqpt sheet ')
|
|
all_rows = lambda sh, start=0: (sh.row(x) for x in range(start, sh.nrows))
|
|
|
|
def read_excel(input_filename):
|
|
with open_workbook(input_filename) as wb:
|
|
# reading Links sheet
|
|
links_sheet = wb.sheet_by_name('Links')
|
|
links = []
|
|
nodeoccuranceinlinks = []
|
|
links_by_src = defaultdict(list)
|
|
links_by_dest = defaultdict(list)
|
|
for row in all_rows(links_sheet, start=5):
|
|
links.append(Shortlink(row[0].value,row[1].value))
|
|
links_by_src[row[0].value].append(Shortnode(row[1].value,''))
|
|
links_by_dest[row[1].value].append(Shortnode(row[0].value,''))
|
|
#print(f'source {links[len(links)-1].src} dest {links[len(links)-1].dest}')
|
|
nodeoccuranceinlinks.append(row[0].value)
|
|
nodeoccuranceinlinks.append(row[1].value)
|
|
|
|
# reading Nodes sheet
|
|
nodes_sheet = wb.sheet_by_name('Nodes')
|
|
nodes = []
|
|
node_degree = []
|
|
for row in all_rows(nodes_sheet, start=5) :
|
|
|
|
temp_eqt = row[6].value
|
|
# verify node degree to confirm eqt type
|
|
node_degree.append(nodeoccuranceinlinks.count(row[0].value))
|
|
if temp_eqt.lower() == 'ila' and nodeoccuranceinlinks.count(row[0].value) !=2 :
|
|
print(f'Inconsistancy: node {nodes[len(nodes)-1]} has degree \
|
|
{node_degree[len(nodes)-1]} and can not be an ILA ... replaced by ROADM')
|
|
temp_eqt = 'ROADM'
|
|
if temp_eqt == '' and nodeoccuranceinlinks.count(row[0].value) == 2 :
|
|
temp_eqt = 'ILA'
|
|
if temp_eqt == '' and nodeoccuranceinlinks.count(row[0].value) != 2 :
|
|
temp_eqt = 'ROADM'
|
|
# print(f'node {nodes[len(nodes)-1]} eqt {temp_eqt}')
|
|
nodes.append(Shortnode(row[0].value,temp_eqt))
|
|
# print(len(nodes)-1)
|
|
print(f'reading: node {nodes[len(nodes)-1].nodename} eqpt {temp_eqt}')
|
|
return links,nodes, links_by_src , links_by_dest
|
|
|
|
def create_eqt_template(links,nodes, links_by_src , links_by_dest, input_filename):
|
|
output_filename = f'{input_filename[:-4]}_eqpt_sheet.txt'
|
|
with open(output_filename, 'w', encoding='utf-8') as my_file:
|
|
# print header similar to excel
|
|
my_file.write('OPTIONAL\n\n\n\
|
|
\t\tNode a egress amp (from a to z)\t\t\t\t\tNode a ingress amp (from z to a) \
|
|
\nNode A \tNode Z \tamp type \tatt_in \tamp gain \ttilt \tatt_out\
|
|
amp type \tatt_in \tamp gain \ttilt \tatt_out\n')
|
|
|
|
tab = []
|
|
temp = []
|
|
i = 0
|
|
for lk in links:
|
|
if [e for n,e in nodes if n==lk.src][0] != 'FUSED' :
|
|
temp = [lk.src , lk.dest]
|
|
tab.append(temp)
|
|
my_file.write(f'{temp[0]}\t{temp[1]}\n')
|
|
for n in nodes :
|
|
if n.eqt.lower() == 'roadm' :
|
|
for src in links_by_dest[n.nodename] :
|
|
temp = [n.nodename , src.nodename]
|
|
tab.append(temp)
|
|
# print(temp)
|
|
my_file.write(f'{temp[0]}\t{temp[1]}\n')
|
|
i = i + 1
|
|
print(f'File {output_filename} successfully created with Node A - Node Z ' +
|
|
' entries for Eqpt sheet in excel file.')
|
|
|
|
if __name__ == '__main__':
|
|
args = parser.parse_args()
|
|
input_filename = args.workbook
|
|
links,nodes,links_by_src, links_by_dest = read_excel(input_filename)
|
|
create_eqt_template(links,nodes, links_by_src , links_by_dest , input_filename)
|