Source code for routeros_telegraf_exporter.views

""" Cornice services.
"""
import os
from queue import Queue
from threading import Thread

from cornice import Service

from .utils import format_values_to_str
from .routeros_exporter import worker
from .models import Args

output_type = os.environ.get("ROUTEROS_EXPORTER_OUTPUT", "influx")
q = Queue()
metrics = Service(name='metrics', path='/metrics', description="RouterOS metrics output", content_type="text/html",
                  renderer="string" if output_type == "influx" else "json")

health = Service(name='health', path='/health/{router_name}', description="RouterOS health output",
                 content_type="text/html",
                 renderer="string" if output_type == "influx" else "json")

worker_args = Args(daemon=True, output_type=output_type)
qworker = Thread(target=worker, args=(worker_args, q,))
qworker.setDaemon(True)
qworker.start()


@metrics.get()
def get_metrics(request):
    """Returns JSON or influx formatted string
    Args:
    request: Pylon requests object
    """
    global output_type, q
    values = []
    direct = 0
    params = request.params
    if params.get("output_type"):
        output_type = params.get("output_type")
    if params.get("direct"):
        direct = int(params.get("direct"))

    if q.empty() and direct == 1:
        worker(worker_args, q, False)
    while not q.empty():
        values.append(q.get())
    if output_type == "influx":
        if values:
            return format_values_to_str(values)
        else:
            return ""
    return values


@health.get()
def get_health(request):
    """Returns JSON or influx formatted string
    Args:
    request: Pylon requests object
    """
    global output_type
    params = request.params
    if params.get("output_type"):
        output_type = params.get("output_type")
    else:
        output_type = os.environ.get("ROUTEROS_EXPORTER_OUTPUT", "influx")
    health_q = Queue()
    worker_args = Args(daemon=False, output_type=output_type, hosts_config_file=None)
    router_name = request.matchdict['router_name']
    worker_args.hosts = router_name
    worker_args.host = router_name
    worker_args.ignore_interval = True
    worker_args.hosts_config = [{router_name: {'resources': [
        {'path': '/system/resource', 'interval': 1, 'tags': ['board-name', 'platform'],
         'values': ['cpu-load']}]}}]
    values = worker(worker_args, health_q, False)
    if output_type == "influx":
        return format_values_to_str(values[0])
    value = values[0][0][0]

    return value