Files
lywsd03mmc_exporter/bt_exporter.py

106 lines
3.5 KiB
Python

"""Xiaomi Mi Temperature and Humidity Monitor 2 (LYWSD03MMC) prom exporter"""
import time
import argparse
import logging
from prometheus_client import start_http_server, Gauge, Enum
from bluepy import btle
logging.basicConfig(
level=logging.INFO,
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
datefmt='%d-%m-%Y %H:%M:%S')
log = logging.getLogger('LYWSD03MMC-exporter')
class XiaoMiTemp(btle.DefaultDelegate):
def __init__(self, mac, location, temp, humid, batt):
btle.DefaultDelegate.__init__(self)
self.mac = mac
self.loc = location
self.temp = temp
self.humid = humid
self.batt = batt
def handleNotification(self, cHandle, data):
databytes = bytearray(data)
self.temp.labels(self.mac, self.loc).set(int.from_bytes(databytes[0:2], "little") / 100)
self.humid.labels(self.mac, self.loc).set(int.from_bytes(databytes[2:3], "little"))
self.batt.labels(self.mac, self.loc).set(int.from_bytes(databytes[3:5], "little") / 1000)
class AppMetrics:
def __init__(self, devices, polling_interval_seconds=180):
self.polling_interval_seconds = polling_interval_seconds
self.devices = devices
self.fetch_count = int(0)
# Metrics scheme
self.temperature = Gauge("lywsd03mmc_temp", "Current temperature", ['mac', 'location'])
self.humidity = Gauge("lywsd03mmc_humid", "Current humidity", ['mac', 'location'])
self.battery = Gauge("lywsd03mmc_batt", "Battery level", ['mac', 'location'])
def run_metrics_loop(self):
"""Metrics fetching loop"""
while True:
self.fetch()
time.sleep(self.polling_interval_seconds)
def fetch(self):
self.fetch_count += 1
log.info(f"Fetch loop started. Iteration {self.fetch_count}")
for mac, location in self.devices:
log.info(f"Fetching '{mac}' ({location})")
p = btle.Peripheral()
p.withDelegate(XiaoMiTemp(mac, location, self.temperature, self.humidity, self.battery))
# BLE performs very poorly, constant errors are not uncommon
for attempt in range(10):
try:
p.connect(mac)
p.waitForNotifications(15.0)
log.info(f"Fetched '{mac}' ({location}). Attempt {attempt}.")
break
except Exception as e:
log.info(f"Failed to fetch '{mac}' ({location}). Attempt {attempt}.")
pass
finally:
p.disconnect()
def main():
parser = argparse.ArgumentParser(
description="Xiaomi Mi Temperature and Humidity Monitor 2 (LYWSD03MMC) prom exporter")
parser.add_argument(
'--device',
action='append',
required=True,
help="BLE Device in 'MAC;location' format")
parser.add_argument(
'--polling-interval',
default=180,
type=int,
help='Polling interval in seconds')
parser.add_argument(
'--port',
default=9877,
type=int,
help='Exporter port')
args = parser.parse_args()
devices = [tuple(device.split(';')) for device in args.device]
polling_interval = args.polling_interval
exporter_port = args.port
app_metrics = AppMetrics(
polling_interval_seconds=polling_interval,
devices=devices
)
start_http_server(exporter_port)
app_metrics.run_metrics_loop()
if __name__ == "__main__":
main()