ChaosStuff/mqtt2prometheus Changeset - 5229d5941a1b · Chaos Computer Club Lëtzebuerg Projects

Changeset - 5229d5941a1b

Parent rev.

Child rev.

[Not reviewed]

default

0 1 0

Dennis Fink - 5 years ago 2020-02-24 18:58:23
dennis.fink@c3l.lu

Simplified sensor handling

1 file changed with 13 insertions and 62 deletions:

mqtt2prometheus.py

0 comments (0 inline, 0 general)

mqtt2prometheus.py

➞

Show inline comments

@@ @@ -30,143 +30,94 @@ temperature = Gauge( @@
     "temperature_celsius", "The current temperature", labelnames=["location"]
+)
 humidity = Gauge("humidity_percent", "The current humidity", labelnames=["location"])
 barometer = Gauge(
     "barometer_hectopascal", "The current air pressure", labelnames=["location"]
+)
 power_consumption = Gauge(
     "power_consumption_watt", "The current power used", labelnames=["location"]
+)
 co2_emission = Gauge(
     "co2_emission_kilogramm", "The CO2 emission in kilogramm", labelnames=["location"]
+)
 energy_consumption = Gauge(
     "energy_consumption_kilowatthour",
     "The energy consumption in kilowatthour",
     labelnames=["location"],
+)
 TOPIC_TO_HANDLER = {
 def on_connect(client, userdata, flags, rc):
     client.publish("availability/mqtt2prometheus", "online", 2, True)
     topics_handler = {
     "availability/#": handle_availability,
     "space/status": handle_space_status,
     "space/member_count": handle_member_count,
     "sensors/people_now_present": handle_people_now_present,
     "sensors/temperature/#": handle_temperature,
     "sensors/humidity/#": handle_humidity,
     "sensors/barometer/#": handle_barometer,
     "sensors/power_consumption/#": handle_power_consumption,
     "sensors/co2_emission/#": handle_co2_emission,
     "sensors/energy_consumption/#": handle_energy_consumption,
         "sensors/#": handle_sensors,
+}
 def on_connect(client, userdata, flags, rc):
     client.publish("availability/mqtt2prometheus", "online", 2, True)
     for topic, handler in TOPIC_TO_HANDLER:
     for topic, handler in topics_handler.items():
         client.message_callback_add(topic, handler)
     for topic in ["availability/#", "space/status", "space/member_count", "sensors/#"]:
         client.subscribe(topic)
 def handle_people_now_present(client, userdata, msg):
     people_now_present.set(json.loads(msg.payload))
 def handle_space_status(client, userdata, msg):
     state = json.loads(msg.payload)
     space_state.set(int(state["open"]))
     space_lastchange.set(state["lastchange"])
 def handle_member_count(client, userdata, msg):
     count = json.loads(msg.payload)
     member_count.labels(count["name"]).set(count["value"])
 def handle_availability(client, userdata, msg):
     t = msg.topic.split("/")
     if len(t) == 2:
         c = t[-1]
     else:
         c = "_".join(t[1:])
     v = 1 if msg.payload == b"online" else 0
     availability.labels(c).set(v)
 def handle_temperature(client, userdata, msg):
     t = msg.topic.split("/")
 def handle_sensor(client, userdata, msg):
     t = msg.topic_split("/")
     location = "-".join(t[2:])
     try:
         temperature.labels(location).set(float(msg.payload))
     except:
         pass
 def handle_humidity(client, userdata, msg):
     t = msg.topic.split("/")
     location = "-".join(t[2:])
     try:
         humidity.labels(location).set(float(msg.payload))
     except:
         pass
 def handle_barometer(client, userdata, msg):
     t = msg.topic.split("/")
     location = "-".join(t[2:])
     if t[1] == "people_now_present":
         return
     metric = globals().get(t[1], None)
     if metric is not None:
     try:
         barometer.labels(location).set(float(msg.payload))
     except:
         pass
 def handle_power_consumption(client, userdata, msg):
     t = msg.topic.split("/")
     location = "-".join(t[2:])
     try:
         power_consumption.labels(location).set(float(msg.payload))
     except:
         pass
 def handle_co2_emission(client, userdata, msg):
     t = msg.topic.split("/")
     location = "-".join(t[2:])
     try:
         co2_emission.labels(location).set(float(msg.payload))
     except:
         pass
 def handle_energy_consumption(client, userdata, msg):
     t = msg.topic.split("/")
     location = "-".join(t[2:])
     try:
         energy_consumption.labels(location).set(float(msg.payload))
             metric.labels(location).set(float(msg.payload))
     except:
         pass
 def main():
     client = mqtt.Client("mqtt2prometheus")
     client.on_connect = on_connect
     client.username_pw_set(CONFIG["mqtt"]["username"], CONFIG["mqtt"]["password"])
     client.will_set("availability/mqtt2prometheus", "offline", 2, True)
     client.connect(CONFIG["mqtt"]["host"])
     client.loop_start()
     start_http_server(CONFIG["prometheus"]["port"])
     while True:
         time.sleep(1)
 if __name__ == "__main__":
     main()

0 comments (0 inline, 0 general)