mirror of
https://github.com/Telecominfraproject/wlan-lanforge-scripts.git
synced 2025-10-31 18:58:01 +00:00
134 lines
4.6 KiB
Python
134 lines
4.6 KiB
Python
#!/usr/bin/env python3
|
|
|
|
#Copies the data from a CSV file from the KPI file generated from a Wifi Capacity test to an Influx database
|
|
|
|
# The CSV requires three columns in order to work: Date, test details, and numeric-score.
|
|
|
|
# Date is a unix timestamp, test details is the variable each datapoint is measuring, and numeric-score is the value for that timepoint and variable.
|
|
|
|
import sys
|
|
import os
|
|
from pprint import pprint
|
|
|
|
if sys.version_info[0] != 3:
|
|
print("This script requires Python 3")
|
|
exit(1)
|
|
|
|
if 'py-json' not in sys.path:
|
|
sys.path.append(os.path.join(os.path.abspath('..'), 'py-json'))
|
|
|
|
import argparse
|
|
from realm import Realm
|
|
import pandas as pd
|
|
|
|
|
|
class CSVtoInflux(Realm):
|
|
def __init__(self,
|
|
lfclient_host="localhost",
|
|
lfclient_port=8080,
|
|
debug=False,
|
|
_exit_on_error=False,
|
|
_exit_on_fail=False,
|
|
_proxy_str=None,
|
|
_capture_signal_list=[],
|
|
influxdb=None,
|
|
df=None):
|
|
super().__init__(lfclient_host=lfclient_host,
|
|
lfclient_port=lfclient_port,
|
|
debug_=debug,
|
|
_exit_on_error=_exit_on_error,
|
|
_exit_on_fail=_exit_on_fail,
|
|
_proxy_str=_proxy_str,
|
|
_capture_signal_list=_capture_signal_list)
|
|
self.influxdb = influxdb
|
|
self.df = df
|
|
|
|
# Submit data to the influx db if configured to do so.
|
|
def post_to_influx(self, kpi):
|
|
dates = list(set(kpi['Date']))
|
|
for date in dates:
|
|
kpi2 = self.df[self.df['Date'] == date][['Date', 'test details', 'numeric-score']]
|
|
metrics = list(set(kpi2['test details']))
|
|
targets = dict()
|
|
for k in metrics:
|
|
targets[k] = [*kpi2[kpi2['test details'] == k]['numeric-score']][0]
|
|
targets[k.replace(' ', '-').lower()] = targets.pop(k)
|
|
targets
|
|
tags = dict()
|
|
tags['script'] = 'WiFi Capacity'
|
|
for k in targets.keys():
|
|
self.influxdb.post_to_influx(k, targets[k], tags, date)
|
|
|
|
|
|
def main():
|
|
lfjson_host = "localhost"
|
|
lfjson_port = 8080
|
|
endp_types = "lf_udp"
|
|
debug = False
|
|
|
|
parser = argparse.ArgumentParser(
|
|
prog='test_l3_longevity.py',
|
|
# formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
formatter_class=argparse.RawTextHelpFormatter,
|
|
epilog='''
|
|
''',
|
|
|
|
description='''\
|
|
csv_to_influx.py:
|
|
--------------------
|
|
|
|
Summary :
|
|
----------
|
|
Copies the data from a CSV file generated by a wifi capacity test to an influx database.
|
|
|
|
Column names are designed for the KPI file generated by our Wifi Capacity Test.
|
|
|
|
A user can of course change the column names to match these in order to input any csv file.
|
|
|
|
The CSV file needs to have the following columns:
|
|
--date - which is a UNIX timestamp
|
|
--test details - which is the variable being measured by the test
|
|
--numeric-score - which is the value for that variable at that point in time.
|
|
|
|
Generic command layout:
|
|
-----------------------
|
|
python .\\csv_to_influx.py
|
|
|
|
|
|
Command:
|
|
python3 csv_to_influx.py --influx_host localhost --influx_org Candela --influx_token random_token --influx_bucket lanforge
|
|
--target_csv kpi.csv
|
|
|
|
|
|
''')
|
|
|
|
parser.add_argument('--influx_host', help='Hostname for the Influx database')
|
|
parser.add_argument('--influx_port', help='IP Port for the Influx database', default=8086)
|
|
parser.add_argument('--influx_org', help='Organization for the Influx database')
|
|
parser.add_argument('--influx_token', help='Token for the Influx database')
|
|
parser.add_argument('--influx_bucket', help='Name of the Influx bucket')
|
|
parser.add_argument('--target_csv', help='CSV file to record to influx database', required=True)
|
|
|
|
args = parser.parse_args()
|
|
|
|
influxdb = None
|
|
if args.influx_bucket is not None:
|
|
from influx2 import RecordInflux
|
|
influxdb = RecordInflux(_lfjson_host=lfjson_host,
|
|
_lfjson_port=lfjson_port,
|
|
_influx_host=args.influx_host,
|
|
_influx_port=args.influx_port,
|
|
_influx_org=args.influx_org,
|
|
_influx_token=args.influx_token,
|
|
_influx_bucket=args.influx_bucket)
|
|
df = pd.read_csv(args.target_csv)
|
|
|
|
csvtoinflux = CSVtoInflux(args=args,
|
|
influxdb=influxdb,
|
|
df=df)
|
|
csvtoinflux.record_kpi()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|