SNMP2JSON/snmp2json.py at master · krisclarkdev/SNMP2JSON · GitHub

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
63
64
65
66
67
68
69
import argparse
import json
from easysnmp import Session

_author_ = 'krisclarkdev'
_project_ = 'SNMP2JSON'

class SNMP2JSON:
    parser = argparse.ArgumentParser(description='Reads an SNMP table into a JSON String')

    def __init__(self):
        SNMP2JSON.parser.add_argument('-c', '--community',
                                        action='store', dest='community',
                                        help='SNMP Community String (Default: public)', default='public')

        SNMP2JSON.parser.add_argument('-n', '--hostname',
                                          action='store', dest='hostname',
                                          help='SNMP Hostname (Default: 192.168.1.1)', default='192.168.1.1')

        SNMP2JSON.parser.add_argument('-p', '--port',
                                          action='store', dest='port',
                                          help='SNMP Port (Default: 161)', default='161')

        SNMP2JSON.parser.add_argument('-t', '--table',
                                          action='store', dest='table',
                                          help='SNMP Port (Default: ifEntry)', default='ifEntry')

        SNMP2JSON.parser.add_argument('-v', '--version',
                                          action='store', dest='version',
                                          help='SNMP Version (Default: 2)', default=2)

        SNMP2JSON.parser.add_argument('-f', '--file',
                                          action='store', dest='filename',
                                          help='Save the results to a file (Default: print to screen)', default=None)

        SNMP2JSON.args = SNMP2JSON.parser.parse_args()

    def fetchTable(self):
        session = Session(hostname=SNMP2JSON.args.hostname + ':' + SNMP2JSON.args.port, community=SNMP2JSON.args.community, version=SNMP2JSON.args.version)
        system_items = session.walk(SNMP2JSON.args.table)

        data = []
        for item in system_items:
            oid = item.oid
            oid_index = item.oid_index
            snmp_type = item.snmp_type
            value = item.value

            test = {
                'oid': oid,
                'oid_index': oid_index,
                'snmp_type': snmp_type,
                'value': value
            }

            data.append(test)

        allData = {
            '' + SNMP2JSON.args.table: data
        }

        if(SNMP2JSON.args.filename == None):
            print json.dumps(allData, indent=4, sort_keys=True)
        else:
            with open(SNMP2JSON.args.filename, 'wt') as f:
                f.write(json.dumps(allData, indent=4, sort_keys=True))
            print 'JSON Flushed to ' + SNMP2JSON.args.filename

SNMP2JSON().fetchTable()