##############################################################################
# Copyright (c) 2017 taseer94@gmail.com and others.
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################

from asq.initiators import query
import click
from prettytable import PrettyTable

from qtip.reporter.console import ConsoleReporter


def extract_section(sections, section_name, node):
    """ Extract information related to QPI """
    qpi = query(sections).where(lambda child: child['name'] == node) \
                         .select_many(lambda child: child['sections']) \
                         .where(lambda child: child['name'] == section_name) \
                         .first()
    return qpi


def display_report(report, section_name, node):
    table_workload = PrettyTable(['Workload', 'Description',
                                  'Result', 'Score'])
    table_workload.align = 'l'

    section_report = extract_section(report['nodes'], section_name, node)

    for metric in section_report['metrics']:
        for wl in metric['workloads']:
            table_workload.add_row([wl['name'],
                                    wl['description'],
                                    wl['result'],
                                    wl['score']])
    return {
        "ss": section_report['score'],
        "desc": section_report['description'],
        "table": table_workload
    }


@click.group()
def cli():
    """ View QTIP results"""
    pass


@cli.command('show')
@click.option('-n', '--node', help="Compute node in OPNFV cluster")
@click.argument('section-name')
def show(node, section_name):
    qpi = ConsoleReporter.load_result()
    result = display_report(qpi, section_name, node)

    click.echo("Node Score: {}".format(qpi['score']))
    click.echo("Section Score: {}".format(result['ss']))
    click.echo("Description: {}".format(result['desc']))
    click.echo(result['table'])