edpibu/hasspy
1
Fork 0
Code Issues Pull requests Releases Activity Actions

Refactor power control logic in MQTT client to use pattern matching for improved readability

Browse source
This commit is contained in:
Edgar P. Burkhart 2025-03-09 11:40:53 +01:00
parent 98e1e970c1
commit be2381429f
Signed by: edpibu
GPG key ID: 9833D3C5A25BD227
1 changed files with 11 additions and 8 deletions

19
hasspy/mqtt.py
View file

@ -74,14 +74,17 @@ class HassClient(Client):
def on_command(self, client: Client, userdata: Any, message: MQTTMessage) -> None:
payload = message.payload.decode("utf-8")
if not self.power_on and payload == "POWER_ON":
log.info("Cancelling shutdown…")
self.power_on = True
run(["systemctl", "poweroff", "--when=cancel"])
elif self.power_on and payload == "POWER_OFF":
log.info("Powering off…")
self.power_on = False
run(["systemctl", "poweroff", "--when=+1m"])
match payload:
case "POWER_ON":
if not self.power_on:
log.info("Cancelling shutdown…")
self.power_on = True
run(["systemctl", "poweroff", "--when=cancel"])
case "POWER_OFF":
if self.power_on:
log.info("Powering off…")
self.power_on = False
run(["systemctl", "poweroff", "--when=+1m"])
self.publish_state()