Add active tray sensor

greghesp · Sep 8, 2023 · 8eca79e · 8eca79e
1 parent 61c69a1
commit 8eca79e
Show file tree

Hide file tree

Showing 3 changed files with 54 additions and 3 deletions.
diff --git a/custom_components/bambu_lab/definitions.py b/custom_components/bambu_lab/definitions.py
@@ -281,7 +281,7 @@ class BambuLabBinarySensorEntityDescription(BinarySensorEntityDescription, Bambu
     ),
     BambuLabSensorEntityDescription(
         key="tray_now",
-        translation_key="active_tray",
+        translation_key="active_tray_index",
         icon="mdi:printer-3d-nozzle",
         available_fn=lambda self: self.coordinator.get_model().supports_feature(
             Features.AMS) and self.coordinator.get_model().ams.tray_now != 255,
@@ -302,6 +302,39 @@ class BambuLabBinarySensorEntityDescription(BinarySensorEntityDescription, Bambu
         available_fn=lambda self: self.coordinator.get_model().info.subtask_name != "",
         value_fn=lambda self: self.coordinator.get_model().info.subtask_name
     ),
+    BambuLabSensorEntityDescription(
+        key="active_tray",
+        translation_key="active_tray",
+        icon="mdi:printer-3d-nozzle",
+        value_fn=lambda self: self.coordinator.get_model().get_active_tray().name,
+        extra_attributes=lambda self:
+        {
+            "color": f"#{self.coordinator.get_model().get_active_tray().color}",
+            "name": self.coordinator.get_model().get_active_tray().name,
+            "nozzle_temp_min": self.coordinator.get_model().get_active_tray().nozzle_temp_min,
+            "nozzle_temp_max": self.coordinator.get_model().get_active_tray().nozzle_temp_max,
+            "type": self.coordinator.get_model().get_active_tray().type,
+        },
+        available_fn=lambda self: self.coordinator.get_model().get_active_tray() is not None,
+        exists_fn=lambda coordinator: coordinator.get_model().supports_feature(Features.AMS) and not coordinator.get_model().supports_feature(Features.K_VALUE)
+    ),
+    BambuLabSensorEntityDescription(
+        key="active_tray",
+        translation_key="active_tray",
+        icon="mdi:printer-3d-nozzle",
+        value_fn=lambda self: self.coordinator.get_model().get_active_tray().name,
+        extra_attributes=lambda self:
+        {
+            "color": f"#{self.coordinator.get_model().get_active_tray().color}",
+            "k_value": self.coordinator.get_model().get_active_tray().k,
+            "name": self.coordinator.get_model().get_active_tray().name,
+            "nozzle_temp_min": self.coordinator.get_model().get_active_tray().nozzle_temp_min,
+            "nozzle_temp_max": self.coordinator.get_model().get_active_tray().nozzle_temp_max,
+            "type": self.coordinator.get_model().get_active_tray().type,
+        },
+        available_fn=lambda self: self.coordinator.get_model().get_active_tray() is not None,
+        exists_fn=lambda coordinator: coordinator.get_model().supports_feature(Features.AMS) and coordinator.get_model().supports_feature(Features.K_VALUE)
+    ),
 )
 
 VIRTUAL_TRAY_SENSORS: tuple[BambuLabSensorEntityDescription, ...] = (

diff --git a/custom_components/bambu_lab/pybambu/models.py b/custom_components/bambu_lab/pybambu/models.py
@@ -1,3 +1,5 @@
+import math
+
 from dataclasses import dataclass
 
 from .utils import \
@@ -29,6 +31,7 @@ def __init__(self, client, device_type, serial):
         self.external_spool = ExternalSpool(client)
         self.hms = HMSList(client)
         self.camera = Camera()
+        self._active_tray = None
 
     def print_update(self, data):
         """Update from dict"""
@@ -83,7 +86,19 @@ def supports_feature(self, feature):
             case Features.CAMERA_RTSP:
                 return self.info.device_type == "X1" or self.info.device_type == "X1C"
         return False
-
+
+    def get_active_tray(self):
+        if self.supports_feature(Features.AMS):
+            if self.ams.tray_now == 255:
+                return None
+            if self.ams.tray_now == 254:
+                return self.external_spool
+            for ams in self.ams.data:
+                active_ams = self.ams.data[math.floor(self.ams.tray_now / 4)]
+                active_tray = self.ams.tray_now % 4
+                return active_ams.tray[active_tray]
+        else:
+            return self.external_spool
 
 @dataclass
 class Lights:
@@ -536,7 +551,7 @@ def __init__(self):
 
     def print_update(self, data):
         if len(data) == 1:
-            # If the day is exactly one entry then it's just the ID and the tray is empty.
+            # If the data is exactly one entry then it's just the ID and the tray is empty.
             self.empty = True
             self.idx = ""
             self.name = "Empty"

diff --git a/custom_components/bambu_lab/translations/en.json b/custom_components/bambu_lab/translations/en.json
@@ -230,6 +230,9 @@
       "active_tray": {
         "name": "Active tray"
       },
+      "active_tray_index": {
+        "name": "Active tray index"
+      },
       "humidity_index": {
         "name": "Humidity index"
       },