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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
#############################################################################
# Copyright (c) 2017 Rajesh Kudaka
#
# Author: Rajesh Kudaka 4k.rajesh@gmail.com
# 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
##############################################################################
""" Handler for yardstick command 'report' """
from __future__ import print_function
from __future__ import absolute_import
import ast
import re
import uuid
from api.utils import influx
from django.conf import settings
from django.template import Context
from django.template import Template
from oslo_utils import encodeutils
from oslo_utils import uuidutils
from yardstick.common import constants as consts
from yardstick.common.html_template import template
from yardstick.common.utils import cliargs
settings.configure()
class Report(object):
"""Report commands.
Set of commands to manage benchmark tasks.
"""
def __init__(self):
self.Timestamp = []
self.yaml_name = ""
self.task_id = ""
def _validate(self, yaml_name, task_id):
if re.match(r"^[\w-]+$", yaml_name):
self.yaml_name = yaml_name
else:
raise ValueError("invalid yaml_name", yaml_name)
if uuidutils.is_uuid_like(task_id):
task_id = '{' + task_id + '}'
task_uuid = (uuid.UUID(task_id))
self.task_id = task_uuid
else:
raise ValueError("invalid task_id", task_id)
def _get_fieldkeys(self):
fieldkeys_cmd = "show field keys from \"%s\""
fieldkeys_query = fieldkeys_cmd % (self.yaml_name)
query_exec = influx.query(fieldkeys_query)
if query_exec:
return query_exec
else:
raise KeyError("Task ID or Test case not found..")
def _get_tasks(self):
task_cmd = "select * from \"%s\" where task_id= '%s'"
task_query = task_cmd % (self.yaml_name, self.task_id)
query_exec = influx.query(task_query)
if query_exec:
return query_exec
else:
raise KeyError("Task ID or Test case not found..")
@cliargs("task_id", type=str, help=" task id", nargs=1)
@cliargs("yaml_name", type=str, help=" Yaml file Name", nargs=1)
def generate(self, args):
"""Start report generation."""
self._validate(args.yaml_name[0], args.task_id[0])
self.db_fieldkeys = self._get_fieldkeys()
self.db_task = self._get_tasks()
field_keys = []
temp_series = []
table_vals = {}
field_keys = [encodeutils.to_utf8(field['fieldKey'])
for field in self.db_fieldkeys]
for key in field_keys:
self.Timestamp = []
series = {}
values = []
for task in self.db_task:
task_time = encodeutils.to_utf8(task['time'])
if not isinstance(task_time, str):
task_time = str(task_time, 'utf8')
key = str(key, 'utf8')
task_time = task_time[11:]
head, _, tail = task_time.partition('.')
task_time = head + "." + tail[:6]
self.Timestamp.append(task_time)
if task[key] is None:
values.append('')
elif isinstance(task[key], (int, float)) is True:
values.append(task[key])
else:
values.append(ast.literal_eval(task[key]))
table_vals['Timestamp'] = self.Timestamp
table_vals[key] = values
series['name'] = key
series['data'] = values
temp_series.append(series)
Template_html = Template(template)
Context_html = Context({"series": temp_series,
"Timestamp": self.Timestamp,
"task_id": self.task_id,
"table": table_vals})
with open(consts.DEFAULT_HTML_FILE, "w") as file_open:
file_open.write(Template_html.render(Context_html))
print("Report generated. View /tmp/yardstick.htm")
|
