From 14f0c8990dfff08c1f045d451570d940455546ef Mon Sep 17 00:00:00 2001
From: Josh <gitlab@breadlysm.com>
Date: Tue, 21 Jul 2020 20:26:27 -0400
Subject: [PATCH] handle packetloss
---
main.py | 9 +++++++--
1 file changed, 7 insertions(+), 2 deletions(-)
diff --git a/main.py b/main.py
index a47ef6d..21b1c1a 100755
--- a/main.py
+++ b/main.py
@@ -7,6 +7,7 @@ from influxdb import InfluxDBClient
# InfluxDB Settings
DB_ADDRESS = os.environ.get('INFLUX_DB_ADDRESS')
+print(os.environ.get('INFLUX_DB_PORT'))
DB_PORT = int(os.environ.get('INFLUX_DB_PORT'))
DB_USER = os.environ.get('INFLUX_DB_USER')
DB_PASSWORD = os.environ.get('INFLUX_DB_PASSWORD')
@@ -28,7 +29,11 @@ def init_db():
DB_DATABASE) # Create if does not exist.
else:
influxdb_client.switch_database(DB_DATABASE) # Switch to if does exist.
-
+def pkt_loss(data):
+ if data['packetLoss']:
+ return data['packetLoss']
+ else:
+ return 0
def format_for_influx(cliout):
data = json.loads(cliout)
@@ -66,7 +71,7 @@ def format_for_influx(cliout):
'measurement': 'packetLoss',
'time': data['timestamp'],
'fields': {
- 'packetLoss': int(data['packetLoss'])
+ 'packetLoss': pkt_loss(data)
}
}
]
--
GitLab