Force some fields to be floats

coredump-ch · Oct 2, 2024 · 95339fe · 95339fe
1 parent 4f1b2c3
commit 95339fe
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/relay.py b/relay.py
@@ -59,6 +59,13 @@ def log_to_influxdb(
     Log the specified data to InfluxDB.
     """
     print(f'Tags: {tags}, Fields: {fields}')
+
+    # Force data type for certain fields
+    if 'bit_rate_download' in fields:
+        fields['bit_rate_download'] = float(fields['bit_rate_download'])
+    if 'bit_rate_upload' in fields:
+        fields['bit_rate_upload'] = float(fields['bit_rate_upload'])
+
     json_body = [{
         'measurement': 'jvb_stats',
         'tags': tags,