moved routes into their own package

inventory_provider/opsdb.py

+import contextlib
+import logging
+
+import mysql.connector
+
+from inventory_provider.constants import DATABASE_LOGGER_NAME
+
+
+@contextlib.contextmanager
+def connection(alarmsdb):
+    cx = None
+    try:
+        cx = mysql.connector.connect(
+            host=alarmsdb["hostname"],
+            user=alarmsdb["username"],
+            passwd=alarmsdb["password"],
+            db=alarmsdb["dbname"])
+        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):
+    database_logger = logging.getLogger(DATABASE_LOGGER_NAME)
+    with cursor(db) as crs:
+        database_logger.debug("_db_test: %r" % router)
+        query = "SELECT absid FROM routers WHERE hostname = %s"
+        crs.execute(query, (router['hostname'],))
+        for (absid,) in crs:
+            database_logger.debug("absid: %r" % absid)

inventory_provider/routes/opsdb.py

+import functools
+import json
+
+from flask import Blueprint, request, Response, current_app
+import redis
+
+routes = Blueprint("inventory-data-query-routes", __name__)
+
+VERSION = {
+    "api": "0.1",
+    "module": "0.1"
+}
+
+
+def require_accepts_json(f):
+    """
+    used as a route handler decorator to return an error
+    unless the request allows responses with type "application/json"
+    :param f: the function to be decorated
+    :return: the decorated function
+    """
+    @functools.wraps(f)
+    def decorated_function(*args, **kwargs):
+        # TODO: use best_match to disallow */* ...?
+        if not request.accept_mimetypes.accept_json:
+            return Response(
+                response="response will be json",
+                status=406,
+                mimetype="text/html")
+        return f(*args, **kwargs)
+    return decorated_function
+
+
+@routes.route("/version", methods=['GET', 'POST'])
+@require_accepts_json
+def version():
+    return Response(
+        json.dumps(VERSION),
+        mimetype="application/json"
+    )
+
+
+@routes.route("/routers", methods=['GET', 'POST'])
+@require_accepts_json
+def routers():
+    redis_config = current_app.config["INVENTORY_PROVIDER_CONFIG"]["redis"]
+    r = redis.StrictRedis(
+        host=redis_config["hostname"],
+        port=redis_config["port"])
+    return Response(
+        json.dumps(list([k.decode("utf-8") for k in r.keys("*")])),
+        mimetype="application/json")