drivers/comedi/drivers/ni_routing/tools/convert_py_to_csv.py

   1 #!/usr/bin/env python3
   2 # SPDX-License-Identifier: GPL-2.0+
   3 # vim: ts=2:sw=2:et:tw=80:nowrap
   4
   5 from os import path
   6 import os, csv
   7 from itertools import chain
   8
   9 from csv_collection import CSVCollection
  10 from ni_names import value_to_name
  11 import ni_values
  12
  13 CSV_DIR = 'csv'
  14
  15 def iter_src_values(D):
  16   return D.items()
  17
  18 def iter_src(D):
  19   for dest in D:
  20     yield dest, 1
  21
  22 def 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
  52 def 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
  66 if __name__ == '__main__':
  67   to_csv()