Fix Sonoff TH fw 3.4.0

AlexxIT · Mar 9, 2020 · 1b44b08 · 1b44b08
1 parent a23d9a6
commit 1b44b08
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -38,7 +38,7 @@ Thanks to these people [@beveradb](https://github.com/beveradb/sonoff-lan-mode-h
 
 - [Sonoff Basic](https://www.itead.cc/sonoff-wifi-wireless-switch.html) fw 3.0.1
 - [Sonoff Mini](https://www.itead.cc/sonoff-mini.html) (no need use DIY-mode) fw 3.3.0
-- [Sonoff TH](https://www.itead.cc/sonoff-th.html) (show temperature and humidity) **control don't work on fw 3.4.0!**
+- [Sonoff TH](https://www.itead.cc/sonoff-th.html) (show temperature and humidity) fw 3.4.0
 - [Sonoff 4CH Pro R2](https://www.itead.cc/sonoff-4ch-pro.html) fw 3.3.0
 - [Sonoff Pow R2](https://www.itead.cc/sonoff-pow-r2.html) (show power consumption)
 - [Sonoff Micro](https://www.itead.cc/sonoff-micro-5v-usb-smart-adaptor.html) fw 3.4.0

diff --git a/custom_components/sonoff/__init__.py b/custom_components/sonoff/__init__.py
@@ -376,6 +376,11 @@ async def send(self, command: str, data: dict):
         except:
             _LOGGER.warning(f"Can't send {command} to {self.deviceid}")
 
+    @property
+    @lru_cache()
+    def is_th_3_4_0(self):
+        return self.state.get('deviceType') in ('normal', 'temperature')
+
     async def turn_on(self, channels: Optional[list]):
         """Включает указанные каналы.
 
@@ -387,6 +392,9 @@ async def turn_on(self, channels: Optional[list]):
                 for channel in channels
             ]
             await self.send('switches', {'switches': switches})
+        elif self.is_th_3_4_0:
+            await self.send('switch', {'switch': 'on', 'mainSwitch': 'on',
+                                       'deviceType': 'normal'})
         else:
             await self.send('switch', {'switch': 'on'})
 
@@ -401,6 +409,9 @@ async def turn_off(self, channels: Optional[list]):
                 for channel in channels
             ]
             await self.send('switches', {'switches': switches})
+        elif self.is_th_3_4_0:
+            await self.send('switch', {'switch': 'off', 'mainSwitch': 'off',
+                                       'deviceType': 'normal'})
         else:
             await self.send('switch', {'switch': 'off'})