1#!/usr/bin/env python3
2# SPDX-License-Identifier: GPL-2.0+
3# vim: ts=2:sw=2:et:tw=80:nowrap
4
5from os import path
6import os, csv
7from itertools import chain
8
9from csv_collection import CSVCollection
10from ni_names import value_to_name
11import ni_values
12
13CSV_DIR = 'csv'
14
15def iter_src_values(D):
16  return D.items()
17
18def iter_src(D):
19  for dest in D:
20    yield dest, 1
21
22def create_csv(name, D, src_iter):
23  # have to change dest->{src:val} to src->{dest:val}
24  fieldnames = [value_to_name[i] for i in sorted(D.keys())]
25  fieldnames.insert(0, CSVCollection.source_column_name)
26
27  S = dict()
28  for dest, srcD in D.items():
29    for src,val in src_iter(srcD):
30      S.setdefault(src,{})[dest] = val
31
32  S = sorted(S.items(), key = lambda src_destD : src_destD[0])
33
34
35  csv_fname = path.join(CSV_DIR, name + '.csv')
36  with open(csv_fname, 'w') as F_csv:
37    dR = csv.DictWriter(F_csv, fieldnames, delimiter=';', quotechar='"')
38    dR.writeheader()
39
40    # now change the json back into the csv dictionaries
41    rows = [
42      dict(chain(
43        ((CSVCollection.source_column_name,value_to_name[src]),),
44        *(((value_to_name[dest],v),) for dest,v in destD.items())
45      ))
46      for src, destD in S
47    ]
48
49    dR.writerows(rows)
50
51
52def to_csv():
53  for d in ['route_values', 'device_routes']:
54    try:
55      os.makedirs(path.join(CSV_DIR,d))
56    except:
57      pass
58
59  for family, dst_src_map in ni_values.ni_route_values.items():
60    create_csv(path.join('route_values',family), dst_src_map, iter_src_values)
61
62  for device, dst_src_map in ni_values.ni_device_routes.items():
63    create_csv(path.join('device_routes',device), dst_src_map, iter_src)
64
65
66if __name__ == '__main__':
67  to_csv()
68