-
-
Notifications
You must be signed in to change notification settings - Fork 568
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge branch 'master' into feature/improve-test-coverage
- Loading branch information
Showing
15 changed files
with
659 additions
and
626 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,180 @@ | ||
| import logging | ||
| from typing import Dict, Any, Optional | ||
| from collections import defaultdict | ||
| from .device import Device | ||
| from .utils import deprecated | ||
|
|
||
| _LOGGER = logging.getLogger(__name__) | ||
|
|
||
| MODEL_CHUANGMI_PLUG_V3 = 'chuangmi.plug.v3' | ||
| MODEL_CHUANGMI_PLUG_V1 = 'chuangmi.plug.v1' | ||
| MODEL_CHUANGMI_PLUG_M1 = 'chuangmi.plug.m1' | ||
| MODEL_CHUANGMI_PLUG_V2 = 'chuangmi.plug.v2' | ||
|
|
||
| AVAILABLE_PROPERTIES = { | ||
| MODEL_CHUANGMI_PLUG_V1: ['on', 'usb_on', 'temperature'], | ||
| MODEL_CHUANGMI_PLUG_V3: ['on', 'usb_on', 'temperature', 'wifi_led'], | ||
| MODEL_CHUANGMI_PLUG_M1: ['power', 'temperature'], | ||
| MODEL_CHUANGMI_PLUG_V2: ['power', 'temperature'], | ||
| } | ||
|
|
||
|
|
||
| class ChuangmiPlugStatus: | ||
| """Container for status reports from the plug.""" | ||
|
|
||
| def __init__(self, data: Dict[str, Any]) -> None: | ||
| """ | ||
| Response of a Chuangmi Plug V1 (chuangmi.plug.v1) | ||
| { 'power': True, 'usb_on': True, 'temperature': 32 } | ||
| Response of a Chuangmi Plug V3 (chuangmi.plug.v3): | ||
| { 'on': True, 'usb_on': True, 'temperature': 32, 'wifi_led': True } | ||
| """ | ||
| self.data = data | ||
|
|
||
| @property | ||
| def power(self) -> bool: | ||
| """Current power state.""" | ||
| if "on" in self.data: | ||
| return self.data["on"] | ||
| if "power" in self.data: | ||
| return self.data["power"] == 'on' | ||
|
|
||
| @property | ||
| def is_on(self) -> bool: | ||
| """True if device is on.""" | ||
| return self.power | ||
|
|
||
| @property | ||
| def temperature(self) -> int: | ||
| return self.data["temperature"] | ||
|
|
||
| @property | ||
| def usb_power(self) -> Optional[bool]: | ||
| """True if USB is on.""" | ||
| if "usb_on" in self.data and self.data["usb_on"] is not None: | ||
| return self.data["usb_on"] | ||
| return None | ||
|
|
||
| @property | ||
| def load_power(self) -> Optional[int]: | ||
| """Current power load, if available.""" | ||
| if "load_power" in self.data and self.data["load_power"] is not None: | ||
| return self.data["load_power"] | ||
| return None | ||
|
|
||
| @property | ||
| def wifi_led(self) -> Optional[bool]: | ||
| """True if the wifi led is turned on.""" | ||
| if "wifi_led" in self.data and self.data["wifi_led"] is not None: | ||
| return self.data["wifi_led"] == "on" | ||
| return None | ||
|
|
||
| def __repr__(self) -> str: | ||
| s = "<ChuangmiPlugStatus " \ | ||
| "power=%s, " \ | ||
| "usb_power=%s, " \ | ||
| "temperature=%s" \ | ||
| "load_power=%s, " \ | ||
| "wifi_led=%s>" % \ | ||
| (self.power, | ||
| self.usb_power, | ||
| self.temperature, | ||
| self.load_power, | ||
| self.wifi_led) | ||
| return s | ||
|
|
||
|
|
||
| class ChuangmiPlug(Device): | ||
| """Main class representing the Chuangmi Plug V1 and V3.""" | ||
|
|
||
| def __init__(self, ip: str = None, token: str = None, start_id: int = 0, | ||
| debug: int = 0, lazy_discover: bool = True, | ||
| model: str = MODEL_CHUANGMI_PLUG_M1) -> None: | ||
| super().__init__(ip, token, start_id, debug, lazy_discover) | ||
|
|
||
| if model in AVAILABLE_PROPERTIES: | ||
| self.model = model | ||
| else: | ||
| self.model = MODEL_CHUANGMI_PLUG_M1 | ||
|
|
||
| def status(self) -> ChuangmiPlugStatus: | ||
| """Retrieve properties.""" | ||
| properties = AVAILABLE_PROPERTIES[self.model] | ||
| values = self.send( | ||
| "get_prop", | ||
| properties | ||
| ) | ||
|
|
||
| properties_count = len(properties) | ||
| values_count = len(values) | ||
| if properties_count != values_count: | ||
| _LOGGER.debug( | ||
| "Count (%s) of requested properties does not match the " | ||
| "count (%s) of received values.", | ||
| properties_count, values_count) | ||
|
|
||
| if self.model == MODEL_CHUANGMI_PLUG_V3: | ||
| load_power = self.send("get_power", []) # Response: [300] | ||
| if len(load_power) == 1: | ||
| properties.append('load_power') | ||
| values.append(load_power[0]) | ||
|
|
||
| return ChuangmiPlugStatus( | ||
| defaultdict(lambda: None, zip(properties, values))) | ||
|
|
||
| def on(self): | ||
| """Power on.""" | ||
| if self.model == MODEL_CHUANGMI_PLUG_V1: | ||
| return self.send("set_on", []) | ||
|
|
||
| return self.send("set_power", ["on"]) | ||
|
|
||
| def off(self): | ||
| """Power off.""" | ||
| if self.model == MODEL_CHUANGMI_PLUG_V1: | ||
| return self.send("set_off", []) | ||
|
|
||
| return self.send("set_power", ["off"]) | ||
|
|
||
| def usb_on(self): | ||
| """Power on.""" | ||
| return self.send("set_usb_on", []) | ||
|
|
||
| def usb_off(self): | ||
| """Power off.""" | ||
| return self.send("set_usb_off", []) | ||
|
|
||
| def set_wifi_led(self, led: bool): | ||
| """Set the wifi led on/off.""" | ||
| if led: | ||
| return self.send("set_wifi_led", ["on"]) | ||
| else: | ||
| return self.send("set_wifi_led", ["off"]) | ||
|
|
||
|
|
||
| @deprecated("This device class is deprecated. Please use the ChuangmiPlug " | ||
| "class in future and select a model by parameter 'model'.") | ||
| class Plug(ChuangmiPlug): | ||
| def __init__(self, ip: str = None, token: str = None, start_id: int = 0, | ||
| debug: int = 0, lazy_discover: bool = True) -> None: | ||
| super().__init__(ip, token, start_id, debug, lazy_discover, | ||
| model=MODEL_CHUANGMI_PLUG_M1) | ||
|
|
||
|
|
||
| @deprecated("This device class is deprecated. Please use the ChuangmiPlug " | ||
| "class in future and select a model by parameter 'model'.") | ||
| class PlugV1(ChuangmiPlug): | ||
| def __init__(self, ip: str = None, token: str = None, start_id: int = 0, | ||
| debug: int = 0, lazy_discover: bool = True) -> None: | ||
| super().__init__(ip, token, start_id, debug, lazy_discover, | ||
| model=MODEL_CHUANGMI_PLUG_V1) | ||
|
|
||
|
|
||
| @deprecated("This device class is deprecated. Please use the ChuangmiPlug " | ||
| "class in future and select a model by parameter 'model'.") | ||
| class PlugV3(ChuangmiPlug): | ||
| def __init__(self, ip: str = None, token: str = None, start_id: int = 0, | ||
| debug: int = 0, lazy_discover: bool = True) -> None: | ||
| super().__init__(ip, token, start_id, debug, lazy_discover, | ||
| model=MODEL_CHUANGMI_PLUG_V3) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file was deleted.
Oops, something went wrong.
Oops, something went wrong.