initial import

config.py

+import contextlib
+import json
+import re
+
+import click
+import mysql.connector
+
+
+
+
+def _validate_config(ctx, param, value):
+    """
+    loads, validates and returns configuration parameters
+
+    :param ctx:
+    :param param:
+    :param value:
+    :return:
+    """
+    return json.loads(value.read())
+
+
+def load_oids(config_file):
+    """
+    :param config_file: file-like object
+    :return:
+    """
+    result = {}
+    for line in config_file:
+        m = re.match(r'^([^=]+)=(.*)\s*$', line)
+        if m:
+            result[m.group(1)] = m.group(2)
+    return result
+
+
+def load_routers(config_file):
+    """
+    :param config_file: file-like object
+    :return:
+    """
+    for line in config_file:
+        m = re.match(r'^([a-z\d]+\.[a-z\d]{3,4}\.[a-z\d]{2}\.(geant|eumedconnect)\d*\.net)\s*=([^,]+)\s*,(.*)\s*$', line)
+        if not m:
+            print("malformed config file line: '%s'" + line.strip())
+            continue
+        yield {
+            "hostname": m.group(1),
+            "community": m.group(3),
+            "address": m.group(4)
+        }
+
+@contextlib.contextmanager
+def connection(alarmsdb):
+    cx = None
+    try:
+        cx = mysql.connector.connect(
+            host=alarmsdb["hostname"],
+            user=alarmsdb["username"],
+            passwd=alarmsdb["password"],
+            db=alarmsdb["dbname"])
+        print(cx)
+        yield cx
+    finally:
+        if cx:
+            cx.close()
+
+
+@contextlib.contextmanager
+def cursor(cnx):
+    csr = None
+    try:
+        csr = cnx.cursor()
+        yield csr
+    finally:
+        if csr:
+            csr.close()
+
+
+def _db_test(db, router):
+    with cursor(db) as crs:
+        print(router)
+        query = "SELECT absid FROM routers WHERE hostname = %s"
+        crs.execute(query, (router['hostname'],))
+        for (absid,) in crs:
+            print(absid)
+
+
+@click.command()
+@click.option(
+    "--config",
+#    required=True,
+    type=click.File(),
+    help="Configuration filename",
+    default=open("config.json"),
+    callback=_validate_config)
+def cli(config):
+    with connection(config["alarms-db"]) as c:
+        with open("routers_community.conf") as f:
+            for r in load_routers(f):
+                _db_test(c, r)
+
+
+if __name__ == "__main__":
+    cli()
+    # with open("oid_list.conf") as f:
+    #     print(load_oids(f))
+    # with open("routers_community.conf") as f:
+    #     for r in load_routers(f):
+    #         _db_test(r)
+

oid_list.conf

+#
+#   This file is located in dbupdates/conf and is used by scripts under dbupdates/scripts.
+#   It holds OID values for retrieving details of a router.
+#
+
+## IPv4
+v4Address=.1.3.6.1.2.1.4.20.1.1
+v4InterfaceOID=.1.3.6.1.2.1.4.20.1.2
+v4InterfaceName=.1.3.6.1.2.1.31.1.1.1.1
+v4Mask=.1.3.6.1.2.1.4.20.1.3
+
+## IPv6
+v6AddressAndMask=.1.3.6.1.2.1.55.1.8.1.2
+v6InterfaceName=.1.3.6.1.2.1.55.1.5.1.2
\ No newline at end of file

routers_community.conf

+######################################################################################################################################
+##                                                                                                                                  ##
+##  This is a configuration file that stores router names and the SNMP community name in <router>=<community>,<IP address> format.  ##
+##                                                                                                                                  ##
+######################################################################################################################################
+
+mx2.ath.gr.geant.net=0pBiFbD,62.40.114.59
+mx1.tal.ee.geant.net=0pBiFbD,62.40.96.1
+mx2.tal.ee.geant.net=0pBiFbD,62.40.96.2
+mx2.rig.lv.geant.net=0pBiFbD,62.40.96.4
+mx1.kau.lt.geant.net=0pBiFbD,62.40.96.6
+mx2.kau.lt.geant.net=0pBiFbD,62.40.96.5
+mx2.zag.hr.geant.net=0pBiFbD,62.40.96.8
+mx2.lju.si.geant.net=0pBiFbD,62.40.96.10
+mx1.bud.hu.geant.net=0pBiFbD,62.40.97.1
+mx1.pra.cz.geant.net=0pBiFbD,62.40.97.2
+mx2.bra.sk.geant.net=0pBiFbD,62.40.97.4
+mx1.lon.uk.geant.net=0pBiFbD,62.40.97.5
+mx1.vie.at.geant.net=0pBiFbD,62.40.97.7
+mx2.bru.be.geant.net=0pBiFbD,62.40.96.20
+mx1.poz.pl.geant.net=0pBiFbD,62.40.97.10
+mx1.ams.nl.geant.net=0pBiFbD,62.40.97.11
+mx1.fra.de.geant.net=0pBiFbD,62.40.97.12
+mx1.par.fr.geant.net=0pBiFbD,62.40.97.13
+mx1.gen.ch.geant.net=0pBiFbD,62.40.97.14
+mx1.mil2.it.geant.net=0pBiFbD,62.40.97.15
+mx1.lis.pt.geant.net=0pBiFbD,62.40.96.16
+mx2.lis.pt.geant.net=0pBiFbD,62.40.96.17
+mx1.mad.es.geant.net=0pBiFbD,62.40.97.16
+mx1.sof.bg.geant.net=0pBiFbD,62.40.96.21
+mx1.buc.ro.geant.net=0pBiFbD,62.40.96.19
+mx1.ham.de.geant.net=0pBiFbD,62.40.96.26
+mx1.dub.ie.geant.net=0pBiFbD,62.40.96.3
+mx1.dub2.ie.geant.net=0pBiFbD,62.40.96.25
+mx1.mar.fr.geant.net=0pBiFbD,62.40.96.12
+mx1.lon2.uk.geant.net=0pBiFbD,62.40.96.15
+rt1.clpk.us.geant.net=GEANT_RO,10.200.64.128
+rt1.denv.us.geant.net=GEANT_RO,10.200.67.128
+mx1.ath2.gr.geant.net=0pBiFbD,62.40.96.39
+qfx.par.fr.geant.net=0pBiFbD,62.40.117.170
+qfx.fra.de.geant.net=0pBiFbD,62.40.117.162