summaryrefslogtreecommitdiffstats
path: root/qtip/reporter/console.py
blob: cb51d9c96702b87480e4fa9fe0b42636433defde (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
##############################################################################
# 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
##############################################################################

import glob
import json
from os import path

from jinja2 import Environment
from jinja2 import FileSystemLoader
from qtip.base import BaseActor

ROOT_DIR = path.join(path.dirname(__file__), path.pardir, path.pardir)


class ConsoleReporter(BaseActor):
    """ report benchmark result to console """

    def __init__(self, config, parent=None):
        super(ConsoleReporter, self).__init__(config, parent=parent)

        # TODO (taseer) load template from config
        tpl_path = path.join(path.dirname(__file__), 'templates')
        tpl_loader = FileSystemLoader(tpl_path)
        self._env = Environment(loader=tpl_loader)

    def load_result(self, result_path):
        result_dirs = glob.glob('{}/qtip-*'.format(result_path))
        # select the last (latest) directory for rendering report, result_dirs[-1]
        with open(path.join(result_path, result_dirs[-1], 'result.json')) as sample:
            result = json.load(sample)
        return result

    def render(self, metric, result_path):
        template = self._env.get_template('base.j2')
        var_dict = self.load_result(result_path)
        var_dict['metric_name'] = metric
        out = template.render(var_dict)
        return out