mirror of
https://github.com/Telecominfraproject/wlan-lanforge-scripts.git
synced 2025-11-02 19:58:03 +00:00
317 lines
15 KiB
Python
317 lines
15 KiB
Python
"""
|
|
Candela Technologies Inc.
|
|
|
|
Info : Standard Script for Webconsole Test Utility
|
|
Date :
|
|
|
|
"""
|
|
import sys
|
|
import os
|
|
import importlib
|
|
import time
|
|
import matplotlib.pyplot as plt
|
|
import threading
|
|
import re
|
|
import json
|
|
|
|
sys.path.append(os.path.join(os.path.abspath(__file__ + "../../../")))
|
|
|
|
lfcli_base = importlib.import_module("py-json.LANforge.lfcli_base")
|
|
LFCliBase = lfcli_base.LFCliBase
|
|
LFUtils = importlib.import_module("py-json.LANforge.LFUtils")
|
|
realm = importlib.import_module("py-json.realm")
|
|
PortUtils = realm.PortUtils
|
|
|
|
webconsole_dir = os.path.dirname(os.path.dirname(os.path.dirname(os.getcwd())))
|
|
updates_path = webconsole_dir + "/web_json/updates.js"
|
|
|
|
|
|
class ClientVisualization(LFCliBase, threading.Thread):
|
|
def __init__(self, lfclient_host="localhost", lfclient_port=8080, num_clients=64, max_data=120, thread_id=None,
|
|
_debug_on=False, _exit_on_error=False, _exit_on_fail=False):
|
|
super().__init__(lfclient_host, lfclient_port, _debug=_debug_on, _exit_on_fail=_exit_on_fail)
|
|
threading.Thread.__init__(self)
|
|
self.num_clients = num_clients
|
|
self.max_data = max_data
|
|
self._stop_event = threading.Event()
|
|
self.client_data = {"down": [], "phantom": [], "ip": [], "scanning": []}
|
|
|
|
def stop(self):
|
|
self._stop_event.set()
|
|
|
|
def stopped(self):
|
|
return self._stop_event.is_set()
|
|
|
|
def run(self):
|
|
self.start_thread()
|
|
|
|
def start_thread(self):
|
|
while True:
|
|
self.scanning = 0
|
|
self.ip = 0
|
|
self.down = 0
|
|
self.phantom = 0
|
|
for i in self.json_get("/port/list?fields=port,alias,parent%20dev,down,phantom,ip,port%20type")['interfaces']:
|
|
|
|
for j in i:
|
|
print(i[j]['port type'])
|
|
if i[j]['port type'] == "WIFI-STA" and i[j]['parent dev'] == "wiphy1" and i[j]['alias'] != 'wlan1':
|
|
# print(j)
|
|
if i[j]['down'] is False and i[j]['phantom'] is False and i[j]['ip'] == '0.0.0.0':
|
|
self.scanning += 1
|
|
elif i[j]['down'] is False and i[j]['phantom'] is True:
|
|
self.phantom += 1
|
|
elif i[j]['down'] is True and i[j]['phantom'] is True:
|
|
self.phantom += 1
|
|
self.client_data['phantom'].append(self.phantom)
|
|
elif i[j]['down'] is True and i[j]['phantom'] is False:
|
|
self.down += 1
|
|
elif i[j]['ip'] != "0.0.0.0":
|
|
self.ip += 1
|
|
else:
|
|
continue
|
|
self.client_data['scanning'].append(self.scanning)
|
|
self.client_data['phantom'].append(self.phantom)
|
|
self.client_data['down'].append(self.down)
|
|
self.client_data['ip'].append(self.ip)
|
|
|
|
for i in self.client_data:
|
|
if len(self.client_data[i]) >= self.max_data:
|
|
self.client_data[i].pop(0)
|
|
time.sleep(1)
|
|
if self.stopped():
|
|
break
|
|
|
|
|
|
class CreateHTML:
|
|
def __init__(self, path="", test_name="", time_snap="", dut_ssid="", test_conf_data={}, objective="",
|
|
test_results={}, chart_data={}, chart_params={}):
|
|
self.head = """
|
|
<html>
|
|
<head>
|
|
<title>""" + test_name + """</title>
|
|
</head>
|
|
<body>
|
|
<div class='Section report_banner-1000x205' style='background-image:url("../../assets/brand/banner.jpg");background-repeat:no-repeat;padding:0;margin:0;min-width:1000px; min-height:205px;width:1000px; height:205px;max-width:1000px; max-height:205px;'>
|
|
<img align='right' style='padding:25;margin:5;width:200px;' src="../../assets/brand/logo.png" border='0' />
|
|
<div class='HeaderStyle'>
|
|
<br>
|
|
<h1 class='TitleFontPrint' style='color:darkgreen;'>""" + test_name + """</h1>
|
|
<h3 class='TitleFontPrint' style="color:darkgreen;">""" + time_snap + """</h3>
|
|
</div>
|
|
</div>
|
|
<br>
|
|
"""
|
|
self.test_conf = """
|
|
<table width="700px" border="1" cellpadding="2" cellspacing="0" style="border-top-color: gray; border-top-style: solid; border-top-width: 1px; border-right-color: gray; border-right-style: solid; border-right-width: 1px; border-bottom-color: gray; border-bottom-style: solid; border-bottom-width: 1px; border-left-color: gray; border-left-style: solid; border-left-width: 1px">
|
|
<tr>
|
|
<th colspan="2">
|
|
Test Setup Information
|
|
</th>
|
|
</tr>
|
|
<tr>
|
|
<td>
|
|
Device Under Test
|
|
</td>
|
|
<td>
|
|
<table width="100%" border="0" cellpadding="2" cellspacing="0" style="border-top-color: gray; border-top-style: solid; border-top-width: 1px; border-right-color: gray; border-right-style: solid; border-right-width: 1px; border-bottom-color: gray; border-bottom-style: solid; border-bottom-width: 1px; border-left-color: gray; border-left-style: solid; border-left-width: 1px">
|
|
<tr>
|
|
<td>
|
|
SSID
|
|
</td>
|
|
<td colspan="3">""" + dut_ssid + """
|
|
</td>
|
|
</tr>
|
|
"""
|
|
|
|
for i in test_conf_data:
|
|
self.test_conf = self.test_conf + """<tr>
|
|
<td>""" + str(i) + """
|
|
</td>
|
|
<td colspan="3">""" + test_conf_data[i] + """
|
|
</td>
|
|
</tr>
|
|
"""
|
|
|
|
self.test_conf = self.test_conf + """ </table>
|
|
</td>
|
|
</tr>
|
|
</table>
|
|
"""
|
|
|
|
self.objective = """
|
|
<br><h2 align="left">Objective</h2> <p align="left" width="900">
|
|
""" + objective + """
|
|
</p>
|
|
<br>
|
|
"""
|
|
|
|
if str(test_results['summary']).__contains__("PASS"):
|
|
self.summary_results = """
|
|
<br>
|
|
<table width="700px" border="1" cellpadding="2" cellspacing="0" style="border-top-color: gray; border-top-style: solid; border-top-width: 1px; border-right-color: gray; border-right-style: solid; border-right-width: 1px; border-bottom-color: gray; border-bottom-style: solid; border-bottom-width: 1px; border-left-color: gray; border-left-style: solid; border-left-width: 1px">
|
|
<tr>
|
|
<th colspan="2">
|
|
Summary Results
|
|
</th>
|
|
</tr>
|
|
<tr align='center' bgcolor="#90EE90">
|
|
<td>
|
|
""" + test_results['summary'] + """
|
|
</td>
|
|
</tr>
|
|
</table>
|
|
<br>
|
|
"""
|
|
else:
|
|
self.summary_results = """
|
|
<br>
|
|
<table width="700px" border="1" cellpadding="2" cellspacing="0" style="border-top-color: gray; border-top-style: solid; border-top-width: 1px; border-right-color: gray; border-right-style: solid; border-right-width: 1px; border-bottom-color: gray; border-bottom-style: solid; border-bottom-width: 1px; border-left-color: gray; border-left-style: solid; border-left-width: 1px">
|
|
<tr>
|
|
<th colspan="2">
|
|
Summary Results
|
|
</th>
|
|
</tr>
|
|
<tr align='center' bgcolor="orange">
|
|
<td>
|
|
""" + test_results['summary'] + """
|
|
</td>
|
|
</tr>
|
|
</table>
|
|
<br>
|
|
"""
|
|
chart_d = []
|
|
chart_label = []
|
|
for i in chart_data:
|
|
chart_label.append(i)
|
|
chart_d.append(chart_data[i])
|
|
|
|
self.detail_result = """<table width="1000px" border="1" cellpadding="2" cellspacing="0" >
|
|
<tr><th colspan="2">Detailed Results</th></tr>
|
|
<table width="1000px" border="1" >
|
|
<tr>
|
|
"""
|
|
for index in test_results['detail']['keys']:
|
|
self.detail_result = self.detail_result + "<th colspan='2'>" + index + "</th>"
|
|
self.detail_result = self.detail_result + "</tr>"
|
|
|
|
for data in test_results['detail']['data']:
|
|
self.detail_result = self.detail_result + "<tr align='center'>"
|
|
print("shivam")
|
|
print(data)
|
|
for i in data:
|
|
print(data[i])
|
|
if str(data[i]).__contains__("PASS"):
|
|
self.detail_result = self.detail_result + "<th colspan='2' bgcolor='#90EE90'>" + str(
|
|
data[i]) + "</th>"
|
|
elif str(data[i]).__contains__("FAIL"):
|
|
self.detail_result = self.detail_result + "<th colspan='2' bgcolor='orange'>" + str(
|
|
data[i]) + "</th>"
|
|
else:
|
|
self.detail_result = self.detail_result + "<th colspan='2'>" + str(data[i]) + "</th>"
|
|
self.detail_result = self.detail_result + "</tr>"
|
|
|
|
self.chart_data = chart_data
|
|
chart_values = []
|
|
for i in self.chart_data:
|
|
chart_values.append(self.chart_data[i])
|
|
plt.bar(list(self.chart_data.keys()), chart_values, tick_label=list(self.chart_data.keys()))
|
|
|
|
plt.xlabel(chart_params['xlabel'])
|
|
# naming the y-axis
|
|
plt.ylabel(chart_params['ylabel'])
|
|
# plot title
|
|
plt.title(chart_params['chart_head'])
|
|
plt.xticks(rotation=90, fontsize=8)
|
|
plt.tight_layout()
|
|
# function to show the plot
|
|
plt.savefig(fname=path + "plot.png")
|
|
plt.close()
|
|
|
|
self.chart = """<img align='center' style='padding:25;margin:5;width:600px;' src="plot.png" border='0' />"""
|
|
|
|
self.end = """</table>
|
|
</table>
|
|
</body>
|
|
</html>
|
|
"""
|
|
self.report = self.head + self.test_conf + self.objective + self.summary_results + self.chart + self.detail_result + self.end
|
|
|
|
|
|
class RuntimeUpdates:
|
|
def __init__(self, session_id, init_data):
|
|
self.session_id = session_id
|
|
self.init_data = init_data
|
|
f = open(updates_path, 'r+')
|
|
data = f.read()
|
|
f.close()
|
|
self.obj = data[data.find('{'): data.rfind('}') + 1]
|
|
self.obj = re.sub('[\']', '"', self.obj)
|
|
data = json.loads(self.obj)
|
|
print(data)
|
|
data["web_updates"].append({"ID": self.session_id, "data": self.init_data})
|
|
print(data)
|
|
f = open(updates_path, 'r+')
|
|
f.seek(0)
|
|
f.truncate()
|
|
f.write("var updates = " + str(data) + ";")
|
|
f.close()
|
|
|
|
def send_update(self, update_data):
|
|
f = open(updates_path, 'r+')
|
|
data = f.read()
|
|
f.close()
|
|
data_obj = data[data.find('{'): data.rfind('}') + 1]
|
|
data_obj = re.sub('[\']', '"', data_obj)
|
|
data = json.loads(data_obj)
|
|
|
|
for update in data["web_updates"]:
|
|
if update["ID"] == self.session_id:
|
|
update["data"] = update_data
|
|
print(data)
|
|
f = open(updates_path, 'r+')
|
|
f.seek(0)
|
|
f.truncate()
|
|
f.write("var updates = " + str(data) + ";")
|
|
f.close()
|
|
|
|
|
|
class StatusSession(LFCliBase):
|
|
def __init__(self, lfclient_host="localhost", lfclient_port=8080,
|
|
_deep_clean=False,
|
|
session_id="0",
|
|
_debug_on=False,
|
|
_exit_on_error=False,
|
|
_exit_on_fail=False):
|
|
super().__init__(lfclient_host, lfclient_port, _debug=_debug_on, _exit_on_fail=_exit_on_fail)
|
|
self.deep_clean = _deep_clean
|
|
self.session_id = session_id
|
|
self.json_put("/status-msg/" + self.session_id, {})
|
|
|
|
def update(self, key, message):
|
|
"""
|
|
Method to add new Message into a session
|
|
"""
|
|
self.json_post("/status-msg/" + self.session_id, {
|
|
"key": key,
|
|
"content-type": "text/plain",
|
|
"message": message
|
|
})
|
|
|
|
def read(self):
|
|
"""
|
|
Method to read all the messages for a particular session
|
|
"""
|
|
keys = []
|
|
for i in self.json_get("/status-msg/" + self.session_id)['messages']:
|
|
keys.append(i['key'])
|
|
json_uri = "/status-msg/" + self.session_id + "/"
|
|
for i in keys:
|
|
json_uri = json_uri + i + ","
|
|
return self.json_get(json_uri)['messages']
|
|
|
|
|
|
if __name__ == "__main__":
|
|
obj = StatusMsg(lfclient_host="localhost", lfclient_port=8090, session_id="01_18_21_20_04_20")
|
|
print(obj.read())
|