osm_mon/dashboarder/backends/grafana.py

# -*- coding: utf-8 -*-

# Copyright 2018 Whitestack, LLC
# *************************************************************

# This file is part of OSM Monitoring module
# All Rights Reserved to Whitestack, LLC

# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at

#         http://www.apache.org/licenses/LICENSE-2.0

# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
# For those usages not covered by the Apache License, Version 2.0 please
# contact: glavado@whitestack.com
##
import logging
import requests

log = logging.getLogger(__name__)

# TODO (lavado): migrate to Class, import config variables to get token
url = "http://grafana:3000/api/"
headers = {
    'content-type': "application/json",
    'authorization': "Basic YWRtaW46YWRtaW4="
    }


def get_all_dashboard_uids():
    response = requests.request("GET", url + "search?query=%", headers=headers)
    dashboards = response.json()
    dashboard_uids = []
    for dashboard in dashboards:
        dashboard_uids.append(dashboard['uid'])
    log.debug("Searching for all dashboard uids: %s", dashboard_uids)
    return dashboard_uids


def get_dashboard_status(uid):
    response = requests.request("GET", url + "dashboards/uid/" + uid, headers=headers)
    log.debug("Searching for dashboard result: %s", response.text)
    return response


def create_dashboard(uid, name, json_file):
    with open(json_file) as f:
        dashboard_data = f.read()

    dashboard_data = dashboard_data.replace('OSM_ID', uid).replace('OSM_NAME', name)

    response = requests.request("POST", url + "dashboards/db/",  data=dashboard_data, headers=headers)
    log.debug("Creating dashboard result: %s", response.text)
    return response


def delete_dashboard(uid):
    response = requests.request("DELETE", url + "dashboards/uid/" + uid, headers=headers)
    log.debug("Delete dashboard result: %s", response.text)
    return response