#!/usr/bin/python

###############################################################
# Copyright (c) 2017 ZTE Corporation
#
# 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
##############################################################################

import json
from numpy import mean
import os

from ansible.plugins.action import ActionBase

from qtip.util.export_to import export_to_file


class ActionModule(ActionBase):
    def run(self, tmp=None, task_vars=None):

        if task_vars is None:
            task_vars = dict()

        result = super(ActionModule, self).run(tmp, task_vars)

        if result.get('skipped', False):
            return result

        basepath = self._task.args.get('basepath')

        return aggregate(
            hosts=task_vars['groups'][self._task.args.get('group')],
            basepath=basepath,
            src=self._task.args.get('src'),
            dest=os.path.join(basepath, self._task.args.get('dest'))
        )


# aggregate QPI results
@export_to_file
def aggregate(hosts, basepath, src):
    host_results = [{'host': host, 'result': json.load(open(os.path.join(basepath, host, src)))} for host in hosts]
    score = int(mean([r['result']['score'] for r in host_results]))
    return {
        'score': score,
        'host_results': host_results
    }