)

co2_emission = Gauge(

    "co2_emission_kilogramm", "The CO2 emission in kilogramm", labelnames=["location"]

)

def on_connect(client, userdata, flags, rc):

    client.publish("availability/mqtt2prometheus", "online", 2, True)

    client.message_callback_add("sensors/people_now_present", handle_people_now_present)

    client.message_callback_add("space/status", handle_space_status)

    client.message_callback_add("space/member_count", handle_member_count)

    client.message_callback_add("availability/#", handle_availability)

    client.message_callback_add("sensors/temperature/#", handle_temperature)

    client.message_callback_add("sensors/humidity/#", handle_humidity)

    client.message_callback_add("sensors/barometer/#", handle_barometer)

    client.message_callback_add("sensors/power_consumption/#", handle_power_consumption)

    client.message_callback_add("sensors/co2_emission/#", handle_co2_emission)

    client.subscribe("sensors/people_now_present")

    client.subscribe("space/status")

    client.subscribe("space/member_count")

    client.subscribe("availability/+")

    client.subscribe("availability/shutdown/+")

    client.subscribe("sensors/temperature/#")

    client.subscribe("sensors/humidity/#")

    client.subscribe("sensors/barometer/#")

    client.subscribe("sensors/power_consumption/#")

    client.subscribe("sensors/co2_emission/#")

def handle_people_now_present(client, userdata, msg):

    people_now_present.set(json.loads(msg.payload))

def handle_co2_emission(client, userdata, msg):

    t = msg.topic.split("/")

    location = "-".join(t[2:])

    co2_emission.labels(location).set(float(msg.payload))

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)