From 99ef9252de107fb9d6eda4c6a4ebfc5301e5c1ae Mon Sep 17 00:00:00 2001 From: Luke Bonaccorsi Date: Tue, 8 Aug 2023 10:51:29 +0100 Subject: [PATCH 1/3] Fix whitespace --- custom_components/robovac/robovac.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/custom_components/robovac/robovac.py b/custom_components/robovac/robovac.py index c0acbf7..6fbdc86 100644 --- a/custom_components/robovac/robovac.py +++ b/custom_components/robovac/robovac.py @@ -118,4 +118,4 @@ class RoboVac(TuyaDevice): def getFanSpeeds(self): - return ROBOVAC_SERIES_FAN_SPEEDS[self.getRoboVacSeries()] + return ROBOVAC_SERIES_FAN_SPEEDS[self.getRoboVacSeries()] From d3139559f1492c8178e05826592fd2b2ba0920c7 Mon Sep 17 00:00:00 2001 From: Luke Bonaccorsi Date: Tue, 8 Aug 2023 12:19:47 +0100 Subject: [PATCH 2/3] Fix loop and incorrect variable name --- custom_components/robovac/robovac.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/custom_components/robovac/robovac.py b/custom_components/robovac/robovac.py index 6fbdc86..02c65c6 100644 --- a/custom_components/robovac/robovac.py +++ b/custom_components/robovac/robovac.py @@ -103,7 +103,7 @@ class RoboVac(TuyaDevice): | VacuumEntityFeature.STOP ) - if self.modelCode in HAS_MAP_FEATURE: + if self.model_code in HAS_MAP_FEATURE: supportedFeatures |= VacuumEntityFeature.MAP return supportedFeatures @@ -112,9 +112,9 @@ class RoboVac(TuyaDevice): return ROBOVAC_SERIES_FEATURES[self.getRoboVacSeries()] def getRoboVacSeries(self): - for series in ROBOVAC_SERIES: - if self.model_code in series.value: - return series.name + for series, models in ROBOVAC_SERIES: + if self.model_code in models: + return series def getFanSpeeds(self): From 057299559dd6a03e3c2ba1d3812b602765af4f6d Mon Sep 17 00:00:00 2001 From: Luke Bonaccorsi Date: Tue, 8 Aug 2023 12:44:49 +0100 Subject: [PATCH 3/3] Fix loop --- custom_components/robovac/robovac.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/custom_components/robovac/robovac.py b/custom_components/robovac/robovac.py index 02c65c6..e247896 100644 --- a/custom_components/robovac/robovac.py +++ b/custom_components/robovac/robovac.py @@ -112,7 +112,7 @@ class RoboVac(TuyaDevice): return ROBOVAC_SERIES_FEATURES[self.getRoboVacSeries()] def getRoboVacSeries(self): - for series, models in ROBOVAC_SERIES: + for series, models in ROBOVAC_SERIES.items(): if self.model_code in models: return series