mirror of
https://github.com/Telecominfraproject/wlan-lanforge-scripts.git
synced 2025-11-06 21:58:04 +00:00
b4bead52ff
Doesn't work yet, calling influx with wrong data structure it seems. But, getting closer. Signed-off-by: Ben Greear <greearb@candelatech.com>
93 lines
3.1 KiB
Python
93 lines
3.1 KiB
Python
#!/usr/bin/env python3
|
|
|
|
# pip3 install influxdb
|
|
|
|
import sys
|
|
|
|
if sys.version_info[0] != 3:
|
|
print("This script requires Python 3")
|
|
exit(1)
|
|
|
|
import requests
|
|
import json
|
|
from influxdb import InfluxDBClient
|
|
import datetime
|
|
from LANforge.lfcli_base import LFCliBase
|
|
import time
|
|
|
|
|
|
class RecordInflux(LFCliBase):
|
|
def __init__(self,
|
|
_lfjson_host="lanforge",
|
|
_lfjson_port=8080,
|
|
_influx_host="localhost",
|
|
_influx_port=8086,
|
|
_influx_user=None,
|
|
_influx_passwd=None,
|
|
_influx_db=None,
|
|
_debug_on=False,
|
|
_exit_on_fail=False):
|
|
super().__init__(_lfjson_host, _lfjson_port,
|
|
_debug=_debug_on,
|
|
_exit_on_fail=_exit_on_fail)
|
|
self.influx_host = _influx_host
|
|
self.influx_port = _influx_port
|
|
self.influx_user = _influx_user
|
|
self.influx_passwd = _influx_passwd
|
|
self.influx_db = _influx_db
|
|
self.client = InfluxDBClient(self.influx_host,
|
|
self.influx_port,
|
|
self.influx_user,
|
|
self.influx_passwd,
|
|
self.influx_db)
|
|
|
|
def post_to_influx(self, key, value, tags):
|
|
data = {}
|
|
data['measurement'] = key
|
|
data['tags'] = {}
|
|
for t in tags:
|
|
data['tags'][t.key] = t.val
|
|
data['time'] = str(datetime.datetime.utcnow().isoformat())
|
|
data['fields'] = {}
|
|
data['fields']['value'] = value
|
|
|
|
# json_body = json.dumps(data)
|
|
|
|
# json_body = [
|
|
# {
|
|
# "measurement": key,
|
|
# "tags": {
|
|
# "host": self.host,
|
|
# "region": "us-west"
|
|
# },
|
|
# "time": str(datetime.datetime.utcnow().isoformat()),
|
|
# "fields": {
|
|
# "value": value
|
|
# }
|
|
# }
|
|
# ]
|
|
self.client.write_points(data)
|
|
|
|
# Don't use this unless you are sure you want to.
|
|
# More likely you would want to generate KPI in the
|
|
# individual test cases and poke those relatively small bits of
|
|
# info into influxdb.
|
|
# This will not return until the 'longevity' timer has expired.
|
|
def monitor_port_data(self,
|
|
lanforge_host="localhost",
|
|
devices=None,
|
|
longevity=None,
|
|
monitor_interval=None):
|
|
url = 'http://' + lanforge_host + ':8080/port/1/1/'
|
|
end = datetime.datetime.now() + datetime.timedelta(0, longevity)
|
|
while datetime.datetime.now() < end:
|
|
for station in devices:
|
|
url1 = url + station
|
|
response = json.loads(requests.get(url1).text)
|
|
|
|
# Poke everything into influx db
|
|
for key in response['interface'].keys():
|
|
self.posttoinflux("%s-%s"%(station, key), response['interface'][key])
|
|
|
|
time.sleep(monitor_interval)
|