kraken3: Improve LCD screen setting checks

liquidctl/driver/kraken3.py

@@ -248,11 +248,11 @@ def initialize(self, direct_access=False, **kwargs):
         self._status = []
 
         self._read_until({b"\x11\x01": self.parse_firm_info, b"\x21\x03": self.parse_led_info})
+        self._status.append(("Firmware version", f"{self.fw[0]}.{self.fw[1]}.{self.fw[2]}", ""))
         return sorted(self._status)
 
     def parse_firm_info(self, msg):
         self.fw = (msg[0x11], msg[0x12], msg[0x13])
-        self._status.append(("Firmware version", f"{self.fw[0]}.{self.fw[1]}.{self.fw[2]}", ""))
 
     def parse_led_info(self, msg):
         channel_count = msg[14]
@@ -647,6 +647,12 @@ def _find_winusb_device(self, vid, pid, serial):
                 return True
         return False
 
+    def _get_fw_version(self, clear_reports=True):
+        if clear_reports:
+            self.device.clear_enqueued_reports()
+        self._write([0x10, 0x01])  # firmware info
+        self._read_until({b"\x11\x01": self.parse_firm_info})
+
     def initialize(self, direct_access=False, **kwargs):
         """Initialize the device and the driver.
 
@@ -679,8 +685,8 @@ def initialize(self, direct_access=False, **kwargs):
         self._status = []
 
         # request static infos
-        self._write([0x10, 0x01])  # firmware info
-        self._read_until({b"\x11\x01": self.parse_firm_info})
+        self._get_fw_version(clear_reports=False)
+        self._status.append(("Firmware version", f"{self.fw[0]}.{self.fw[1]}.{self.fw[2]}", ""))
 
         self._write([0x30, 0x01])  # lcd info
         self._read_until({b"\x31\x01": self.parse_lcd_info})
@@ -772,14 +778,16 @@ def parse_lcd_info(msg):
             self.orientation = msg[0x1A]
 
         # Firmware 2.0.0 and onwards broke the implemented image setting mechanism for Kraken 2023
-        # (standard and elite). In those cases, show an error until issue #631 is resolved.
+        # (non-elite). In those cases, show an error until issue #631 is resolved.
         def unsupported_fw_version():
-            device_product_id = self.bulk_device.product_id
-            if device_product_id in (0x300C, 0x300E) and self.fw[0] == 2:
-                _LOGGER.error(
-                    "setting images is not supported on firmware 2.X.Y, please see issue #631"
-                )
-                return True
+            device_product_id = self.device.product_id
+            if device_product_id == 0x300E:
+                self._get_fw_version()
+                if self.fw[0] == 2:
+                    _LOGGER.error(
+                        "setting images is not supported on firmware 2.X.Y, please see issue #631"
+                    )
+                    return True
             return False
 
         self._read_until({b"\x31\x01": parse_lcd_info})