pyh0n/pyhon/appliances/wd.py

from pyhon.appliances.base import ApplianceBase


class Appliance(ApplianceBase):
    def data(self, data):
        super().data(data)
        if data["attributes"]["lastConnEvent"]["category"] == "DISCONNECTED":
            data["attributes"]["parameters"]["machMode"] = "0"
        data["active"] = bool(data.get("attributes", {}).get("activity"))
        data["pause"] = data["attributes"]["parameters"]["machMode"] == "3"
        return data

    def settings(self, settings):
        return settings