add weight from MI_SCALE_WEIGHT_SAMPLE

This commit is contained in:
Oliver Großkloß 2025-07-17 12:56:00 +02:00
parent 958d33f366
commit bcf905b33b

34
main.py
View File

@ -128,6 +128,17 @@ class GadgetbridgeMQTTPublisher:
"device_class": "battery", "device_class": "battery",
} }
# Weight sensor
weight_config = {
"name": f"{self.device_name.replace('_', ' ').title()} Weight",
"unique_id": f"{self.device_name}_weight",
"state_topic": f"gadgetbridge/{self.device_name}/weight",
"unit_of_measurement": "kg",
"icon": "mdi:scale-bathroom",
"device": device_info,
"state_class": "measurement",
}
await self.publish_home_assistant_discovery( await self.publish_home_assistant_discovery(
"sensor", "daily_steps", steps_config "sensor", "daily_steps", steps_config
) )
@ -143,6 +154,7 @@ class GadgetbridgeMQTTPublisher:
await self.publish_home_assistant_discovery( await self.publish_home_assistant_discovery(
"sensor", "battery_level", battery_config "sensor", "battery_level", battery_config
) )
await self.publish_home_assistant_discovery("sensor", "weight", weight_config)
def get_steps_data(self) -> Dict: def get_steps_data(self) -> Dict:
"""Extract steps data from Gadgetbridge database""" """Extract steps data from Gadgetbridge database"""
@ -243,6 +255,7 @@ class GadgetbridgeMQTTPublisher:
"monthly": f"gadgetbridge/{self.device_name}/steps/monthly", "monthly": f"gadgetbridge/{self.device_name}/steps/monthly",
"last_sync": f"gadgetbridge/{self.device_name}/last_sync", "last_sync": f"gadgetbridge/{self.device_name}/last_sync",
"battery": f"gadgetbridge/{self.device_name}/battery", "battery": f"gadgetbridge/{self.device_name}/battery",
"weight": f"gadgetbridge/{self.device_name}/weight",
} }
try: try:
await self.mqtt_client.publish( await self.mqtt_client.publish(
@ -264,6 +277,12 @@ class GadgetbridgeMQTTPublisher:
await self.mqtt_client.publish( await self.mqtt_client.publish(
topics["battery"], str(battery_level), qos=1 topics["battery"], str(battery_level), qos=1
) )
# Publish latest weight
latest_weight = self.get_latest_weight()
if latest_weight is not None:
await self.mqtt_client.publish(
topics["weight"], str(latest_weight), qos=1
)
self.logger.info( self.logger.info(
f"Published steps data: Daily={data['daily_steps']}, Weekly={data['weekly_steps']}, Monthly={data['monthly_steps']}, Battery={battery_level}" f"Published steps data: Daily={data['daily_steps']}, Weekly={data['weekly_steps']}, Monthly={data['monthly_steps']}, Battery={battery_level}"
) )
@ -315,6 +334,21 @@ class GadgetbridgeMQTTPublisher:
self.logger.error(f"Error querying battery level: {e}") self.logger.error(f"Error querying battery level: {e}")
return None return None
def get_latest_weight(self) -> Optional[float]:
"""Get the latest weight in kg from MI_SCALE_WEIGHT_SAMPLE table"""
try:
conn = sqlite3.connect(self.db_path)
cursor = conn.cursor()
cursor.execute(
"SELECT WEIGHT_KG FROM MI_SCALE_WEIGHT_SAMPLE ORDER BY TIMESTAMP DESC LIMIT 1"
)
row = cursor.fetchone()
conn.close()
return row[0] if row else None
except Exception as e:
self.logger.error(f"Error querying latest weight: {e}")
return None
def get_all_device_names(db_path): def get_all_device_names(db_path):
"""Returns a list of all unique device names from the database.""" """Returns a list of all unique device names from the database."""