1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# netmap-csv.py
#
# Copyright 2015 Neil Williams <codehelp@debian.org>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
# MA 02110-1301, USA.
#
#
import csv
import yaml
import argparse
def main(args):
parser = argparse.ArgumentParser(description='network map CSV import')
parser.add_argument(
'--input', metavar='FILE', type=str,
dest='filename',
help='CSV file')
args = parser.parse_args()
rows = []
network = {}
with open(args.filename, 'r') as csvfile:
content = csv.DictReader(csvfile)
for row in content:
rows.append(row)
for row in rows:
for key, value in row.items():
if key == 'Switch' and value and 'Switch Port' in row:
network.setdefault('switches', {})
network['switches'].setdefault(str(value), [])
data = {'port': row['Switch Port']}
data['device'] = {
'interface': row['Device eth port'],
'mac': row['Device eth MAC/unique ID'],
'sysfs': '',
'hostname': row['Device'].replace('*', '')
}
network['switches'][str(value)].append(data)
with open('./output/output.yaml', 'w') as output:
yaml.dump(network, output, default_flow_style=False)
return 0
if __name__ == '__main__':
import sys
sys.exit(main(sys.argv))
|