diff options
Diffstat (limited to 'custom_components/skykettle/skykettle.py')
-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=[]): |