Added wifi 5G support

README.md

@@ -3,17 +3,20 @@ Home Assistant component for TP-Link router administration based on the [TP-Link
 
 See [Supported routers](#supports)
 
-<img src="https://raw.githubusercontent.com/AlexandrErohin/home-assistant-tplink-router/master/docs/media/sensors.png" width="70%">
+<img src="https://raw.githubusercontent.com/AlexandrErohin/home-assistant-tplink-router/master/docs/media/sensors.png" width="48%"> <img src="https://raw.githubusercontent.com/AlexandrErohin/home-assistant-tplink-router/master/docs/media/switches.png" width="48%">
 
 ## Components
 ### Switches
  - Router Reboot
  - 2.4Ghz main wifi Enable/Disable
  - 5Ghz main wifi Enable/Disable
+ - 6Ghz main wifi Enable/Disable
  - 2.4Ghz guest wifi Enable/Disable
  - 5Ghz guest wifi Enable/Disable
+ - 6Ghz guest wifi Enable/Disable
  - 2.4Ghz IoT wifi network Enable/Disable
  - 5Ghz IoT wifi network Enable/Disable
+ - 6Ghz IoT wifi network Enable/Disable
 
 ### Sensors
  - Total amount of wired clients
@@ -24,7 +27,7 @@ See [Supported routers](#supports)
  - Memory used
 
 ### Device Tracker
- - Track clients by MAC address across 2.4Ghz, 5Ghz guest and main wifi with connection information
+ - Track clients by MAC address across 2.4Ghz, 5Ghz, 6Ghz guest and main wifi with connection information
 
 To find your device - Go to `Developer tools` and search for your MAC address - you’ll find sensor like `device_tracker.YOUR_MAC` or `device_tracker.YOUR_PHONE_NAME`.
 

custom_components/tplink_router/manifest.json

@@ -7,5 +7,5 @@
     "iot_class": "local_polling",
     "issue_tracker": "https://github.com/AlexandrErohin/home-assistant-tplink-router/issues",
     "requirements": ["tplinkrouterc6u==3.3.0"],
-    "version": "1.9.0"
+    "version": "1.10.0"
   }

custom_components/tplink_router/switch.py

@@ -41,6 +41,14 @@ class TPLinkRouterSwitchEntityDescription(SwitchEntityDescription, TPLinkRouterS
         property='guest_5g_enable',
         method=lambda coordinator, value: coordinator.set_wifi(Wifi.WIFI_GUEST_5G, value),
     ),
+    TPLinkRouterSwitchEntityDescription(
+        key="wifi_guest_6g",
+        name="Guest WIFI 6G",
+        icon="mdi:wifi",
+        entity_category=EntityCategory.CONFIG,
+        property='guest_6g_enable',
+        method=lambda coordinator, value: coordinator.set_wifi(Wifi.WIFI_GUEST_6G, value),
+    ),
     TPLinkRouterSwitchEntityDescription(
         key="wifi_24g",
         name="WIFI 2.4G",
@@ -57,6 +65,14 @@ class TPLinkRouterSwitchEntityDescription(SwitchEntityDescription, TPLinkRouterS
         property='wifi_5g_enable',
         method=lambda coordinator, value: coordinator.set_wifi(Wifi.WIFI_5G, value),
     ),
+    TPLinkRouterSwitchEntityDescription(
+        key="wifi_6g",
+        name="WIFI 6G",
+        icon="mdi:wifi",
+        entity_category=EntityCategory.CONFIG,
+        property='wifi_6g_enable',
+        method=lambda coordinator, value: coordinator.set_wifi(Wifi.WIFI_6G, value),
+    ),
     TPLinkRouterSwitchEntityDescription(
         key="iot_24g",
         name="IoT WIFI 2.4G",
@@ -73,6 +89,14 @@ class TPLinkRouterSwitchEntityDescription(SwitchEntityDescription, TPLinkRouterS
         property='iot_5g_enable',
         method=lambda coordinator, value: coordinator.set_wifi(Wifi.WIFI_IOT_5G, value),
     ),
+    TPLinkRouterSwitchEntityDescription(
+        key="iot_6g",
+        name="IoT WIFI 6G",
+        icon="mdi:wifi",
+        entity_category=EntityCategory.CONFIG,
+        property='iot_6g_enable',
+        method=lambda coordinator, value: coordinator.set_wifi(Wifi.WIFI_IOT_6G, value),
+    ),
 )