Create dishStatusInflux_cron.py

2021-01-08 22:27:54 -08:00 · 2021-01-08 22:27:54 -08:00 · fe3cf90612
commit fe3cf90612
parent 49cdcaa18c
1 changed files with 93 additions and 0 deletions
--- a/dishStatusInflux_cron.py
+++ b/dishStatusInflux_cron.py
@ -0,0 +1,93 @@
+#!/usr/bin/python3
+######################################################################
+#
+# Write get_status info to an InfluxDB database.
+#
+# This script will poll current status and write it to
+# the specified InfluxDB database.
+#
+######################################################################
+import os
+import grpc
+import spacex.api.device.device_pb2
+import spacex.api.device.device_pb2_grpc
+
+from influxdb import InfluxDBClient
+from influxdb import SeriesHelper
+
+influxdb_host = os.environ.get("INFLUXDB_HOST")
+influxdb_port = os.environ.get("INFLUXDB_PORT")
+influxdb_user = os.environ.get("INFLUXDB_USER")
+influxdb_pwd = os.environ.get("INFLUXDB_PWD")
+influxdb_db = os.environ.get("INFLUXDB_DB")
+
+class DeviceStatusSeries(SeriesHelper):
+    class Meta:
+        series_name = "spacex.starlink.user_terminal.status"
+        fields = [
+            "hardware_version",
+            "software_version",
+            "state",
+            "alert_motors_stuck",
+            "alert_thermal_throttle",
+            "alert_thermal_shutdown",
+            "alert_unexpected_location",
+            "snr",
+            "seconds_to_first_nonempty_slot",
+            "pop_ping_drop_rate",
+            "downlink_throughput_bps",
+            "uplink_throughput_bps",
+            "pop_ping_latency_ms",
+            "currently_obstructed",
+            "fraction_obstructed"]
+        tags = ["id"]
+
+influx_client = InfluxDBClient(host=influxdb_host, port=influxdb_port, username=influxdb_user, password=influxdb_pwd, database=influxdb_db, ssl=False, retries=1, timeout=15)
+
+dish_channel = None
+last_id = None
+last_failed = False
+
+while True:
+    try:
+        if dish_channel is None:
+            dish_channel = grpc.insecure_channel("192.168.100.1:9200")
+        stub = spacex.api.device.device_pb2_grpc.DeviceStub(dish_channel)
+        response = stub.Handle(spacex.api.device.device_pb2.Request(get_status={}))
+        status = response.dish_get_status
+        DeviceStatusSeries(
+            id=status.device_info.id,
+            hardware_version=status.device_info.hardware_version,
+            software_version=status.device_info.software_version,
+            state=spacex.api.device.dish_pb2.DishState.Name(status.state),
+            alert_motors_stuck=status.alerts.motors_stuck,
+            alert_thermal_throttle=status.alerts.thermal_throttle,
+            alert_thermal_shutdown=status.alerts.thermal_shutdown,
+            alert_unexpected_location=status.alerts.unexpected_location,
+            snr=status.snr,
+            seconds_to_first_nonempty_slot=status.seconds_to_first_nonempty_slot,
+            pop_ping_drop_rate=status.pop_ping_drop_rate,
+            downlink_throughput_bps=status.downlink_throughput_bps,
+            uplink_throughput_bps=status.uplink_throughput_bps,
+            pop_ping_latency_ms=status.pop_ping_latency_ms,
+            currently_obstructed=status.obstruction_stats.currently_obstructed,
+            fraction_obstructed=status.obstruction_stats.fraction_obstructed)
+        last_id = status.device_info.id
+        last_failed = False
+    except grpc.RpcError:
+        if dish_channel is not None:
+            dish_channel.close()
+            dish_channel = None
+        if last_failed:
+            if last_id is not None:
+                DeviceStatusSeries(id=last_id, state="DISH_UNREACHABLE")
+        else:
+            # Retry once, because the connection may have been lost while
+            # we were sleeping
+            last_failed = True
+            continue
+    try:
+        DeviceStatusSeries.commit(influx_client)
+    except Exception as e:
+        print("Failed to write: " + str(e))
+    break