Changeset - 5229d5941a1b
[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 19 insertions and 68 deletions:
0 comments (0 inline, 0 general)
mqtt2prometheus.py
Show inline comments
 
@@ -51,27 +51,19 @@ energy_consumption = Gauge(
 
)
 

	
 

	
 
TOPIC_TO_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,
 
}
 

	
 

	
 
def on_connect(client, userdata, flags, rc):
 
    client.publish("availability/mqtt2prometheus", "online", 2, True)
 

	
 
    for topic, handler in TOPIC_TO_HANDLER:
 
    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/#": handle_sensors,
 
    }
 

	
 
    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)
 

	
 

	
 
@@ -100,58 +92,17 @@ def handle_availability(client, userdata
 
    availability.labels(c).set(v)
 

	
 

	
 
def handle_temperature(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("/")
 
def handle_sensor(client, userdata, msg):
 
    t = msg.topic_split("/")
 
    location = "-".join(t[2:])
 
    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))
 
    except:
 
        pass
 
    if t[1] == "people_now_present":
 
        return
 
    metric = globals().get(t[1], None)
 
    if metric is not None:
 
        try:
 
            metric.labels(location).set(float(msg.payload))
 
        except:
 
            pass
 

	
 

	
 
def main():
0 comments (0 inline, 0 general)