diff options
author | Alexey 'Cluster' Avdyukhin <clusterrr@clusterrr.com> | 2022-08-16 00:17:06 +0300 |
---|---|---|
committer | Alexey 'Cluster' Avdyukhin <clusterrr@clusterrr.com> | 2022-08-16 00:17:06 +0300 |
commit | 0b4166ce68cab806cfc2eac61c73083774cfdb73 (patch) | |
tree | 241c6658b51f54f3ffdb4f6b71e7b0ea70dec997 | |
parent | 62c776deb3691b988f97ea5535bd69ea74b261ee (diff) |
Ignore '-E' suffix
-rw-r--r-- | custom_components/skykettle/skykettle.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/custom_components/skykettle/skykettle.py b/custom_components/skykettle/skykettle.py index 52c0af2..0fa52d5 100644 --- a/custom_components/skykettle/skykettle.py +++ b/custom_components/skykettle/skykettle.py @@ -22,7 +22,6 @@ class SkyKettle(): #"RK-M173S": MODELS_2, # need more investigation "RK-G200": MODELS_3, "RK-G200S": MODELS_4, - "RK-G200S-E": MODELS_4, "RK-G201S": MODELS_4, "RK-G202S": MODELS_4, "RK-G203S": MODELS_4, @@ -40,7 +39,6 @@ class SkyKettle(): "RK-G240S": MODELS_4, "RK-M215S": MODELS_4, "RK-M216S": MODELS_4, - "RK-M216S-E": MODELS_4, "RK-M223S": MODELS_4, "RK-M136S": MODELS_4, "RK-M139S": MODELS_4, @@ -118,7 +116,11 @@ class SkyKettle(): @staticmethod def get_model_code(model): - return SkyKettle.MODEL_TYPE.get(model, None) + if model in SkyKettle.MODEL_TYPE: + return SkyKettle.MODEL_TYPE[model] + if model.endswith("-E"): + return SkyKettle.MODEL_TYPE.get(model[:-2], None) + return None @abstractmethod async def command(self, command, params=[]): |